• Home
  • Line#
  • Scopes#
  • Navigate#
  • Raw
  • Download
1 //
2 // Copyright © 2017 Arm Ltd. All rights reserved.
3 // SPDX-License-Identifier: MIT
4 //
5 
6 #include "TensorCopyUtils.hpp"
7 #include <Half.hpp>
8 
CopyDataToITensorHandle(armnn::ITensorHandle * tensorHandle,const void * memory)9 void CopyDataToITensorHandle(armnn::ITensorHandle* tensorHandle, const void* memory)
10 {
11     tensorHandle->CopyInFrom(memory);
12 }
13 
CopyDataFromITensorHandle(void * memory,const armnn::ITensorHandle * tensorHandle)14 void CopyDataFromITensorHandle(void* memory, const armnn::ITensorHandle* tensorHandle)
15 {
16     tensorHandle->CopyOutTo(memory);
17 }
18 
AllocateAndCopyDataToITensorHandle(armnn::ITensorHandle * tensorHandle,const void * memory)19 void AllocateAndCopyDataToITensorHandle(armnn::ITensorHandle* tensorHandle, const void* memory)
20 {
21     tensorHandle->Allocate();
22     CopyDataToITensorHandle(tensorHandle, memory);
23 }
24