TensorCopyUtils.cpp File Reference

#include <armnn/backends/ITensorHandle.hpp>
#include <armnnTestUtils/TensorCopyUtils.hpp>
#include <Half.hpp>

Include dependency graph for TensorCopyUtils.cpp:

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 20 of file TensorCopyUtils.cpp.

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

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

CopyDataFromITensorHandle()

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

Definition at line 15 of file TensorCopyUtils.cpp.

{
    tensorHandle->CopyOutTo(memory);
}

References ITensorHandle::CopyOutTo().

CopyDataToITensorHandle()

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

Definition at line 10 of file TensorCopyUtils.cpp.

{
    tensorHandle->CopyInFrom(memory);
}

References ITensorHandle::CopyInFrom().

Referenced by AllocateAndCopyDataToITensorHandle().