#include <Tensor.hpp>

Public Member Functions
	TensorShape ()
	Empty (invalid) constructor. More...

	TensorShape (unsigned int numDimensions, bool initDimensionsSpecificity=true)
	Constructor for TensorShape. More...

	TensorShape (unsigned int numDimensions, const unsigned int *dimensionSizes)
	Constructor for TensorShape. More...

	TensorShape (std::initializer_list< unsigned int > dimensionSizeList)
	Constructor for TensorShape. More...

	TensorShape (const TensorShape &other)
	Copy Constructor for TensorShape. More...

	TensorShape (unsigned int numDimensions, const unsigned int dimensionSizes, const bool dimensionsSpecificity)
	Constructor for TensorShape. More...

	TensorShape (std::initializer_list< unsigned int > dimensionSizeList, std::initializer_list< bool > dimensionsSpecificityList)
	Constructor for TensorShape. More...

	TensorShape (Dimensionality dimensionality)
	Constructor for TensorShape. More...

TensorShape &	operator= (const TensorShape &other)
	Assignation function. More...

unsigned int	operator[] (unsigned int i) const
	Read only operator. More...

unsigned int &	operator[] (unsigned int i)
	Read and write operator. More...

bool	operator== (const TensorShape &other) const
	Equality comparison operator. More...

bool	operator!= (const TensorShape &other) const
	Inequality comparison operator. More...

unsigned int	GetNumDimensions () const
	Function that returns the tensor rank. More...

unsigned int	GetNumElements () const
	Function that calculates the tensor elements by multiplying all dimension size which are Specified. More...

Dimensionality	GetDimensionality () const
	Function that returns the tensor type. More...

bool	GetDimensionSpecificity (unsigned int i) const
	Gets information about if the dimension size has been specified or not. More...

void	SetNumDimensions (unsigned int numDimensions, bool initDimensionsSpecificity=false)
	Sets the tensor rank and therefore the Dimensionality is set to Specified if it was not. More...

void	SetDimensionSize (unsigned int i, unsigned int dimensionSize)
	Sets the size of the indicated dimension and Specificity for that dimension is set to true. More...

bool	AreAllDimensionsSpecified () const
	Checks if there is at least one dimension not specified. More...

bool	IsAtLeastOneDimensionSpecified () const
	Checks if there is at least one dimension specified. More...

Detailed Description

Definition at line 20 of file Tensor.hpp.

Constructor & Destructor Documentation

◆ TensorShape() [1/8]

TensorShape ( )

Empty (invalid) constructor.

Definition at line 26 of file Tensor.cpp.

References armnn::Specified.

Referenced by TensorShape::TensorShape().

  : m_NumDimensions(0), m_Dimensionality(Dimensionality::Specified)
 {
 }

◆ TensorShape() [2/8]

TensorShape	(	unsigned int	numDimensions,
		bool	initDimensionsSpecificity = `true`
	)

explicit

Constructor for TensorShape.

Parameters

numDimensions	- Tensor rank.
initDimensionsSpecificity	(optional) - value to initialize the specificity of each dimension size.

Definition at line 31 of file Tensor.cpp.

  : m_NumDimensions(numDimensions), m_Dimensionality(Dimensionality::Specified)
 {
     CheckValidNumDimensions(numDimensions);
 
     std::fill(m_Dimensions.begin(), m_Dimensions.begin() + m_NumDimensions, 0);
     std::fill(m_DimensionsSpecificity.begin(), m_DimensionsSpecificity.begin() + m_NumDimensions,
               initDimensionsSpecificity);
 }

◆ TensorShape() [3/8]

TensorShape	(	unsigned int	numDimensions,
		const unsigned int *	dimensionSizes
	)

Constructor for TensorShape.

Parameters

numDimensions	- Tensor rank.
dimensionSizes	- Size of each of dimension.

Definition at line 41 of file Tensor.cpp.

  : m_NumDimensions(numDimensions), m_Dimensionality(Dimensionality::Specified)
 {
     CheckValidNumDimensions(numDimensions);
 
     if (dimensionSizes == nullptr)
     {
         throw InvalidArgumentException("Tensor dimensionSizes must not be NULL");
     }
 
     std::copy(dimensionSizes, dimensionSizes + numDimensions, m_Dimensions.begin());
     std::fill(m_DimensionsSpecificity.begin(), m_DimensionsSpecificity.begin() + m_NumDimensions, true);
 }

◆ TensorShape() [4/8]

TensorShape ( std::initializer_list< unsigned int > dimensionSizeList )

Constructor for TensorShape.

Parameters

dimensionSizeList - Size of each of dimension.

Definition at line 55 of file Tensor.cpp.

  : TensorShape(boost::numeric_cast<unsigned int>(dimensionSizeList.size()), dimensionSizeList.begin())
 {
 }

◆ TensorShape() [5/8]

TensorShape ( const TensorShape & other )

Copy Constructor for TensorShape.

Parameters

other - TensorShape to copy from.

Definition at line 117 of file Tensor.cpp.

  : m_NumDimensions(other.m_NumDimensions), m_Dimensionality(other.m_Dimensionality)
 {
     std::copy(other.m_Dimensions.cbegin(), other.m_Dimensions.cbegin() + other.m_NumDimensions, m_Dimensions.begin());
     std::copy(other.m_DimensionsSpecificity.cbegin(), other.m_DimensionsSpecificity.cbegin() + other.m_NumDimensions,
               m_DimensionsSpecificity.begin());
 }

◆ TensorShape() [6/8]

TensorShape	(	unsigned int	numDimensions,
		const unsigned int *	dimensionSizes,
		const bool *	dimensionsSpecificity
	)

Constructor for TensorShape.

Parameters

numDimensions	- Tensor rank.
dimensionSizes	- Size of each of dimension.
dimensionsSpecificity	- Flags to indicate which dimension has its size specified.

Definition at line 60 of file Tensor.cpp.

                        : m_NumDimensions(numDimensions), m_Dimensionality(Dimensionality::Specified)
 {
     CheckValidNumDimensions(numDimensions);
 
     if (dimensionSizes == nullptr)
     {
         throw InvalidArgumentException("Tensor dimensionSizes must not be NULL");
     }
 
     if (dimensionsSpecificity == nullptr)
     {
         throw InvalidArgumentException("Tensor dimensionsSpecificity must not be NULL");
     }
 
     std::copy(dimensionSizes, dimensionSizes + numDimensions, m_Dimensions.begin());
     std::copy(dimensionsSpecificity, dimensionsSpecificity + numDimensions, m_DimensionsSpecificity.begin());
 }

◆ TensorShape() [7/8]

TensorShape	(	std::initializer_list< unsigned int >	dimensionSizeList,
		std::initializer_list< bool >	dimensionsSpecificityList
	)

Constructor for TensorShape.

Parameters

dimensionSizeList	- Size of each of dimension.
dimensionsSpecificityList	- Flags to indicate which dimension size is specified.

Definition at line 81 of file Tensor.cpp.

References TensorShape::TensorShape().

 {
     auto numDimensions = static_cast<unsigned int>(dimensionSizeList.size());
     if (dimensionsSpecificityList.size() != numDimensions)
     {
         throw InvalidArgumentException("Tensors dimensionSizeList and dimensionsSpecificityList must be same size");
     }
 
     *this = TensorShape(numDimensions, dimensionSizeList.begin(), dimensionsSpecificityList.begin());
 }

◆ TensorShape() [8/8]

TensorShape ( Dimensionality dimensionality )

explicit

Constructor for TensorShape.

Parameters

dimensionality - Parameter to indicate if the Tensor is a Scalar, a Tensor of known dimensionality or a Tensor of unknown dimensionality.

Definition at line 93 of file Tensor.cpp.

References armnn::NotSpecified, armnn::Scalar, and armnn::Specified.

 : m_Dimensionality(dimensionality)
 {
     switch (dimensionality)
     {
         case Dimensionality::Specified:
             throw InvalidArgumentException("Use other constructor to specify the rest of the values, this one is only "
                                            "for tensors that have an unknown number of dimensions or that are scalar");
             break;
         case Dimensionality::NotSpecified:
             m_NumDimensions = 0;
             m_Dimensions = {0};
             m_DimensionsSpecificity = {false};
             break;
         case Dimensionality::Scalar:
             m_NumDimensions = 1;
             m_Dimensions = {1};
             m_DimensionsSpecificity = {true};
             break;
         default:
             throw InvalidArgumentException("Invalid Dimensionality value");
     }
 }

Member Function Documentation

◆ AreAllDimensionsSpecified()

bool AreAllDimensionsSpecified ( ) const

Checks if there is at least one dimension not specified.

AND of all array elements.

Returns: - True when all dimension sizes are specified. False when at least one dimension size is not specified.

Definition at line 242 of file Tensor.cpp.

Referenced by BOOST_AUTO_TEST_CASE(), TensorShape::GetDimensionality(), ConstantLayer::ValidateTensorShapesFromInputs(), and Layer::VerifyShapeInferenceType().

 {
     CheckUnspecifiedNumDimensions();
 
     bool areAllDimensionsSpecified = true;
     for (unsigned int i = 0; i < m_NumDimensions; ++i)
     {
         if (!m_DimensionsSpecificity[i])
         {
             areAllDimensionsSpecified = false;
             break;
         }
     }
     return areAllDimensionsSpecified;
 }

◆ GetDimensionality()

Dimensionality GetDimensionality ( ) const

inline

Function that returns the tensor type.

Returns: - Parameter to indicate if the Tensor is a scalar, a Tensor of known dimensionality or a Tensor of unknown dimensionality

Definition at line 92 of file Tensor.hpp.

References TensorShape::AreAllDimensionsSpecified(), TensorShape::GetDimensionSpecificity(), TensorShape::IsAtLeastOneDimensionSpecified(), TensorShape::SetDimensionSize(), TensorShape::SetNumDimensions(), and armnn::Specified.

Referenced by BOOST_AUTO_TEST_CASE(), Layer::ValidateAndCopyShape(), ConstantLayer::ValidateTensorShapesFromInputs(), Layer::VerifyShapeInferenceType(), and SerializerVisitor::VisitQuantizedLstmLayer().

92 { return m_Dimensionality; }

◆ GetDimensionSpecificity()

bool GetDimensionSpecificity ( unsigned int i ) const

Gets information about if the dimension size has been specified or not.

Parameters

i	- Dimension index.

Returns: - Flag to indicate if the dimension "i" has a specified size.

Definition at line 212 of file Tensor.cpp.

Referenced by BOOST_AUTO_TEST_CASE(), TensorShape::GetDimensionality(), and Layer::ValidateAndCopyShape().

 {
     CheckUnspecifiedNumDimensions();
     CheckDimensionIndex(i);
 
     return m_DimensionsSpecificity[i];
 }

◆ GetNumDimensions()

unsigned int GetNumDimensions ( ) const

Function that returns the tensor rank.

Returns: - Tensor rank.

Definition at line 175 of file Tensor.cpp.

Referenced by armnn::BatchToSpaceNd(), BOOST_AUTO_TEST_CASE(), armnn::boost_test_print_type(), ShapesAreBroadcastCompatible::CalcInputSize(), armnnDeserializer::CheckShape(), ClStackWorkload::ClStackWorkload(), armnnUtils::ConvertTensorShapeToString(), armnn::CopyTensorContentsGeneric(), OnnxParser::CreateNetworkFromString(), RefL2NormalizationWorkload::Execute(), armnnUtils::ExpandDims(), ExpandTensorShapeTo3dForPermute(), armnnUtils::GetNumElementsAfter(), armnnUtils::GetNumElementsBetween(), ArgMinMaxLayer::InferOutputShapes(), ComparisonLayer::InferOutputShapes(), PreluLayer::InferOutputShapes(), StackLayer::InferOutputShapes(), PadLayer::InferOutputShapes(), armnn::LogSoftmax(), NeonStackWorkload::NeonStackWorkload(), armnn::operator<<(), armnnTfParser::OutputShapeOfExpandDims(), armnnTfParser::OutputShapeOfSqueeze(), armnn::Pad(), armnnUtils::Permuted(), ParserPrototxtFixture< armnnOnnxParser::IOnnxParser >::RunTest(), ShapesAreBroadcastCompatible::ShapesAreBroadcastCompatible(), ShapesAreSameRank::ShapesAreSameRank(), armnn::Slice(), armnn::Softmax(), TensorBufferArrayView< DataType >::TensorBufferArrayView(), TfParser::TfParser(), armnnUtils::TransposeTensorShape(), StackQueueDescriptor::Validate(), ArgMinMaxQueueDescriptor::Validate(), Layer::ValidateAndCopyShape(), SerializerVisitor::VisitQuantizedLstmLayer(), SerializerVisitor::VisitReshapeLayer(), and SerializerVisitor::VisitStackLayer().

 {
     CheckUnspecifiedNumDimensions();
 
     return m_NumDimensions;
 }

◆ GetNumElements()

unsigned int GetNumElements ( ) const

Function that calculates the tensor elements by multiplying all dimension size which are Specified.

Returns: - Total number of elements in the tensor.

Definition at line 182 of file Tensor.cpp.

Referenced by BOOST_AUTO_TEST_CASE(), armnn::DepthToSpace(), armnn::Fill(), armnnUtils::FindMinMax(), and armnn::TransposeConvolution2dImpl().

 {
     CheckUnspecifiedNumDimensions();
 
     if (m_NumDimensions == 0)
     {
         return 0;
     }
 
     unsigned int count = 1;
     bool atLeastOneDimensionSpecified = false;
     for (unsigned int i = 0; i < m_NumDimensions; ++i)
     {
         if (m_DimensionsSpecificity[i])
         {
             atLeastOneDimensionSpecified = true;
             count *= m_Dimensions[i];
         }
     }
 
     if (atLeastOneDimensionSpecified)
     {
         return count;
     } 
     else 
     {
         return 0;
     }
 }

◆ IsAtLeastOneDimensionSpecified()

bool IsAtLeastOneDimensionSpecified ( ) const

Checks if there is at least one dimension specified.

OR of all array elements.

Returns: - True at least one dimension sizes is specified. False when all dimension sizes are not specified.

Definition at line 258 of file Tensor.cpp.

References CHECK_LOCATION, armnn::MaxNumOfTensorDimensions, armnn::NotSpecified, armnn::Scalar, and armnn::Specified.

Referenced by BOOST_AUTO_TEST_CASE(), and TensorShape::GetDimensionality().

 {
     CheckUnspecifiedNumDimensions();
 
     bool isAtLeastOneDimensionSpecified = false;
     for (unsigned int i = 0; i < m_NumDimensions; ++i)
     {
         if (m_DimensionsSpecificity[i])
         {
             isAtLeastOneDimensionSpecified = true;
             break;
         }
     }
     return isAtLeastOneDimensionSpecified;
 }

◆ operator!=()

bool operator!= ( const TensorShape & other ) const

Inequality comparison operator.

Parameters

other - TensorShape to compare with.

Definition at line 170 of file Tensor.cpp.

 {
     return !(*this == other);
 }

◆ operator=()

TensorShape & operator= ( const TensorShape & other )

Assignation function.

Parameters

other - TensorShape to copy from.

Definition at line 125 of file Tensor.cpp.

 {
     m_NumDimensions = other.m_NumDimensions;
     m_Dimensionality = other.m_Dimensionality;
     std::copy(other.m_Dimensions.cbegin(), other.m_Dimensions.cbegin() + other.m_NumDimensions, m_Dimensions.begin());
     std::copy(other.m_DimensionsSpecificity.cbegin(), other.m_DimensionsSpecificity.cbegin() + other.m_NumDimensions,
               m_DimensionsSpecificity.begin());
     return *this;
 }

◆ operator==()

bool operator== ( const TensorShape & other ) const

Equality comparison operator.

Parameters

other - TensorShape to compare with.

Definition at line 161 of file Tensor.cpp.

Referenced by TensorInfo::HasMultipleQuantizationScales().

 {
     return ((m_NumDimensions == other.m_NumDimensions) &&
             (m_Dimensionality == other.m_Dimensionality) &&
              std::equal(m_Dimensions.cbegin(), m_Dimensions.cbegin() + m_NumDimensions, other.m_Dimensions.cbegin()) &&
              std::equal(m_DimensionsSpecificity.cbegin(), m_DimensionsSpecificity.cbegin() + m_NumDimensions,
                         other.m_DimensionsSpecificity.cbegin()));
 }

◆ operator[]() [1/2]

unsigned int operator[] ( unsigned int i ) const

Read only operator.

Parameters

i	- Dimension index.

Definition at line 136 of file Tensor.cpp.

 {
     CheckUnspecifiedNumDimensions();
     CheckDimensionIndex(i);
     CheckDimensionSpecified(i);
 
     return m_Dimensions.at(i);
 }

◆ operator[]() [2/2]

unsigned int & operator[] ( unsigned int i )

Read and write operator.

Parameters

i	- Dimension index.

Definition at line 146 of file Tensor.cpp.

References CHECK_LOCATION, and armnn::Scalar.

 {
     if (Dimensionality::Scalar == m_Dimensionality)
     {
         std::stringstream errorMessage;
         errorMessage << "TensorShape with Dimensionality::Scalar must be const to use operator[]";
         throw InvalidArgumentException(errorMessage.str(), CHECK_LOCATION());
     }
     CheckUnspecifiedNumDimensions();
     CheckDimensionIndex(i);
     CheckDimensionSpecified(i);
 
     return m_Dimensions.at(i);
 }

◆ SetDimensionSize()

void SetDimensionSize	(	unsigned int	i,
		unsigned int	dimensionSize
	)

Sets the size of the indicated dimension and Specificity for that dimension is set to true.

Parameters

i	- Dimension index.
dimensionSize	- size of one dimension.

Definition at line 233 of file Tensor.cpp.

Referenced by BOOST_AUTO_TEST_CASE(), and TensorShape::GetDimensionality().

 {
     CheckScalar();
     CheckDimensionIndex(i);
 
     m_Dimensions[i] = dimensionSize;
     m_DimensionsSpecificity[i] = true;
 }

◆ SetNumDimensions()

void SetNumDimensions	(	unsigned int	numDimensions,
		bool	initDimensionsSpecificity = `false`
	)

Sets the tensor rank and therefore the Dimensionality is set to Specified if it was not.

Parameters

numDimensions	- Tensor rank.
initDimensionsSpecificity	(optional) - value to initialize the specificity of each dimension size.

Definition at line 220 of file Tensor.cpp.

References armnn::Specified.

Referenced by BOOST_AUTO_TEST_CASE(), and TensorShape::GetDimensionality().

 {
     CheckScalar();
     CheckSpecifiedNumDimensions();
     CheckValidNumDimensions(numDimensions);
 
     m_NumDimensions = numDimensions;
     m_Dimensionality = Dimensionality::Specified;
     std::fill(m_Dimensions.begin(), m_Dimensions.begin() + m_NumDimensions, 0);
     std::fill(m_DimensionsSpecificity.begin(), m_DimensionsSpecificity.begin() + m_NumDimensions,
               initDimensionsSpecificity);
 }

The documentation for this class was generated from the following files:

include/armnn/Tensor.hpp
src/armnn/Tensor.cpp

Public Member Functions

Detailed Description

Constructor & Destructor Documentation

◆ TensorShape() [1/8]

◆ TensorShape() [2/8]

◆ TensorShape() [3/8]

◆ TensorShape() [4/8]

◆ TensorShape() [5/8]

◆ TensorShape() [6/8]

◆ TensorShape() [7/8]

◆ TensorShape() [8/8]

Member Function Documentation

◆ AreAllDimensionsSpecified()

◆ GetDimensionality()

◆ GetDimensionSpecificity()

◆ GetNumDimensions()

◆ GetNumElements()

◆ IsAtLeastOneDimensionSpecified()

◆ operator!=()

◆ operator=()

◆ operator==()

◆ operator[]() [1/2]

◆ operator[]() [2/2]

◆ SetDimensionSize()

◆ SetNumDimensions()