diff options
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; }; } |