diff options
Diffstat (limited to 'src/backends/neon')
-rw-r--r-- | src/backends/neon/NeonTensorHandleFactory.cpp | 8 | ||||
-rw-r--r-- | src/backends/neon/NeonTensorHandleFactory.hpp | 6 |
2 files changed, 5 insertions, 9 deletions
diff --git a/src/backends/neon/NeonTensorHandleFactory.cpp b/src/backends/neon/NeonTensorHandleFactory.cpp index 4ccbb7b64f..80f46d2237 100644 --- a/src/backends/neon/NeonTensorHandleFactory.cpp +++ b/src/backends/neon/NeonTensorHandleFactory.cpp @@ -60,7 +60,7 @@ std::unique_ptr<ITensorHandle> NeonTensorHandleFactory::CreateTensorHandle(const } // If we are not Managing the Memory then we must be importing tensorHandle->SetImportEnabledFlag(!IsMemoryManaged); - tensorHandle->SetImportFlags(m_ImportFlags); + tensorHandle->SetImportFlags(GetImportFlags()); return tensorHandle; } @@ -76,7 +76,7 @@ std::unique_ptr<ITensorHandle> NeonTensorHandleFactory::CreateTensorHandle(const } // If we are not Managing the Memory then we must be importing tensorHandle->SetImportEnabledFlag(!IsMemoryManaged); - tensorHandle->SetImportFlags(m_ImportFlags); + tensorHandle->SetImportFlags(GetImportFlags()); return tensorHandle; } @@ -99,12 +99,12 @@ bool NeonTensorHandleFactory::SupportsSubTensors() const MemorySourceFlags NeonTensorHandleFactory::GetExportFlags() const { - return m_ExportFlags; + return 0; } MemorySourceFlags NeonTensorHandleFactory::GetImportFlags() const { - return m_ImportFlags; + return static_cast<MemorySourceFlags>(MemorySource::Malloc); } } // namespace armnn diff --git a/src/backends/neon/NeonTensorHandleFactory.hpp b/src/backends/neon/NeonTensorHandleFactory.hpp index d9b64045e6..8a8ac5cdcb 100644 --- a/src/backends/neon/NeonTensorHandleFactory.hpp +++ b/src/backends/neon/NeonTensorHandleFactory.hpp @@ -17,9 +17,7 @@ class NeonTensorHandleFactory : public ITensorHandleFactory { public: NeonTensorHandleFactory(std::weak_ptr<NeonMemoryManager> mgr) - : m_MemoryManager(mgr), - m_ImportFlags(static_cast<MemorySourceFlags>(MemorySource::Malloc)), - m_ExportFlags(static_cast<MemorySourceFlags>(MemorySource::Malloc)) + : m_MemoryManager(mgr) {} std::unique_ptr<ITensorHandle> CreateSubTensorHandle(ITensorHandle& parent, @@ -50,8 +48,6 @@ public: private: mutable std::shared_ptr<NeonMemoryManager> m_MemoryManager; - MemorySourceFlags m_ImportFlags; - MemorySourceFlags m_ExportFlags; }; } // namespace armnn |