10 #include <boost/assert.hpp> 15 unsigned int numberOfChannels,
21 unsigned int numberOfChannels,
32 unsigned int firstAxisInclusive,
33 unsigned int lastAxisExclusive);
35 unsigned int GetUnsignedAxis(
const unsigned int inputDimension,
const int axis);
armnn::TensorShape GetTensorShape(unsigned int numberOfBatches, unsigned int numberOfChannels, unsigned int height, unsigned int width, const armnn::DataLayout dataLayout)
std::pair< float, float > FindMinMax(armnn::ITensorHandle *tensorHandle)
unsigned int GetUnsignedAxis(const unsigned int inputDimension, const int axis)
armnn::TensorInfo GetTensorInfo(unsigned int numberOfBatches, unsigned int numberOfChannels, unsigned int height, unsigned int width, const armnn::DataLayout dataLayout, const armnn::DataType dataType)
std::pair< unsigned int, std::vector< float > > GetPerAxisParams(const armnn::TensorInfo &info)
armnn::TensorShape ExpandDims(const armnn::TensorShape &tensorShape, int axis)
unsigned int GetNumElementsBetween(const armnn::TensorShape &shape, unsigned int firstAxisInclusive, unsigned int lastAxisExclusive)
unsigned int GetNumElementsAfter(const armnn::TensorShape &shape, unsigned int axis)