TensorUtils.cpp File Reference

#include <armnnUtils/TensorUtils.hpp>
#include <armnn/backends/ITensorHandle.hpp>
#include <armnn/utility/Assert.hpp>
#include <armnn/utility/NumericCast.hpp>
#include <fmt/format.h>

Go to the source code of this file.

Namespaces
	armnnUtils

Functions
armnn::TensorShape	GetTensorShape (unsigned int numberOfBatches, unsigned int numberOfChannels, unsigned int height, unsigned int width, const armnn::DataLayout dataLayout)
armnn::TensorInfo	GetTensorInfo (unsigned int numberOfBatches, unsigned int numberOfChannels, unsigned int height, unsigned int width, const armnn::DataLayout dataLayout, const armnn::DataType dataType)
armnn::TensorInfo	GetTensorInfo (unsigned int numberOfBatches, unsigned int numberOfChannels, unsigned int depth, unsigned int height, unsigned int width, const armnn::DataLayout dataLayout, const armnn::DataType dataType)
std::pair< float, float >	FindMinMax (armnn::ITensorHandle *tensorHandle)
armnn::TensorShape	ReduceDims (const armnn::TensorShape &tensorInfo, unsigned int dimensions)
armnn::TensorInfo	ReduceDims (const armnn::TensorInfo &tensorInfo, unsigned int dimensions)
armnn::TensorShape	ExpandDims (const armnn::TensorShape &tensorShape, int axis)
armnn::TensorShape	ExpandDimsToRank (const armnn::TensorShape &tensorShape, unsigned int rank)
std::vector< unsigned int >	SqueezeDims (const armnn::TensorShape &tensorShape)
unsigned int	GetNumElementsBetween (const armnn::TensorShape &shape, unsigned int firstAxisInclusive, unsigned int lastAxisExclusive)
unsigned int	GetUnsignedAxis (const unsigned int inputDimension, const int axis)
unsigned int	GetNumElementsAfter (const armnn::TensorShape &shape, unsigned int axis)
std::pair< unsigned int, std::vector< float > >	GetPerAxisParams (const armnn::TensorInfo &info)
template<typename PrimitiveType >
void	CheckSizes (const std::vector< PrimitiveType > &data, const armnn::TensorInfo &tensorInfo, unsigned int size=1)
template<typename PrimitiveType >
std::unique_ptr< float[]>	ToFloatArray (const std::vector< PrimitiveType > &data, const armnn::TensorInfo &tensorInfo)
std::unique_ptr< float[]>	ToFloatArray (const std::vector< uint8_t > &data, const armnn::TensorInfo &tensorInfo)