TensorShape Class Reference

#include <Tensor.hpp>

Public Member Functions
	TensorShape ()
	Empty (invalid) constructor. More...
	TensorShape (unsigned int numDimensions)
	TensorShape (unsigned int numDimensions, const unsigned int *dimensionSizes)
	TensorShape (std::initializer_list< unsigned int > dimensionSizeList)
	TensorShape (const TensorShape &other)
TensorShape &	operator= (const TensorShape &other)
unsigned int	operator[] (unsigned int i) const
unsigned int &	operator[] (unsigned int i)
bool	operator== (const TensorShape &other) const
bool	operator!= (const TensorShape &other) const
unsigned int	GetNumDimensions () const
unsigned int	GetNumElements () const

Detailed Description

Definition at line 20 of file Tensor.hpp.

Constructor & Destructor Documentation

TensorShape() [1/5]

TensorShape

(

)

Empty (invalid) constructor.

Definition at line 24 of file Tensor.cpp.

 : m_NumDimensions(0)
{
}

TensorShape() [2/5]

TensorShape

(

unsigned int

numDimensions

)

Definition at line 29 of file Tensor.cpp.

References armnn::MaxNumOfTensorDimensions.

 : m_NumDimensions(numDimensions)
{
    if (numDimensions < 1)
    {
        throw InvalidArgumentException("Tensor numDimensions must be greater than 0");
    }

    if (numDimensions > MaxNumOfTensorDimensions)
    {
        throw InvalidArgumentException("Tensor numDimensions must be less than or equal to MaxNumOfTensorDimensions");
    }

    std::fill(m_Dimensions.begin(), m_Dimensions.begin() + m_NumDimensions, 0);
}

TensorShape() [3/5]

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

Definition at line 45 of file Tensor.cpp.

References armnn::MaxNumOfTensorDimensions.

 : m_NumDimensions(numDimensions)
{
    if (numDimensions < 1)
    {
        throw InvalidArgumentException("Tensor numDimensions must be greater than 0");
    }

    if (numDimensions > MaxNumOfTensorDimensions)
    {
        throw InvalidArgumentException("Tensor numDimensions must be less than or equal to MaxNumOfTensorDimensions");
    }

    if (dimensionSizes == nullptr)
    {
        throw InvalidArgumentException("Tensor dimensionSizes must not be NULL");
    }

    std::copy(dimensionSizes, dimensionSizes + numDimensions, m_Dimensions.begin());
}

TensorShape() [4/5]

TensorShape

(

std::initializer_list< unsigned int >

dimensionSizeList

)

Definition at line 66 of file Tensor.cpp.

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

TensorShape() [5/5]

TensorShape

(

const TensorShape &

other

)

Definition at line 71 of file Tensor.cpp.

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

Member Function Documentation

GetNumDimensions()

unsigned int GetNumDimensions ( ) const

inline

Definition at line 43 of file Tensor.hpp.

References TensorShape::GetNumElements().

Referenced by armnn::BatchToSpaceNd(), 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(), PreluLayer::InferOutputShapes(), ComparisonLayer::InferOutputShapes(), StackLayer::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(), SerializerVisitor::VisitQuantizedLstmLayer(), SerializerVisitor::VisitReshapeLayer(), and SerializerVisitor::VisitStackLayer().

{ return m_NumDimensions; }

GetNumElements()

unsigned int GetNumElements

(

)

const

Definition at line 107 of file Tensor.cpp.

References CHECK_LOCATION.

Referenced by armnn::DepthToSpace(), armnnUtils::FindMinMax(), TensorShape::GetNumDimensions(), and armnn::TransposeConvolution2dImpl().

{
    if (m_NumDimensions == 0)
    {
        return 0;
    }

    unsigned int count = 1;
    for (unsigned int i = 0; i < m_NumDimensions; i++)
    {
        count *= m_Dimensions[i];
    }

    return count;
}

operator!=()

bool operator!=

(

const TensorShape &

other

)

const

Definition at line 102 of file Tensor.cpp.

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

operator=()

TensorShape & operator=

(

const TensorShape &

other

)

Definition at line 77 of file Tensor.cpp.

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

operator==()

bool operator==

(

const TensorShape &

other

)

const

Definition at line 96 of file Tensor.cpp.

Referenced by TensorInfo::HasMultipleQuantizationScales().

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

operator[]() [1/2]

unsigned int operator[]

(

unsigned int

i

)

const

Definition at line 84 of file Tensor.cpp.

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

operator[]() [2/2]

unsigned int & operator[]

(

unsigned int

i

)

Definition at line 90 of file Tensor.cpp.

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

---

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

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