TensorCopyUtils.cpp File Reference

#include <armnnTestUtils/TensorCopyUtils.hpp>
#include <Half.hpp>

Go to the source code of this file.

Functions
void	CopyDataToITensorHandle (armnn::ITensorHandle *tensorHandle, const void *memory)
void	CopyDataFromITensorHandle (void *memory, const armnn::ITensorHandle *tensorHandle)
void	AllocateAndCopyDataToITensorHandle (armnn::ITensorHandle *tensorHandle, const void *memory)

Function Documentation

AllocateAndCopyDataToITensorHandle()

void AllocateAndCopyDataToITensorHandle	(	armnn::ITensorHandle *	tensorHandle,
		const void *	memory
	)

Definition at line 19 of file TensorCopyUtils.cpp.

References ITensorHandle::Allocate(), and CopyDataToITensorHandle().

{
    tensorHandle->Allocate();
    CopyDataToITensorHandle(tensorHandle, memory);
}

CopyDataFromITensorHandle()

void CopyDataFromITensorHandle	(	void *	memory,
		const armnn::ITensorHandle *	tensorHandle
	)

Definition at line 14 of file TensorCopyUtils.cpp.

References ITensorHandle::CopyOutTo().

{
    tensorHandle->CopyOutTo(memory);
}

CopyDataToITensorHandle()

void CopyDataToITensorHandle	(	armnn::ITensorHandle *	tensorHandle,
		const void *	memory
	)

Definition at line 9 of file TensorCopyUtils.cpp.

References ITensorHandle::CopyInFrom().

Referenced by AllocateAndCopyDataToITensorHandle().

{
    tensorHandle->CopyInFrom(memory);
}