RefWorkloadUtils.hpp File Reference

#include <backendsCommon/CpuTensorHandle.hpp>
#include <armnn/Tensor.hpp>
#include <armnn/Types.hpp>
#include <reference/RefTensorHandle.hpp>
#include <Half.hpp>
#include <boost/polymorphic_cast.hpp>

Go to the source code of this file.

Namespaces
	armnn
	Copyright (c) 2020 ARM Limited.

Functions
const TensorInfo &	GetTensorInfo (const ITensorHandle *tensorHandle)
	float32 helpers More...
template<typename DataType , typename PayloadType >
const DataType *	GetInputTensorData (unsigned int idx, const PayloadType &data)
template<typename DataType , typename PayloadType >
DataType *	GetOutputTensorData (unsigned int idx, const PayloadType &data)
template<typename PayloadType >
const float *	GetInputTensorDataFloat (unsigned int idx, const PayloadType &data)
template<typename PayloadType >
float *	GetOutputTensorDataFloat (unsigned int idx, const PayloadType &data)
template<typename PayloadType >
const Half *	GetInputTensorDataHalf (unsigned int idx, const PayloadType &data)
template<typename PayloadType >
Half *	GetOutputTensorDataHalf (unsigned int idx, const PayloadType &data)
template<typename T >
std::vector< float >	Dequantize (const T *quant, const TensorInfo &info)
	u8 helpers More...
template<typename T >
void	Dequantize (const T *inputData, float *outputData, const TensorInfo &info)
void	Quantize (uint8_t *quant, const float *dequant, const TensorInfo &info)