diff options
author | Narumol Prangnawarat <narumol.prangnawarat@arm.com> | 2020-06-26 11:00:21 +0100 |
---|---|---|
committer | Narumol Prangnawarat <narumol.prangnawarat@arm.com> | 2020-06-30 08:39:55 +0000 |
commit | 3b90af6bb79639d3813f21f3c3017503ee0c66af (patch) | |
tree | c5cccdb45ed268a47f5b4c8ac39a389fe037597c /src/backends/reference/RefTensorHandle.hpp | |
parent | c0761e9a82ec4d6bd4a81fda018d624659b629a7 (diff) | |
download | armnn-3b90af6bb79639d3813f21f3c3017503ee0c66af.tar.gz |
IVGCVSW-5036 Do not allocate memory when import is enabled
Signed-off-by: Narumol Prangnawarat <narumol.prangnawarat@arm.com>
Change-Id: Ideaae5280702aae6c73f3b4e4cee9f71a8386fda
Diffstat (limited to 'src/backends/reference/RefTensorHandle.hpp')
-rw-r--r-- | src/backends/reference/RefTensorHandle.hpp | 6 |
1 files changed, 3 insertions, 3 deletions
diff --git a/src/backends/reference/RefTensorHandle.hpp b/src/backends/reference/RefTensorHandle.hpp index 6cde3263a0..8c64dfbe63 100644 --- a/src/backends/reference/RefTensorHandle.hpp +++ b/src/backends/reference/RefTensorHandle.hpp @@ -17,8 +17,7 @@ class RefTensorHandle : public ITensorHandle public: RefTensorHandle(const TensorInfo& tensorInfo, std::shared_ptr<RefMemoryManager> &memoryManager); - RefTensorHandle(const TensorInfo& tensorInfo, std::shared_ptr<RefMemoryManager> &memoryManager, - MemorySourceFlags importFlags); + RefTensorHandle(const TensorInfo& tensorInfo, MemorySourceFlags importFlags); ~RefTensorHandle(); @@ -73,9 +72,10 @@ private: std::shared_ptr<RefMemoryManager> m_MemoryManager; RefMemoryManager::Pool* m_Pool; - mutable void *m_UnmanagedMemory; + mutable void* m_UnmanagedMemory; MemorySourceFlags m_ImportFlags; bool m_Imported; + bool m_IsImportEnabled; }; } |