diff options
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 |