diff options
author | Narumol Prangnawarat <narumol.prangnawarat@arm.com> | 2020-08-14 11:51:12 +0100 |
---|---|---|
committer | Jim Flynn <jim.flynn@arm.com> | 2020-08-16 14:47:00 +0000 |
commit | b8d771ac2e6f847a64b3b203591c5b1c3e198d3a (patch) | |
tree | 289b769e179e495e45f2d2c2f9374703be32f9c2 /src/backends/neon/NeonTensorHandleFactory.hpp | |
parent | 9e132f57e3fc9d1cd12e3bca2dd3eb82549d7d84 (diff) | |
download | armnn-b8d771ac2e6f847a64b3b203591c5b1c3e198d3a.tar.gz |
IVGCVSW-5012 Enable zero copy for Neon
* Allow memory import if padding is not required in Neon
* AddMockImportBackend for fallback tests
* Refactor GraphUtils
* Memory import unit tests
* Fallback unit tests
Signed-off-by: Narumol Prangnawarat <narumol.prangnawarat@arm.com>
Change-Id: Ic2e141e12774bf6d915e77745b6f6d2d83d9b82d
Diffstat (limited to 'src/backends/neon/NeonTensorHandleFactory.hpp')
-rw-r--r-- | src/backends/neon/NeonTensorHandleFactory.hpp | 6 |
1 files changed, 5 insertions, 1 deletions
diff --git a/src/backends/neon/NeonTensorHandleFactory.hpp b/src/backends/neon/NeonTensorHandleFactory.hpp index ae45aadb7c..2ca67c9d6e 100644 --- a/src/backends/neon/NeonTensorHandleFactory.hpp +++ b/src/backends/neon/NeonTensorHandleFactory.hpp @@ -41,7 +41,9 @@ class NeonTensorHandleFactory : public ITensorHandleFactory { public: NeonTensorHandleFactory(std::weak_ptr<NeonMemoryManager> mgr) - : m_MemoryManager(mgr) + : m_MemoryManager(mgr), + m_ImportFlags(static_cast<MemorySourceFlags>(MemorySource::Malloc)), + m_ExportFlags(static_cast<MemorySourceFlags>(MemorySource::Malloc)) {} std::unique_ptr<ITensorHandle> CreateSubTensorHandle(ITensorHandle& parent, @@ -76,6 +78,8 @@ public: private: mutable std::shared_ptr<NeonMemoryManager> m_MemoryManager; + MemorySourceFlags m_ImportFlags; + MemorySourceFlags m_ExportFlags; }; } // namespace armnn |