From c30abd843e68dfbd186ca25e8a8ecaefcf95776f Mon Sep 17 00:00:00 2001 From: Matthew Bentham Date: Wed, 23 Nov 2022 12:11:32 +0000 Subject: 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 Change-Id: Ic629858920f7dd32f99ee27f150b81d8b67144cf --- src/backends/reference/RefTensorHandle.hpp | 8 ++------ 1 file changed, 2 insertions(+), 6 deletions(-) (limited to 'src/backends/reference/RefTensorHandle.hpp') 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 &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 m_MemoryManager; RefMemoryManager::Pool* m_Pool; mutable void* m_UnmanagedMemory; - MemorySourceFlags m_ImportFlags; bool m_Imported; bool m_IsImportEnabled; }; -- cgit v1.2.1