aboutsummaryrefslogtreecommitdiff
path: root/src/dynamic/sample/SampleTensorHandle.cpp
diff options
context:
space:
mode:
Diffstat (limited to 'src/dynamic/sample/SampleTensorHandle.cpp')
-rw-r--r--src/dynamic/sample/SampleTensorHandle.cpp19
1 files changed, 16 insertions, 3 deletions
diff --git a/src/dynamic/sample/SampleTensorHandle.cpp b/src/dynamic/sample/SampleTensorHandle.cpp
index 48f8cf44fa..fb4eddcb55 100644
--- a/src/dynamic/sample/SampleTensorHandle.cpp
+++ b/src/dynamic/sample/SampleTensorHandle.cpp
@@ -21,14 +21,13 @@ SampleTensorHandle::SampleTensorHandle(const TensorInfo &tensorInfo,
}
SampleTensorHandle::SampleTensorHandle(const TensorInfo& tensorInfo,
- std::shared_ptr<SampleMemoryManager> &memoryManager,
MemorySourceFlags importFlags)
: m_TensorInfo(tensorInfo),
- m_MemoryManager(memoryManager),
+ m_MemoryManager(nullptr),
m_Pool(nullptr),
m_UnmanagedMemory(nullptr),
m_ImportFlags(importFlags),
- m_Imported(false)
+ m_Imported(true)
{
}
@@ -134,4 +133,18 @@ bool SampleTensorHandle::Import(void* memory, MemorySource source)
return false;
}
+void SampleTensorHandle::CopyOutTo(void* dest) const
+{
+ const void *src = GetPointer();
+ ARMNN_ASSERT(src);
+ memcpy(dest, src, m_TensorInfo.GetNumBytes());
+}
+
+void SampleTensorHandle::CopyInFrom(const void* src)
+{
+ void *dest = GetPointer();
+ ARMNN_ASSERT(dest);
+ memcpy(dest, src, m_TensorInfo.GetNumBytes());
+}
+
}