TensorInfo Class Reference

#include <Tensor.hpp>

Public Member Functions
	TensorInfo ()
	Empty (invalid) constructor. More...
	TensorInfo (const TensorShape &shape, DataType dataType, float quantizationScale=0.0f, int32_t quantizationOffset=0)
	TensorInfo (unsigned int numDimensions, const unsigned int *dimensionSizes, DataType dataType, float quantizationScale=0.0f, int32_t quantizationOffset=0)
	TensorInfo (const TensorShape &shape, DataType dataType, const std::vector< float > &quantizationScales, unsigned int quantizationDim)
	TensorInfo (unsigned int numDimensions, const unsigned int *dimensionSizes, DataType dataType, const std::vector< float > &quantizationScales, unsigned int quantizationDim)
	TensorInfo (const TensorInfo &other)
TensorInfo &	operator= (const TensorInfo &other)
bool	operator== (const TensorInfo &other) const
bool	operator!= (const TensorInfo &other) const
const TensorShape &	GetShape () const
TensorShape &	GetShape ()
void	SetShape (const TensorShape &newShape)
unsigned int	GetNumDimensions () const
unsigned int	GetNumElements () const
DataType	GetDataType () const
void	SetDataType (DataType type)
bool	HasMultipleQuantizationScales () const
bool	HasPerAxisQuantization () const
std::vector< float >	GetQuantizationScales () const
void	SetQuantizationScales (const std::vector< float > &scales)
float	GetQuantizationScale () const
void	SetQuantizationScale (float scale)
int32_t	GetQuantizationOffset () const
void	SetQuantizationOffset (int32_t offset)
Optional< unsigned int >	GetQuantizationDim () const
void	SetQuantizationDim (const Optional< unsigned int > &quantizationDim)
bool	IsQuantized () const
bool	IsTypeSpaceMatch (const TensorInfo &other) const
	Check that the types are the same and, if quantize, that the quantization parameters are the same. More...
unsigned int	GetNumBytes () const

Detailed Description

Definition at line 53 of file Tensor.hpp.

Constructor & Destructor Documentation

TensorInfo() [1/6]

TensorInfo

(

)

Empty (invalid) constructor.

Definition at line 137 of file Tensor.cpp.

: m_DataType(DataType::Float32)
{
}

TensorInfo() [2/6]

TensorInfo	(	const TensorShape &	shape,
		DataType	dataType,
		float	quantizationScale = 0.0f,
		int32_t	quantizationOffset = 0
	)

Definition at line 142 of file Tensor.cpp.

References TensorInfo::SetQuantizationOffset(), and TensorInfo::SetQuantizationScale().

    : m_Shape(shape)
    , m_DataType(dataType)
{
    SetQuantizationScale(quantizationScale);
    SetQuantizationOffset(quantizationOffset);
}

TensorInfo() [3/6]

TensorInfo	(	unsigned int	numDimensions,
		const unsigned int *	dimensionSizes,
		DataType	dataType,
		float	quantizationScale = 0.0f,
		int32_t	quantizationOffset = 0
	)

Definition at line 153 of file Tensor.cpp.

References TensorInfo::SetQuantizationOffset(), and TensorInfo::SetQuantizationScale().

    : m_Shape(numDimensions, dimensionSizes)
    , m_DataType(dataType)
{
    SetQuantizationScale(quantizationScale);
    SetQuantizationOffset(quantizationOffset);
}

TensorInfo() [4/6]

TensorInfo	(	const TensorShape &	shape,
		DataType	dataType,
		const std::vector< float > &	quantizationScales,
		unsigned int	quantizationDim
	)

Definition at line 165 of file Tensor.cpp.

References TensorInfo::SetQuantizationDim(), and TensorInfo::SetQuantizationScales().

    : m_Shape(shape)
    , m_DataType(dataType)
{
    SetQuantizationScales(quantizationScales);
    SetQuantizationDim(MakeOptional<unsigned int>(quantizationDim));
}

TensorInfo() [5/6]

TensorInfo	(	unsigned int	numDimensions,
		const unsigned int *	dimensionSizes,
		DataType	dataType,
		const std::vector< float > &	quantizationScales,
		unsigned int	quantizationDim
	)

Definition at line 176 of file Tensor.cpp.

References TensorInfo::SetQuantizationDim(), and TensorInfo::SetQuantizationScales().

    : m_Shape(numDimensions, dimensionSizes)
    , m_DataType(dataType)
{
    SetQuantizationScales(quantizationScales);
    SetQuantizationDim(MakeOptional<unsigned int>(quantizationDim));
}

TensorInfo() [6/6]

TensorInfo

(

const TensorInfo &

other

)

Definition at line 188 of file Tensor.cpp.

: m_Shape(other.m_Shape)
, m_DataType(other.m_DataType)
, m_Quantization(other.m_Quantization)
{}

Member Function Documentation

GetDataType()

DataType GetDataType ( ) const

inline

Definition at line 95 of file Tensor.hpp.

Referenced by BiasAndWeightsTypesCompatible::BiasAndWeightsTypesCompatible(), BiasAndWeightsTypesMatch::BiasAndWeightsTypesMatch(), armnn::BOOST_AUTO_TEST_CASE(), BOOST_AUTO_TEST_CASE(), armnnUtils::CalculateReducedOutputTensoInfo(), armnnUtils::CalculateStridedSliceOutputTensorInfo(), armnn::CheckScaleSetOnQuantizedType(), armnn::CheckTensorDataTypesEqual(), armnn::ClConvertFp16ToFp32WorkloadValidate(), armnn::ClConvertFp32ToFp16WorkloadValidate(), armnn::ConvertBf16ToFp32Weight(), armnn::optimizations::ConvertWeight(), armnn::CreateQuantizedConst(), RefDepthToSpaceWorkload::Execute(), RefStridedSliceWorkload::Execute(), RefSliceWorkload::Execute(), RefLstmWorkload::Execute(), BFloat16ToFloat32::Func(), Float16ToFloat32::Func(), Float32ToBFloat16::Func(), Float32ToFloat16::Func(), armnn::GetBiasDataType(), TfLiteParser::GetBuffer(), Layer::GetDataType(), armnnUtils::GetPerAxisParams(), armnn::GetUnpaddedTensorStrides(), armnn::InitializeArmComputeTensorData(), armnn::InsertConvertBf16ToFp32LayersBefore(), armnn::InsertConvertFp16ToFp32LayersBefore(), armnn::InsertConvertFp32ToBf16LayersAfter(), armnn::InsertConvertFp32ToBf16LayersBefore(), armnn::InsertConvertFp32ToFp16LayersAfter(), SampleDynamicLayerSupport::IsAdditionSupported(), RefLayerSupport::IsConvertFp16ToFp32Supported(), RefLayerSupport::IsConvertFp32ToFp16Supported(), RefLayerSupport::IsConvolution2dSupported(), RefLayerSupport::IsDepthwiseConvolutionSupported(), NeonLayerSupport::IsFloorSupported(), armnn::IsFloorSupported(), RefLayerSupport::IsFullyConnectedSupported(), ClLayerSupport::IsQLstmSupported(), NeonLayerSupport::IsQLstmSupported(), ClLayerSupport::IsSplitterSupported(), NeonLayerSupport::IsSplitterSupported(), RefLayerSupport::IsTransposeConvolution2dSupported(), main(), armnn::MakeDecoder(), armnn::MakeEncoder(), armnnTfParser::OutputShapeOfExpandDims(), TfLiteParser::OutputShapeOfReshape(), armnnTfParser::OutputShapeOfSqueeze(), armnn::PermuteTensor(), ConvertFp32NetworkToFp16Impl::Run(), TypeAnyOf::TypeAnyOf(), TypeIs::TypeIs(), ArgMinMaxQueueDescriptor::Validate(), FullyConnectedQueueDescriptor::Validate(), Convolution2dQueueDescriptor::Validate(), DepthwiseConvolution2dQueueDescriptor::Validate(), QuantizeQueueDescriptor::Validate(), EqualQueueDescriptor::Validate(), ConvertBf16ToFp32QueueDescriptor::Validate(), ConvertFp32ToBf16QueueDescriptor::Validate(), ConvertFp16ToFp32QueueDescriptor::Validate(), ConvertFp32ToFp16QueueDescriptor::Validate(), GreaterQueueDescriptor::Validate(), GatherQueueDescriptor::Validate(), DequantizeQueueDescriptor::Validate(), TransposeConvolution2dQueueDescriptor::Validate(), ComparisonQueueDescriptor::Validate(), armnn::VerifyTensorInfoDataType(), QuantizerVisitor::VisitInputLayer(), QuantizerVisitor::VisitOutputLayer(), and SerializerVisitor::VisitQuantizedLstmLayer().

{ return m_DataType; }

GetNumBytes()

unsigned int GetNumBytes

(

)

const

Definition at line 214 of file Tensor.cpp.

References armnn::GetDataTypeSize(), and TensorInfo::GetNumElements().

Referenced by RefTensorHandle::Allocate(), SampleTensorHandle::Allocate(), ScopedCpuTensorHandle::Allocate(), ClDepthwiseConvolutionWorkload::ClDepthwiseConvolutionWorkload(), armnnTfLiteParser::ComputeWrappedIndex(), ConstTensor::ConstTensor(), TfLiteParser::CreateNetworkFromBinary(), RefReshapeWorkload::Execute(), RefTensorHandle::Manage(), SampleTensorHandle::Manage(), RefTensorHandle::Map(), NeonDepthwiseConvolutionWorkload::NeonDepthwiseConvolutionWorkload(), OnnxParser::OnnxParser(), TfLiteParser::OutputShapeOfReshape(), TfLiteParser::OutputShapeOfSqueeze(), armnn::PermuteTensor(), RefConstantWorkload::PostAllocationConfigure(), armnn::ReorderWeightChannelsForAcl(), and ScopedCpuTensorHandle::ScopedCpuTensorHandle().

{
    return GetDataTypeSize(m_DataType) * GetNumElements();
}

GetNumDimensions()

unsigned int GetNumDimensions ( ) const

inline

Definition at line 92 of file Tensor.hpp.

Referenced by armnn::ArgMinMax(), BOOST_AUTO_TEST_CASE(), BOOST_FIXTURE_TEST_CASE(), armnn::boost_test_print_type(), armnnTfParser::CalculatePaddedOutputTensorInfo(), armnnUtils::CalculateReducedOutputTensoInfo(), armnnUtils::CalculateStridedSliceOutputTensorInfo(), armnnTfParser::CheckPaddingTensor(), armnn::ComputeSoftmaxAclAxis(), armnnTfLiteParser::ComputeWrappedIndex(), armnn::Concatenate(), TfLiteParser::CreateNetworkFromBinary(), armnn::Debug(), armnn::Gather(), armnn::GetBiasDataType(), GetTensorShapeAsArray(), RefLayerSupport::IsBatchToSpaceNdSupported(), RefLayerSupport::IsMeanSupported(), armnn::LogSoftmax(), MakeTensor(), armnn::Mean(), armnnTfParser::OutputShapeOfExpandDims(), TfLiteParser::OutputShapeOfReshape(), TfLiteParser::OutputShapeOfSqueeze(), armnnTfParser::OutputShapeOfSqueeze(), CaffeParserBase::ParseConcatLayer(), CaffeParserBase::ParseInnerProductLayer(), RefFullyConnectedWorkload::PostAllocationConfigure(), armnnUtils::ProcessConcatInputTensorInfo(), PermuteAndBatchToSpaceAsDepthToSpaceImpl< PermuteType >::Run(), ParserFlatbuffersFixture::RunTest(), armnn::Softmax(), armnn::Split(), armnn::Splitter(), armnn::Stack(), armnnCaffeParser::TensorDescToBlobShape(), TensorNumDimensionsAreCorrect::TensorNumDimensionsAreCorrect(), TfLiteParser::TfLiteParser(), TfParser::TfParser(), FullyConnectedQueueDescriptor::Validate(), MeanQueueDescriptor::Validate(), PadQueueDescriptor::Validate(), InstanceNormalizationQueueDescriptor::Validate(), L2NormalizationQueueDescriptor::Validate(), StridedSliceQueueDescriptor::Validate(), GatherQueueDescriptor::Validate(), SliceQueueDescriptor::Validate(), MeanLayer::ValidateTensorShapesFromInputs(), and GatherLayer::ValidateTensorShapesFromInputs().

{ return m_Shape.GetNumDimensions(); }

GetNumElements()

unsigned int GetNumElements ( ) const

inline

Definition at line 93 of file Tensor.hpp.

Referenced by armnn::Activation(), BOOST_FIXTURE_TEST_CASE(), armnnTfLiteParser::ComputeWrappedIndex(), Concat2dTestImpl(), Concat3dTestImpl(), Concat4dTestImpl(), armnn::ConvertBf16ToFp32Weight(), armnnTfParser::ConvertTfTensorDataType(), armnn::optimizations::ConvertWeight(), TfLiteParser::CreateNetworkFromBinary(), armnn::CreateQuantizedConst(), armnn::Debug(), armnn::Dequantize(), armnn::DetectionPostProcess(), RefConvertBf16ToFp32Workload::Execute(), RefConvertFp16ToFp32Workload::Execute(), RefConvertFp32ToBf16Workload::Execute(), RefFakeQuantizationFloat32Workload::Execute(), RefFloorWorkload::Execute(), SampleDynamicAdditionWorkload::Execute(), RefConvertFp32ToFp16Workload::Execute(), RefStackWorkload::Execute(), RefDebugWorkload< DataType >::Execute(), BFloat16ToFloat32::Func(), Float16ToFloat32::Func(), Float32ToBFloat16::Func(), Float32ToFloat16::Func(), armnn::Gather(), armnn::GetBiasDataType(), TensorInfo::GetNumBytes(), MakeRandomTensor(), MakeTensor(), OnnxParser::OnnxParser(), Deserializer::OutputShapeOfReshape(), TfLiteParser::OutputShapeOfReshape(), TfLiteParser::OutputShapeOfSqueeze(), armnn::Pad(), PermuteOutputForConcat(), PermuteTensorData(), armnn::Quantize(), QuantizeData(), ShapesAreSameTotalSize::ShapesAreSameTotalSize(), armnn::Splitter(), TfParser::TfParser(), armnnDeserializer::ToConstTensor(), and SpaceToBatchNdQueueDescriptor::Validate().

{ return m_Shape.GetNumElements(); }

GetQuantizationDim()

Optional< unsigned int > GetQuantizationDim

(

)

const

Definition at line 281 of file Tensor.cpp.

Referenced by armnn::GetBiasDataType(), armnnUtils::GetPerAxisParams(), and SerializerVisitor::VisitQuantizedLstmLayer().

{
    return m_Quantization.m_QuantizationDim;
}

GetQuantizationOffset()

int32_t GetQuantizationOffset

(

)

const

Definition at line 265 of file Tensor.cpp.

Referenced by BOOST_AUTO_TEST_CASE(), armnn::BOOST_AUTO_TEST_CASE(), armnn::CheckScaleSetOnQuantizedType(), DepthwiseConvolution2dAsymmetricTestImpl(), DepthwiseConvolution2dDepthMul1TestImpl(), DepthwiseConvolution2dTestImpl(), armnn::Dequantize(), armnn::GetBiasDataType(), TensorInfo::IsTypeSpaceMatch(), armnn::MakeDecoder(), armnn::MakeEncoder(), MakeRandomTensor(), TfLiteParser::OutputShapeOfReshape(), QuantizationParametersAreEqual::QuantizationParametersAreEqual(), armnn::Quantize(), QuantizeData(), SimpleConvolution2dTestImpl(), and SerializerVisitor::VisitQuantizedLstmLayer().

{
    if (!m_Quantization.m_Offset.has_value())
    {
        // NOTE: old default for backward compatibility
        return 0;
    }

    return m_Quantization.m_Offset.value();
}

GetQuantizationScale()

float GetQuantizationScale

(

)

const

Definition at line 248 of file Tensor.cpp.

References ARMNN_ASSERT, and TensorInfo::HasMultipleQuantizationScales().

Referenced by BOOST_AUTO_TEST_CASE(), armnn::BOOST_AUTO_TEST_CASE(), armnn::CheckScaleSetOnQuantizedType(), DepthwiseConvolution2dAsymmetricTestImpl(), DepthwiseConvolution2dDepthMul1TestImpl(), DepthwiseConvolution2dTestImpl(), armnn::Dequantize(), armnn::GetBiasDataType(), TensorInfo::IsTypeSpaceMatch(), armnn::MakeDecoder(), armnn::MakeEncoder(), MakeRandomTensor(), TfLiteParser::OutputShapeOfReshape(), QuantizationParametersAreEqual::QuantizationParametersAreEqual(), armnn::Quantize(), QuantizeData(), QuantizerVisitor::QuantizerVisitor(), SimpleConvolution2dTestImpl(), and SerializerVisitor::VisitQuantizedLstmLayer().

{
    if (m_Quantization.m_Scales.empty())
    {
        // NOTE: old default for backward compatibility
        return 1.0f;
    }

    ARMNN_ASSERT(!HasMultipleQuantizationScales());
    return m_Quantization.m_Scales[0];
}

GetQuantizationScales()

std::vector< float > GetQuantizationScales

(

)

const

Definition at line 238 of file Tensor.cpp.

Referenced by armnn::GetBiasDataType(), armnnUtils::GetPerAxisParams(), and SerializerVisitor::VisitQuantizedLstmLayer().

{
    return m_Quantization.m_Scales;
}

GetShape() [1/2]

const TensorShape& GetShape ( ) const

inline

Definition at line 88 of file Tensor.hpp.

Referenced by armnn::ArgMinMax(), armnn::BatchNormImpl(), armnn::BatchToSpaceNd(), BOOST_AUTO_TEST_CASE(), armnn::BOOST_AUTO_TEST_CASE(), BOOST_FIXTURE_TEST_CASE(), armnn::boost_test_print_type(), armnnTfParser::CalculatePaddedOutputTensorInfo(), armnnUtils::CalculateReducedOutputTensoInfo(), armnnUtils::CalculateStridedSliceOutputTensorInfo(), armnnTfLiteParser::ComputeWrappedIndex(), armnn::Concatenate(), Concatenate(), armnn::ConvertBf16ToFp32Weight(), armnnTfParser::ConvertTfTensorDataType(), armnn::optimizations::ConvertWeight(), armnn::ConvertWeightTensorFromArmnnToAcl(), armnn::CreateAclNormalizationLayerInfoForL2Normalization(), TfLiteParser::CreateNetworkFromBinary(), OnnxParser::CreateNetworkFromString(), armnn::CreateQuantizedConst(), ConcatLayer::CreateWorkload(), SplitterLayer::CreateWorkload(), armnn::Debug(), armnn::DepthToSpace(), DepthwiseConvolution2dAsymmetricTestImpl(), DepthwiseConvolution2dDepthMul1TestImpl(), DepthwiseConvolution2dTestImpl(), armnn::DetectionPostProcess(), RefLstmWorkload::Execute(), RefQLstmWorkload::Execute(), RefComparisonWorkload::Execute(), RefElementwiseUnaryWorkload::Execute(), RefElementwiseWorkload< Functor, ParentDescriptor, DebugString >::Execute(), BFloat16ToFloat32::Func(), Float16ToFloat32::Func(), Float32ToBFloat16::Func(), Float32ToFloat16::Func(), armnn::Gather(), GetBias(), armnn::GetBiasDataType(), armnnUtils::GetPerAxisParams(), Deserializer::GetQLstmDescriptor(), RefTensorHandle::GetShape(), SampleTensorHandle::GetShape(), ConstCpuTensorHandle::GetShape(), GetTensorShapeAsArray(), armnn::GetUnpaddedTensorStrides(), armnn::InstanceNorm(), armnn::IsFloorSupported(), ClLayerSupport::IsResizeBilinearSupported(), NeonLayerSupport::IsResizeBilinearSupported(), ClLayerSupport::IsSplitterSupported(), NeonLayerSupport::IsSplitterSupported(), armnn::LogSoftmax(), main(), MakeTensor(), armnn::Mean(), armnnTfParser::OutputShapeOfExpandDims(), Deserializer::OutputShapeOfReshape(), TfLiteParser::OutputShapeOfReshape(), TfLiteParser::OutputShapeOfSqueeze(), armnnTfParser::OutputShapeOfSqueeze(), armnn::Pad(), CaffeParserBase::ParseBatchNormLayer(), CaffeParserBase::ParseConcatLayer(), CaffeParserBase::ParseInnerProductLayer(), CaffeParserBase::ParsePoolingLayer(), CaffeParserBase::ParseScaleLayer(), armnnUtils::Permuted(), PermuteInputsForConcat(), armnn::PermuteTensor(), PermuteTensorNchwToNhwc(), PermuteTensorNhwcToNchw(), armnn::Pooling2d(), RefDepthwiseConvolution2dWorkload::PostAllocationConfigure(), RefConvolution2dWorkload::PostAllocationConfigure(), RefFullyConnectedWorkload::PostAllocationConfigure(), RefTransposeConvolution2dWorkload::PostAllocationConfigure(), armnn::PreluImpl(), armnnUtils::ProcessConcatInputTensorInfo(), RefConvolution2dWorkload::RefConvolution2dWorkload(), RefDepthwiseConvolution2dWorkload::RefDepthwiseConvolution2dWorkload(), RefFullyConnectedWorkload::RefFullyConnectedWorkload(), armnn::ReshapeWeightsForAcl(), armnn::Resize(), PermuteAsReshapeImpl::Run(), TransposeAsReshapeImpl::Run(), OptimizeConsecutiveReshapesImpl::Run(), DepthwiseConvolution2dLayer::SerializeLayerParameters(), Convolution2dLayer::SerializeLayerParameters(), Graph::SerializeToDot(), ShapesAreBroadcastCompatible::ShapesAreBroadcastCompatible(), ShapesAreSameRank::ShapesAreSameRank(), SimpleConvolution2dTestImpl(), armnn::Slice(), armnn::Softmax(), armnn::SpaceToBatchNd(), armnn::SpaceToDepth(), armnn::Split(), armnn::Splitter(), armnn::Stack(), armnn::StridedSlice(), armnnCaffeParser::TensorDescToBlobShape(), TfLiteParser::TfLiteParser(), TfParser::TfParser(), TransposeConvolution2dEndToEnd(), armnnUtils::TransposeTensorShape(), ArgMinMaxQueueDescriptor::Validate(), PermuteQueueDescriptor::Validate(), DepthwiseConvolution2dQueueDescriptor::Validate(), DetectionPostProcessQueueDescriptor::Validate(), ResizeBilinearQueueDescriptor::Validate(), ResizeQueueDescriptor::Validate(), SpaceToBatchNdQueueDescriptor::Validate(), SpaceToDepthQueueDescriptor::Validate(), TransposeQueueDescriptor::Validate(), SliceQueueDescriptor::Validate(), DepthToSpaceQueueDescriptor::Validate(), ElementwiseBaseLayer::ValidateTensorShapesFromInputs(), QuantizeLayer::ValidateTensorShapesFromInputs(), ActivationLayer::ValidateTensorShapesFromInputs(), ConvertFp32ToFp16Layer::ValidateTensorShapesFromInputs(), NormalizationLayer::ValidateTensorShapesFromInputs(), DebugLayer::ValidateTensorShapesFromInputs(), SliceLayer::ValidateTensorShapesFromInputs(), MemImportLayer::ValidateTensorShapesFromInputs(), FloorLayer::ValidateTensorShapesFromInputs(), BatchToSpaceNdLayer::ValidateTensorShapesFromInputs(), AbsLayer::ValidateTensorShapesFromInputs(), DequantizeLayer::ValidateTensorShapesFromInputs(), InstanceNormalizationLayer::ValidateTensorShapesFromInputs(), Pooling2dLayer::ValidateTensorShapesFromInputs(), SoftmaxLayer::ValidateTensorShapesFromInputs(), MemCopyLayer::ValidateTensorShapesFromInputs(), ConvertBf16ToFp32Layer::ValidateTensorShapesFromInputs(), SwitchLayer::ValidateTensorShapesFromInputs(), ConvertFp16ToFp32Layer::ValidateTensorShapesFromInputs(), FakeQuantizationLayer::ValidateTensorShapesFromInputs(), ResizeLayer::ValidateTensorShapesFromInputs(), ConvertFp32ToBf16Layer::ValidateTensorShapesFromInputs(), MergeLayer::ValidateTensorShapesFromInputs(), StackLayer::ValidateTensorShapesFromInputs(), RsqrtLayer::ValidateTensorShapesFromInputs(), L2NormalizationLayer::ValidateTensorShapesFromInputs(), GatherLayer::ValidateTensorShapesFromInputs(), TransposeLayer::ValidateTensorShapesFromInputs(), MeanLayer::ValidateTensorShapesFromInputs(), LogSoftmaxLayer::ValidateTensorShapesFromInputs(), PermuteLayer::ValidateTensorShapesFromInputs(), ConstantLayer::ValidateTensorShapesFromInputs(), ReshapeLayer::ValidateTensorShapesFromInputs(), DetectionPostProcessLayer::ValidateTensorShapesFromInputs(), StridedSliceLayer::ValidateTensorShapesFromInputs(), ArgMinMaxLayer::ValidateTensorShapesFromInputs(), ElementwiseUnaryLayer::ValidateTensorShapesFromInputs(), DepthwiseConvolution2dLayer::ValidateTensorShapesFromInputs(), TransposeConvolution2dLayer::ValidateTensorShapesFromInputs(), FullyConnectedLayer::ValidateTensorShapesFromInputs(), SpaceToDepthLayer::ValidateTensorShapesFromInputs(), PreluLayer::ValidateTensorShapesFromInputs(), SpaceToBatchNdLayer::ValidateTensorShapesFromInputs(), ComparisonLayer::ValidateTensorShapesFromInputs(), DepthToSpaceLayer::ValidateTensorShapesFromInputs(), Convolution2dLayer::ValidateTensorShapesFromInputs(), SplitterLayer::ValidateTensorShapesFromInputs(), BatchNormalizationLayer::ValidateTensorShapesFromInputs(), QuantizedLstmLayer::ValidateTensorShapesFromInputs(), LstmLayer::ValidateTensorShapesFromInputs(), QLstmLayer::ValidateTensorShapesFromInputs(), armnn::VerifyTensorInfoDataType(), and SerializerVisitor::VisitQuantizedLstmLayer().

{ return m_Shape; }

GetShape() [2/2]

TensorShape& GetShape ( )

inline

Definition at line 89 of file Tensor.hpp.

{ return m_Shape; }

HasMultipleQuantizationScales()

bool HasMultipleQuantizationScales ( ) const

inline

Definition at line 98 of file Tensor.hpp.

References TensorShape::operator==(), SetQuantizationOffset(), and SetQuantizationScale().

Referenced by armnn::GetBiasDataType(), TensorInfo::GetQuantizationScale(), TensorInfo::HasPerAxisQuantization(), and TensorInfo::IsTypeSpaceMatch().

{ return m_Quantization.m_Scales.size() > 1; }

HasPerAxisQuantization()

bool HasPerAxisQuantization

(

)

const

Definition at line 233 of file Tensor.cpp.

References TensorInfo::HasMultipleQuantizationScales().

Referenced by armnn::GetBiasDataType(), armnnUtils::GetPerAxisParams(), armnn::MakeDecoder(), armnn::MakeEncoder(), TypeNotPerAxisQuantized::TypeNotPerAxisQuantized(), and SerializerVisitor::VisitQuantizedLstmLayer().

{
    return HasMultipleQuantizationScales() || m_Quantization.m_QuantizationDim.has_value();
}

IsQuantized()

bool IsQuantized

(

)

const

Definition at line 291 of file Tensor.cpp.

References armnn::IsQuantizedType().

Referenced by armnn::GetBiasDataType(), TensorInfo::IsTypeSpaceMatch(), and TypeNotPerAxisQuantized::TypeNotPerAxisQuantized().

{
    return IsQuantizedType(m_DataType);
}

IsTypeSpaceMatch()

bool IsTypeSpaceMatch

(

const TensorInfo &

other

)

const

Check that the types are the same and, if quantize, that the quantization parameters are the same.

Definition at line 219 of file Tensor.cpp.

References TensorInfo::GetQuantizationOffset(), TensorInfo::GetQuantizationScale(), TensorInfo::HasMultipleQuantizationScales(), and TensorInfo::IsQuantized().

Referenced by ConcatLayer::CreateWorkload(), SplitterLayer::CreateWorkload(), armnn::GetBiasDataType(), ClLayerSupport::IsConcatSupported(), NeonLayerSupport::IsConcatSupported(), ClLayerSupport::IsSplitterSupported(), and NeonLayerSupport::IsSplitterSupported().

{
    bool match = true;

    match &= m_DataType == other.m_DataType;

    if (IsQuantized() && !HasMultipleQuantizationScales())
    {
        match &= GetQuantizationScale() == other.GetQuantizationScale() &&
                 GetQuantizationOffset() == other.GetQuantizationOffset();
    }
    return match;
}

operator!=()

bool operator!=

(

const TensorInfo &

other

)

const

Definition at line 209 of file Tensor.cpp.

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

operator=()

TensorInfo & operator=

(

const TensorInfo &

other

)

Definition at line 194 of file Tensor.cpp.

{
    m_Shape = other.m_Shape;
    m_DataType = other.m_DataType;
    m_Quantization = other.m_Quantization;
    return *this;
}

operator==()

bool operator==

(

const TensorInfo &

other

)

const

Definition at line 202 of file Tensor.cpp.

{
    return ((m_Shape == other.m_Shape) &&
            (m_DataType == other.m_DataType) &&
            (m_Quantization == other.m_Quantization));
}

SetDataType()

void SetDataType ( DataType  type )

inline

Definition at line 96 of file Tensor.hpp.

Referenced by BOOST_AUTO_TEST_CASE(), ConvertToDataType(), armnn::InsertConvertBf16ToFp32LayersBefore(), armnn::InsertConvertFp16ToFp32LayersBefore(), armnn::InsertConvertFp32ToBf16LayersAfter(), armnn::InsertConvertFp32ToBf16LayersBefore(), armnn::InsertConvertFp32ToFp16LayersAfter(), armnnTfParser::OutputShapeOfExpandDims(), and ConvertFp32NetworkToFp16Impl::Run().

{ m_DataType = type; }

SetQuantizationDim()

void SetQuantizationDim

(

const Optional< unsigned int > &

quantizationDim

)

Definition at line 286 of file Tensor.cpp.

Referenced by TensorInfo::TensorInfo().

{
    m_Quantization.m_QuantizationDim = quantizationDim;
}

SetQuantizationOffset()

void SetQuantizationOffset

(

int32_t

offset

)

Definition at line 276 of file Tensor.cpp.

Referenced by AdditionBroadcast1ElementTestImpl(), AdditionBroadcastTestImpl(), ArgMaxChannelTest(), ArgMaxHeightTest(), ArgMaxSimpleTest(), ArgMinChannelTest(), ArgMinSimpleTest(), ArgMinWidthTest(), BOOST_AUTO_TEST_CASE(), armnn::CheckScaleSetOnQuantizedType(), CompareActivationTestImpl(), ConstantLinearActivationTestCommon(), Convolution2d3x3DilationTestCommon(), DepthwiseConvolution2d3x3DilationTestCommon(), DepthwiseConvolution2dAsymmetricTestImpl(), DepthwiseConvolution2dDepthMul1TestImpl(), DepthwiseConvolution2dTestImpl(), DetectionPostProcessFastNmsQuantizedTest(), DetectionPostProcessRegularNmsQuantizedTest(), FullyConnectedLargeTestCommon(), GetSoftmaxProfilerJson(), PermuteValueSet1Test(), PermuteValueSet2Test(), PermuteValueSet3Test(), QuantizeData(), SimpleConvolution2dTestImpl(), SimplePermuteTest(), SimpleTransposeTest(), TensorInfo::TensorInfo(), TransposeValueSet1Test(), TransposeValueSet2Test(), and TransposeValueSet3Test().

{
    m_Quantization.m_Offset = MakeOptional<int32_t>(offset);
}

SetQuantizationScale()

void SetQuantizationScale

(

float

scale

)

Definition at line 260 of file Tensor.cpp.

Referenced by AdditionBroadcast1ElementTestImpl(), AdditionBroadcastTestImpl(), ArgMaxChannelTest(), ArgMaxHeightTest(), ArgMaxSimpleTest(), ArgMinChannelTest(), ArgMinSimpleTest(), ArgMinWidthTest(), BOOST_AUTO_TEST_CASE(), BoundedReLuTestCommon(), armnn::CheckScaleSetOnQuantizedType(), CompareActivationTestImpl(), ConcatUint16Test(), ConcatUint8Test(), ConstantLinearActivationTestCommon(), Convolution1dTestImpl(), Convolution2d3x3DilationTestCommon(), DepthwiseConvolution2d3x3DilationTestCommon(), DepthwiseConvolution2dAsymmetricTestImpl(), DepthwiseConvolution2dDepthMul1TestImpl(), DepthwiseConvolution2dTestImpl(), DetectionPostProcessFastNmsQuantizedTest(), DetectionPostProcessRegularNmsQuantizedTest(), ElementwiseTestHelper(), ElementwiseUnaryTestHelper(), FullyConnectedLargeTestCommon(), FullyConnectedTest(), GetSoftmaxProfilerJson(), PermuteValueSet1Test(), PermuteValueSet2Test(), PermuteValueSet3Test(), PreluTest(), QuantizeData(), Reshape5dTest(), armnn::SetupQuantize(), SimpleActivationTest(), SimpleConvolution2dTestImpl(), SimpleFloorTest(), SimplePermuteTest(), SimpleReshapeTest(), SimpleTransposeTest(), TensorInfo::TensorInfo(), TransposeValueSet1Test(), TransposeValueSet2Test(), and TransposeValueSet3Test().

{
    m_Quantization.m_Scales = { scale };
}

SetQuantizationScales()

void SetQuantizationScales

(

const std::vector< float > &

scales

)

Definition at line 243 of file Tensor.cpp.

Referenced by TensorInfo::TensorInfo().

{
    m_Quantization.m_Scales = scales;
}

SetShape()

void SetShape ( const TensorShape &  newShape )

inline

Definition at line 90 of file Tensor.hpp.

Referenced by BOOST_AUTO_TEST_CASE(), armnnTfParser::CalculatePaddedOutputTensorInfo(), TfLiteParser::CreateNetworkFromBinary(), OnnxParser::CreateNetworkFromString(), armnnTfParser::OutputShapeOfExpandDims(), Deserializer::OutputShapeOfReshape(), TfLiteParser::OutputShapeOfReshape(), TfLiteParser::OutputShapeOfSqueeze(), armnnTfParser::OutputShapeOfSqueeze(), CaffeParserBase::ParseInputLayer(), armnnUtils::Permuted(), PermuteInputsForConcat(), armnn::ReshapeWeightsForAcl(), TfLiteParser::TfLiteParser(), TfParser::TfParser(), and armnnUtils::TransposeTensorShape().

{ m_Shape = newShape; }

---

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

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