diff options
Diffstat (limited to 'src/backends/backendsCommon/CpuTensorHandle.cpp')
-rw-r--r-- | src/backends/backendsCommon/CpuTensorHandle.cpp | 10 |
1 files changed, 10 insertions, 0 deletions
diff --git a/src/backends/backendsCommon/CpuTensorHandle.cpp b/src/backends/backendsCommon/CpuTensorHandle.cpp index 46a7cb8251..fe0c634e7c 100644 --- a/src/backends/backendsCommon/CpuTensorHandle.cpp +++ b/src/backends/backendsCommon/CpuTensorHandle.cpp @@ -84,6 +84,16 @@ void ScopedCpuTensorHandle::Allocate() } } +void ScopedCpuTensorHandle::CopyOutTo(void* memory) const +{ + memcpy(memory, GetTensor<void>(), GetTensorInfo().GetNumBytes()); +} + +void ScopedCpuTensorHandle::CopyInFrom(const void* memory) +{ + memcpy(GetTensor<void>(), memory, GetTensorInfo().GetNumBytes()); +} + void ScopedCpuTensorHandle::CopyFrom(const ScopedCpuTensorHandle& other) { CopyFrom(other.GetTensor<void>(), other.GetTensorInfo().GetNumBytes()); |