RefWorkloadUtils.hpp File Reference

#include <armnn/backends/TensorHandle.hpp>
#include <armnn/Tensor.hpp>
#include <armnn/Types.hpp>
#include <armnn/utility/PolymorphicDowncast.hpp>
#include <reference/RefTensorHandle.hpp>
#include <BFloat16.hpp>
#include <Half.hpp>

Include dependency graph for RefWorkloadUtils.hpp:

This graph shows which files directly or indirectly include this file:

Go to the source code of this file.

Namespaces
	armnn
	Copyright (c) 2021 ARM Limited and Contributors.

Macros
#define	ARMNN_SCOPED_PROFILING_EVENT_REF_NAME_GUID(label)
	Creates a profiling event that uses GetGuid() and GetName() from the calling class. More...

Functions
template<typename TensorHandleType = RefTensorHandle>
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 DataType >
DataType *	GetOutputTensorData (ITensorHandle *tensorHandle)
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 PayloadType >
const BFloat16 *	GetInputTensorDataBFloat16 (unsigned int idx, const PayloadType &data)
template<typename PayloadType >
BFloat16 *	GetOutputTensorDataBFloat16 (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)

Macro Definition Documentation

ARMNN_SCOPED_PROFILING_EVENT_REF_NAME_GUID

#define ARMNN_SCOPED_PROFILING_EVENT_REF_NAME_GUID

(

label

)

Value:

ARMNN_SCOPED_PROFILING_EVENT_WITH_INSTRUMENTS(armnn::Compute::CpuRef, \
                                              this->GetGuid(), \
                                              this->GetName() + "_" + label, \
                                              armnn::WallClockTimer())

Creates a profiling event that uses GetGuid() and GetName() from the calling class.

Definition at line 22 of file RefWorkloadUtils.hpp.