diff options
author | Matthew Bentham <matthew.bentham@arm.com> | 2022-11-23 12:11:32 +0000 |
---|---|---|
committer | TeresaARM <teresa.charlinreyes@arm.com> | 2022-12-14 12:53:00 +0000 |
commit | c30abd843e68dfbd186ca25e8a8ecaefcf95776f (patch) | |
tree | e2f17d59df2dbef75cc056dd355560df22e93d78 /src/backends/reference/RefTensorHandle.hpp | |
parent | 6d2647df4ce2e15bff8548e74993aa4b12ea8f34 (diff) | |
download | armnn-c30abd843e68dfbd186ca25e8a8ecaefcf95776f.tar.gz |
Refactor: Remove m_ImportFlags from RefTensorHandle
The import flags for a RefTensorHandle shouldn't be a data member,
as RefTensorHandle can only import from MemorySource::Malloc. Instead,
use m_ImportEnabled to determine what to return from GetImportFlags().
Simplifies the code in Import and CanBeImported.
Signed-off-by: Matthew Bentham <matthew.bentham@arm.com>
Change-Id: Ic629858920f7dd32f99ee27f150b81d8b67144cf
Diffstat (limited to 'src/backends/reference/RefTensorHandle.hpp')
-rw-r--r-- | src/backends/reference/RefTensorHandle.hpp | 8 |
1 files changed, 2 insertions, 6 deletions
diff --git a/src/backends/reference/RefTensorHandle.hpp b/src/backends/reference/RefTensorHandle.hpp index a7eab034b2..d916b39ed9 100644 --- a/src/backends/reference/RefTensorHandle.hpp +++ b/src/backends/reference/RefTensorHandle.hpp @@ -17,7 +17,7 @@ class RefTensorHandle : public ITensorHandle public: RefTensorHandle(const TensorInfo& tensorInfo, std::shared_ptr<RefMemoryManager> &memoryManager); - RefTensorHandle(const TensorInfo& tensorInfo, MemorySourceFlags importFlags); + RefTensorHandle(const TensorInfo& tensorInfo); ~RefTensorHandle(); @@ -51,10 +51,7 @@ public: return m_TensorInfo; } - virtual MemorySourceFlags GetImportFlags() const override - { - return m_ImportFlags; - } + virtual MemorySourceFlags GetImportFlags() const override; virtual bool Import(void* memory, MemorySource source) override; virtual bool CanBeImported(void* memory, MemorySource source) override; @@ -74,7 +71,6 @@ private: std::shared_ptr<RefMemoryManager> m_MemoryManager; RefMemoryManager::Pool* m_Pool; mutable void* m_UnmanagedMemory; - MemorySourceFlags m_ImportFlags; bool m_Imported; bool m_IsImportEnabled; }; |