From c30abd843e68dfbd186ca25e8a8ecaefcf95776f Mon Sep 17 00:00:00 2001
From: Matthew Bentham <matthew.bentham@arm.com>
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 <matthew.bentham@arm.com>
Change-Id: Ic629858920f7dd32f99ee27f150b81d8b67144cf
---
 src/backends/reference/RefTensorHandleFactory.cpp | 6 +++---
 1 file changed, 3 insertions(+), 3 deletions(-)

(limited to 'src/backends/reference/RefTensorHandleFactory.cpp')

diff --git a/src/backends/reference/RefTensorHandleFactory.cpp b/src/backends/reference/RefTensorHandleFactory.cpp
index ade27dd733..da3b798d3d 100644
--- a/src/backends/reference/RefTensorHandleFactory.cpp
+++ b/src/backends/reference/RefTensorHandleFactory.cpp
@@ -48,7 +48,7 @@ std::unique_ptr<ITensorHandle> RefTensorHandleFactory::CreateTensorHandle(const
     }
     else
     {
-        return std::make_unique<RefTensorHandle>(tensorInfo, m_ImportFlags);
+        return std::make_unique<RefTensorHandle>(tensorInfo);
     }
 }
 
@@ -63,7 +63,7 @@ std::unique_ptr<ITensorHandle> RefTensorHandleFactory::CreateTensorHandle(const
     }
     else
     {
-        return std::make_unique<RefTensorHandle>(tensorInfo, m_ImportFlags);
+        return std::make_unique<RefTensorHandle>(tensorInfo);
     }
 }
 
@@ -87,4 +87,4 @@ MemorySourceFlags RefTensorHandleFactory::GetImportFlags() const
     return m_ImportFlags;
 }
 
-} // namespace armnn
\ No newline at end of file
+} // namespace armnn
-- 
cgit v1.2.1