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, bool isConstant=false)
	TensorInfo (unsigned int numDimensions, const unsigned int *dimensionSizes, DataType dataType, float quantizationScale=0.0f, int32_t quantizationOffset=0, bool isConstant=false)
	TensorInfo (const TensorShape &shape, DataType dataType, const std::vector< float > &quantizationScales, unsigned int quantizationDim, bool isConstant=false)
	TensorInfo (unsigned int numDimensions, const unsigned int *dimensionSizes, DataType dataType, const std::vector< float > &quantizationScales, unsigned int quantizationDim, bool isConstant=false)
	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	IsConstant () const
void	SetConstant (const bool IsConstant=true)
	Marks the data corresponding to this tensor info as constant. More...
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

Examples:

AsyncExecutionSample.cpp, CustomMemoryAllocatorSample.cpp, DynamicSample.cpp, and SimpleSample.cpp.

Definition at line 152 of file Tensor.hpp.

Constructor & Destructor Documentation

TensorInfo() [1/6]

TensorInfo

(

)

Empty (invalid) constructor.

Definition at line 341 of file Tensor.cpp.

: m_DataType(DataType::Float32), m_IsConstant(false)
{
}

TensorInfo() [2/6]

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

Definition at line 346 of file Tensor.cpp.

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

    : m_Shape(shape)
    , m_DataType(dataType)
    , m_IsConstant(isConstant)
{
    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,
		bool	isConstant = false
	)

Definition at line 359 of file Tensor.cpp.

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

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

TensorInfo() [4/6]

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

Definition at line 371 of file Tensor.cpp.

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

    : m_Shape(shape)
    , m_DataType(dataType)
    , m_IsConstant(isConstant)
{
    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,
		bool	isConstant = false
	)

Definition at line 384 of file Tensor.cpp.

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

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

TensorInfo() [6/6]

TensorInfo

(

const TensorInfo &

other

)

Definition at line 398 of file Tensor.cpp.

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

Member Function Documentation

GetDataType()

DataType GetDataType ( ) const

inline

Definition at line 198 of file Tensor.hpp.

Referenced by armnnTfLiteParser::AsFloatArray(), BiasAndWeightsTypesCompatible::BiasAndWeightsTypesCompatible(), BiasAndWeightsTypesMatch::BiasAndWeightsTypesMatch(), armnnUtils::CalculateReducedOutputTensoInfo(), armnnUtils::CalculateStridedSliceOutputTensorInfo(), MockTensorHandle::CanBeImported(), RefTensorHandle::CanBeImported(), armnn::CheckScaleSetOnQuantizedType(), armnn::ClConvertFp16ToFp32WorkloadValidate(), armnn::ClConvertFp32ToFp16WorkloadValidate(), armnnTfLiteParser::ComputeWrappedIndex(), armnn::ConvertBf16ToFp32Weight(), armnn::optimizations::ConvertWeight(), RefDepthToSpaceWorkload::ExecuteAsync(), RefStridedSliceWorkload::ExecuteAsync(), RefArgMinMaxWorkload::ExecuteAsync(), RefSliceWorkload::ExecuteAsync(), RefShapeWorkload::ExecuteAsync(), BFloat16ToFloat32::Func(), Float16ToFloat32::Func(), Float32ToBFloat16::Func(), Float32ToFloat16::Func(), armnn::GetBiasDataType(), TfLiteParserImpl::GetBuffer(), Layer::GetDataType(), armnnSerializer::GetFlatBufferArgMinMaxFunction(), armnn::optimizations::pad_fold::GetLowestElement(), TfLiteParserImpl::GetOutputTensorIds(), 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::IsConvolution3dSupported(), RefLayerSupport::IsDepthwiseConvolutionSupported(), NeonLayerSupport::IsFloorSupported(), RefLayerSupport::IsFullyConnectedSupported(), ClLayerSupport::IsQLstmSupported(), NeonLayerSupport::IsQLstmSupported(), RefLayerSupport::IsTransposeConvolution2dSupported(), NeonLayerSupport::IsUnidirectionalSequenceLstmSupported(), armnn::LstmImpl(), main(), armnn::MakeDecoder(), armnn::MakeEncoder(), TfLiteParserImpl::OutputShapeOfReshape(), TfLiteParserImpl::OutputShapeOfSqueeze(), armnn::PermuteTensor(), ConvertConstDequantisationLayersToConstLayersImpl::Run(), ConvertFp32NetworkToFp16Impl::Run(), ConvertConstPermuteLayersToConstLayers::Run(), TEST_SUITE(), TypeAnyOf::TypeAnyOf(), TypeIs::TypeIs(), ArgMinMaxQueueDescriptor::Validate(), FullyConnectedQueueDescriptor::Validate(), Convolution2dQueueDescriptor::Validate(), Convolution3dQueueDescriptor::Validate(), DepthwiseConvolution2dQueueDescriptor::Validate(), QuantizeQueueDescriptor::Validate(), EqualQueueDescriptor::Validate(), ConvertBf16ToFp32QueueDescriptor::Validate(), ConvertFp32ToBf16QueueDescriptor::Validate(), ConvertFp16ToFp32QueueDescriptor::Validate(), ConvertFp32ToFp16QueueDescriptor::Validate(), GreaterQueueDescriptor::Validate(), GatherNdQueueDescriptor::Validate(), GatherQueueDescriptor::Validate(), TransposeConvolution2dQueueDescriptor::Validate(), ComparisonQueueDescriptor::Validate(), LogicalBinaryQueueDescriptor::Validate(), LayerVerifierBase::VerifyConstTensors(), LayerVerifierBase::VerifyNameAndConnections(), and armnn::VerifyTensorInfoDataType().

{ return m_DataType; }

GetNumBytes()

unsigned int GetNumBytes

(

)

const

Definition at line 427 of file Tensor.cpp.

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

Referenced by MockTensorHandle::Allocate(), RefTensorHandle::Allocate(), SampleTensorHandle::Allocate(), ScopedTensorHandle::Allocate(), armnnTfLiteParser::AsFloatArray(), armnnTfLiteParser::ComputeWrappedIndex(), ConstTensor::ConstTensor(), armnnOnnxParser::CreateConstTensorImpl(), RefReshapeWorkload::ExecuteAsync(), IDeserializer::DeserializerImpl::GetNetworkOutputBindingInfo(), TfLiteParserImpl::GetOutputTensorIds(), SampleTensorHandle::Import(), MockTensorHandle::Manage(), RefTensorHandle::Manage(), SampleTensorHandle::Manage(), MockTensorHandle::Map(), RefTensorHandle::Map(), TfLiteParserImpl::OutputShapeOfReshape(), TfLiteParserImpl::OutputShapeOfSqueeze(), armnn::PermuteTensor(), armnn::ReorderWeightChannelsForAcl(), and ScopedTensorHandle::ScopedTensorHandle().

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

GetNumDimensions()

unsigned int GetNumDimensions ( ) const

inline

Definition at line 195 of file Tensor.hpp.

Referenced by armnn::ArgMinMax(), armnnTfLiteParser::AsFloatArray(), armnn::CalculateGatherNdKeyIndices(), armnnUtils::CalculateReducedOutputTensoInfo(), armnnUtils::CalculateStridedSliceOutputTensorInfo(), armnn::ComputeAclAxis(), armnn::ComputePositiveAxis(), armnn::ComputeReductionTensorShape(), armnn::ComputeSoftmaxAclAxis(), armnnTfLiteParser::ComputeWrappedIndex(), armnn::Concatenate(), armnn::Debug(), RefShapeWorkload::ExecuteAsync(), RefRankWorkload::ExecuteAsync(), armnn::Gather(), armnn::GetBiasDataType(), armnn::GetNumActivations(), RefLayerSupport::IsBatchToSpaceNdSupported(), RefLayerSupport::IsMeanSupported(), armnn::LogSoftmax(), TfLiteParserImpl::OutputShapeOfReshape(), TfLiteParserImpl::OutputShapeOfSqueeze(), armnnUtils::ProcessConcatInputTensorInfo(), armnn::Reduce(), PermuteAndBatchToSpaceAsDepthToSpaceImpl< PermuteType >::Run(), AddBroadcastReshapeLayerImpl::Run(), ParserFlatbuffersFixture::RunTest(), armnn::Softmax(), armnn::Split(), armnn::Splitter(), armnn::Stack(), TensorNumDimensionsAreCorrect::TensorNumDimensionsAreCorrect(), TEST_SUITE(), FullyConnectedQueueDescriptor::Validate(), MeanQueueDescriptor::Validate(), PadQueueDescriptor::Validate(), InstanceNormalizationQueueDescriptor::Validate(), L2NormalizationQueueDescriptor::Validate(), StridedSliceQueueDescriptor::Validate(), GatherNdQueueDescriptor::Validate(), GatherQueueDescriptor::Validate(), SliceQueueDescriptor::Validate(), QueueDescriptor::ValidateTensorNumDimensions(), ReduceLayer::ValidateTensorShapesFromInputs(), FillLayer::ValidateTensorShapesFromInputs(), and MeanLayer::ValidateTensorShapesFromInputs().

{ return m_Shape.GetNumDimensions(); }

GetNumElements()

unsigned int GetNumElements ( ) const

inline

Examples:

CustomMemoryAllocatorSample.cpp.

Definition at line 196 of file Tensor.hpp.

Referenced by armnn::Activation(), armnnTfLiteParser::AsFloatArray(), BoundedReLuUint8UpperAndLowerBoundTest(), CompareActivationTestImpl(), CompareAdditionTest(), CompareBatchNormTest(), CompareConvolution2dTestImpl(), CompareDepthwiseConvolution2dTestImpl(), CompareMultiplicationTest(), armnnTfLiteParser::ComputeWrappedIndex(), Concat2dTestImpl(), Concat3dTestImpl(), Concat4dTestImpl(), ConstantLinearActivationTestCommon(), armnn::ConvertBf16ToFp32Weight(), armnn::optimizations::ConvertWeight(), armnnOnnxParser::CreateConstTensorImpl(), armnn::Debug(), DepthwiseConvolution2dAsymmetricTestImpl(), DepthwiseConvolution2dDepthMul1TestImpl(), DepthwiseConvolution2dTestImpl(), armnn::Dequantize(), armnn::DetectionPostProcess(), SampleDynamicAdditionWorkload::Execute(), RefGatherNdWorkload::ExecuteAsync(), RefConvertBf16ToFp32Workload::ExecuteAsync(), RefConvertFp16ToFp32Workload::ExecuteAsync(), RefConvertFp32ToBf16Workload::ExecuteAsync(), RefFakeQuantizationFloat32Workload::ExecuteAsync(), RefConvertFp32ToFp16Workload::ExecuteAsync(), RefFloorWorkload::ExecuteAsync(), RefCastWorkload::ExecuteAsync(), RefUnidirectionalSequenceLstmWorkload::ExecuteAsync(), RefDebugWorkload< DataType >::ExecuteAsync(), BFloat16ToFloat32::Func(), Float16ToFloat32::Func(), Float32ToBFloat16::Func(), Float32ToFloat16::Func(), armnn::Gather(), armnn::GetBiasDataType(), TensorInfo::GetNumBytes(), L2Normalization2dShapeTest(), LayerTestResult< T, n >::LayerTestResult(), MakeRandomTensor(), armnn::MirrorPad(), MirrorPad2dTestCommon(), MirrorPad3dTestCommon(), MirrorPad4dTestCommon(), IDeserializer::DeserializerImpl::OutputShapeOfReshape(), TfLiteParserImpl::OutputShapeOfReshape(), TfLiteParserImpl::OutputShapeOfSqueeze(), armnn::Pad(), Pad2dTestCommon(), Pad3dTestCommon(), Pad4dTestCommon(), PadQAsymmTestCommon(), PermuteOutputForConcat(), PermuteTensorData(), armnn::Quantize(), QuantizeData(), RankTest(), armnn::Reduce(), ConvertConstDequantisationLayersToConstLayersImpl::Run(), ConvertConstPermuteLayersToConstLayers::Run(), ParserFlatbuffersFixture::RunTest(), ShapesAreSameTotalSize::ShapesAreSameTotalSize(), ShapeTest(), SimpleConvolution2dTestImpl(), SimpleFloorTest(), SimpleFullyConnectedTestImpl(), SimplePermuteTestImpl(), SimpleTransposeTestImpl(), armnn::Splitter(), armnn::Stack(), TEST_SUITE(), armnnDeserializer::ToConstTensor(), TransposeConvolution2dPerAxisQuantTest(), and SpaceToBatchNdQueueDescriptor::Validate().

{ return m_Shape.GetNumElements(); }

GetQuantizationDim()

Optional< unsigned int > GetQuantizationDim

(

)

const

Definition at line 494 of file Tensor.cpp.

Referenced by armnnTfLiteParser::AsFloatArray(), armnn::GetBiasDataType(), armnnSerializer::GetFlatBufferArgMinMaxFunction(), armnnUtils::GetPerAxisParams(), and armnnUtils::Permuted().

{
    return m_Quantization.m_QuantizationDim;
}

GetQuantizationOffset()

int32_t GetQuantizationOffset

(

)

const

Definition at line 478 of file Tensor.cpp.

Referenced by armnnTfLiteParser::AsFloatArray(), armnn::CheckScaleSetOnQuantizedType(), Convolution2d3x3DilationTestCommon(), DepthwiseConvolution2d3x3DilationTestCommon(), DepthwiseConvolution2dAsymmetricTestImpl(), DepthwiseConvolution2dDepthMul1TestImpl(), DepthwiseConvolution2dTestImpl(), armnn::Dequantize(), armnn::GetBiasDataType(), armnnSerializer::GetFlatBufferArgMinMaxFunction(), armnn::optimizations::pad_fold::GetLowestElement(), armnn::optimizations::pad_fold::GetZeroElement(), TensorInfo::IsTypeSpaceMatch(), armnn::MakeDecoder(), armnn::MakeEncoder(), MakeRandomTensor(), TfLiteParserImpl::OutputShapeOfReshape(), QuantizationParametersAreEqual::QuantizationParametersAreEqual(), armnn::Quantize(), QuantizeData(), SimpleConvolution2dTestImpl(), TEST_SUITE(), and LayerVerifierBase::VerifyNameAndConnections().

{
    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 461 of file Tensor.cpp.

References ARMNN_ASSERT, and TensorInfo::HasMultipleQuantizationScales().

Referenced by armnnTfLiteParser::AsFloatArray(), armnn::CheckScaleSetOnQuantizedType(), Convolution2d3x3DilationTestCommon(), DepthwiseConvolution2d3x3DilationTestCommon(), DepthwiseConvolution2dAsymmetricTestImpl(), DepthwiseConvolution2dDepthMul1TestImpl(), DepthwiseConvolution2dTestImpl(), armnn::Dequantize(), armnn::GetBiasDataType(), armnnSerializer::GetFlatBufferArgMinMaxFunction(), armnn::optimizations::pad_fold::GetLowestElement(), TensorInfo::IsTypeSpaceMatch(), armnn::MakeDecoder(), armnn::MakeEncoder(), MakeRandomTensor(), TfLiteParserImpl::OutputShapeOfReshape(), QuantizationParametersAreEqual::QuantizationParametersAreEqual(), armnn::Quantize(), QuantizeData(), SimpleConvolution2dTestImpl(), TEST_SUITE(), and LayerVerifierBase::VerifyNameAndConnections().

{
    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 451 of file Tensor.cpp.

Referenced by armnnTfLiteParser::AsFloatArray(), armnn::GetBiasDataType(), armnnSerializer::GetFlatBufferArgMinMaxFunction(), armnnUtils::GetPerAxisParams(), and LayerVerifierBase::VerifyNameAndConnections().

{
    return m_Quantization.m_Scales;
}

GetShape() [1/2]

const TensorShape& GetShape ( ) const

inline

Definition at line 191 of file Tensor.hpp.

Referenced by armnn::ArgMinMax(), armnnTfLiteParser::AsFloatArray(), armnn::BatchNormImpl(), armnn::BatchToSpaceNd(), armnn::CalculateGatherNdKeyIndices(), armnnUtils::CalculateReducedOutputTensoInfo(), armnnUtils::CalculateStridedSliceOutputTensorInfo(), armnn::ClUnidirectionalSequenceLstmFloatWorkloadValidate(), CompareActivationTestImpl(), CompareAdditionTest(), CompareBatchNormTest(), CompareConvolution2dTestImpl(), CompareDepthwiseConvolution2dTestImpl(), CompareMultiplicationTest(), armnnTfLiteParser::ComputeWrappedIndex(), armnn::Concatenate(), Concatenate(), ConstantLinearActivationTestCommon(), armnn::Convert1HWOTensorInfoToAcl(), armnn::Convert1HWOTensorToAcl(), armnn::Convert1HWOtoMIHW(), armnn::ConvertBf16ToFp32Weight(), armnn::ConvertWeightTensorFromArmnnToAcl(), Convolution2d3x3DilationTestCommon(), armnn::CreateAclNormalizationLayerInfoForL2Normalization(), armnnOnnxParser::CreateConstTensorImpl(), OnnxParserImpl::CreateNetworkFromString(), SplitterLayer::CreateWorkload(), ConcatLayer::CreateWorkload(), armnn::Debug(), armnn::DepthToSpace(), DepthwiseConvolution2d3x3DilationTestCommon(), DepthwiseConvolution2dAsymmetricTestImpl(), DepthwiseConvolution2dDepthMul1TestImpl(), DepthwiseConvolution2dTestImpl(), armnn::DetectionPostProcess(), RefChannelShuffleWorkload::ExecuteAsync(), RefDepthwiseConvolution2dWorkload::ExecuteAsync(), RefLstmWorkload::ExecuteAsync(), RefQLstmWorkload::ExecuteAsync(), RefConvolution3dWorkload::ExecuteAsync(), RefElementwiseUnaryWorkload::ExecuteAsync(), RefLogicalBinaryWorkload::ExecuteAsync(), RefLogicalUnaryWorkload::ExecuteAsync(), RefComparisonWorkload::ExecuteAsync(), RefShapeWorkload::ExecuteAsync(), RefUnidirectionalSequenceLstmWorkload::ExecuteAsync(), RefElementwiseWorkload< Functor, ParentDescriptor, DebugString >::ExecuteAsync(), BFloat16ToFloat32::Func(), Float16ToFloat32::Func(), Float32ToBFloat16::Func(), Float32ToFloat16::Func(), armnn::Gather(), GetBias(), GetBiasData(), armnn::GetBiasDataType(), TfLiteParserImpl::GetBuffer(), armnnSerializer::GetFlatBufferArgMinMaxFunction(), armnn::GetNumActivations(), armnnUtils::GetPerAxisParams(), IDeserializer::DeserializerImpl::GetQLstmDescriptor(), MockTensorHandle::GetShape(), RefTensorHandle::GetShape(), SampleTensorHandle::GetShape(), ConstTensorHandle::GetShape(), armnn::GetUnpaddedTensorStrides(), armnn::InstanceNorm(), ClLayerSupport::IsSplitterSupported(), NeonLayerSupport::IsSplitterSupported(), L2Normalization2dShapeTest(), LayerTestResult< T, n >::LayerTestResult(), armnn::LogSoftmax(), armnn::LstmImpl(), main(), armnn::MirrorPad(), MirrorPad2dTestCommon(), MirrorPad3dTestCommon(), MirrorPad4dTestCommon(), armnn::NeonUnidirectionalSequenceLstmFloatWorkloadValidate(), armnn::NeonUnidirectionalSequenceLstmWorkloadValidate(), IDeserializer::DeserializerImpl::OutputShapeOfReshape(), TfLiteParserImpl::OutputShapeOfReshape(), TfLiteParserImpl::OutputShapeOfSqueeze(), armnn::Pad(), Pad2dTestCommon(), Pad3dTestCommon(), Pad4dTestCommon(), PadQAsymmTestCommon(), armnnUtils::Permuted(), PermuteInputsForConcat(), armnn::PermuteTensor(), PermuteTensorNcdhwToNdhwc(), PermuteTensorNchwToNhwc(), PermuteTensorNdhwcToNcdhw(), PermuteTensorNhwcToNchw(), armnn::Pooling2d(), armnn::Pooling3d(), armnn::PreluImpl(), armnnUtils::ProcessConcatInputTensorInfo(), RankTest(), armnn::Reduce(), armnn::ReshapeWeightsForAcl(), armnn::Resize(), TransposeAsReshapeImpl::Run(), PermuteAsReshapeImpl::Run(), OptimizeConsecutiveReshapesImpl::Run(), ConvertConstPermuteLayersToConstLayers::Run(), AddBroadcastReshapeLayerImpl::Run(), FuseBatchNorm< ConvLayer, ArmnnType, T >::Run(), Convolution3dLayer::SerializeLayerParameters(), DepthwiseConvolution2dLayer::SerializeLayerParameters(), Convolution2dLayer::SerializeLayerParameters(), Graph::SerializeToDot(), ShapesAreBroadcastCompatible::ShapesAreBroadcastCompatible(), ShapesAreSameRank::ShapesAreSameRank(), ShapeTest(), SimpleConvolution2dTestImpl(), SimpleFloorTest(), SimplePermuteTestImpl(), SimpleTransposeTestImpl(), armnn::Slice(), armnn::Softmax(), armnn::SpaceToBatchNd(), armnn::SpaceToDepth(), armnn::Split(), armnn::Splitter(), armnn::Stack(), armnn::StridedSlice(), TEST_SUITE(), TransposeConvolution2dEndToEnd(), TransposeConvolution2dPerAxisQuantTest(), armnnUtils::TransposeTensorShape(), ArgMinMaxQueueDescriptor::Validate(), PermuteQueueDescriptor::Validate(), Pooling3dQueueDescriptor::Validate(), DepthwiseConvolution2dQueueDescriptor::Validate(), DetectionPostProcessQueueDescriptor::Validate(), ResizeQueueDescriptor::Validate(), SpaceToBatchNdQueueDescriptor::Validate(), SpaceToDepthQueueDescriptor::Validate(), TransposeQueueDescriptor::Validate(), SliceQueueDescriptor::Validate(), DepthToSpaceQueueDescriptor::Validate(), QueueDescriptor::ValidateTensorNumDimensions(), ElementwiseBaseLayer::ValidateTensorShapesFromInputs(), RankLayer::ValidateTensorShapesFromInputs(), QuantizeLayer::ValidateTensorShapesFromInputs(), ActivationLayer::ValidateTensorShapesFromInputs(), ConvertFp32ToFp16Layer::ValidateTensorShapesFromInputs(), FillLayer::ValidateTensorShapesFromInputs(), ReduceLayer::ValidateTensorShapesFromInputs(), ResizeLayer::ValidateTensorShapesFromInputs(), RsqrtLayer::ValidateTensorShapesFromInputs(), MergeLayer::ValidateTensorShapesFromInputs(), ConvertFp32ToBf16Layer::ValidateTensorShapesFromInputs(), SliceLayer::ValidateTensorShapesFromInputs(), SwitchLayer::ValidateTensorShapesFromInputs(), SoftmaxLayer::ValidateTensorShapesFromInputs(), ShapeLayer::ValidateTensorShapesFromInputs(), AbsLayer::ValidateTensorShapesFromInputs(), L2NormalizationLayer::ValidateTensorShapesFromInputs(), Pooling2dLayer::ValidateTensorShapesFromInputs(), InstanceNormalizationLayer::ValidateTensorShapesFromInputs(), Pooling3dLayer::ValidateTensorShapesFromInputs(), DequantizeLayer::ValidateTensorShapesFromInputs(), FloorLayer::ValidateTensorShapesFromInputs(), MemImportLayer::ValidateTensorShapesFromInputs(), StackLayer::ValidateTensorShapesFromInputs(), MemCopyLayer::ValidateTensorShapesFromInputs(), ConvertFp16ToFp32Layer::ValidateTensorShapesFromInputs(), NormalizationLayer::ValidateTensorShapesFromInputs(), BatchToSpaceNdLayer::ValidateTensorShapesFromInputs(), FakeQuantizationLayer::ValidateTensorShapesFromInputs(), DebugLayer::ValidateTensorShapesFromInputs(), ConvertBf16ToFp32Layer::ValidateTensorShapesFromInputs(), CastLayer::ValidateTensorShapesFromInputs(), ChannelShuffleLayer::ValidateTensorShapesFromInputs(), TransposeLayer::ValidateTensorShapesFromInputs(), MeanLayer::ValidateTensorShapesFromInputs(), LogSoftmaxLayer::ValidateTensorShapesFromInputs(), PadLayer::ValidateTensorShapesFromInputs(), ReshapeLayer::ValidateTensorShapesFromInputs(), ConstantLayer::ValidateTensorShapesFromInputs(), PermuteLayer::ValidateTensorShapesFromInputs(), Convolution3dLayer::ValidateTensorShapesFromInputs(), GatherNdLayer::ValidateTensorShapesFromInputs(), ArgMinMaxLayer::ValidateTensorShapesFromInputs(), ElementwiseUnaryLayer::ValidateTensorShapesFromInputs(), GatherLayer::ValidateTensorShapesFromInputs(), DetectionPostProcessLayer::ValidateTensorShapesFromInputs(), StridedSliceLayer::ValidateTensorShapesFromInputs(), SpaceToBatchNdLayer::ValidateTensorShapesFromInputs(), ComparisonLayer::ValidateTensorShapesFromInputs(), SpaceToDepthLayer::ValidateTensorShapesFromInputs(), LogicalBinaryLayer::ValidateTensorShapesFromInputs(), DepthToSpaceLayer::ValidateTensorShapesFromInputs(), TransposeConvolution2dLayer::ValidateTensorShapesFromInputs(), PreluLayer::ValidateTensorShapesFromInputs(), FullyConnectedLayer::ValidateTensorShapesFromInputs(), SplitterLayer::ValidateTensorShapesFromInputs(), DepthwiseConvolution2dLayer::ValidateTensorShapesFromInputs(), UnidirectionalSequenceLstmLayer::ValidateTensorShapesFromInputs(), ConcatLayer::ValidateTensorShapesFromInputs(), Convolution2dLayer::ValidateTensorShapesFromInputs(), LstmLayer::ValidateTensorShapesFromInputs(), BatchNormalizationLayer::ValidateTensorShapesFromInputs(), QuantizedLstmLayer::ValidateTensorShapesFromInputs(), QLstmLayer::ValidateTensorShapesFromInputs(), LayerVerifierBase::VerifyConstTensors(), LayerVerifierBase::VerifyNameAndConnections(), and armnn::VerifyTensorInfoDataType().

{ return m_Shape; }

GetShape() [2/2]

TensorShape& GetShape ( )

inline

Definition at line 192 of file Tensor.hpp.

{ return m_Shape; }

HasMultipleQuantizationScales()

bool HasMultipleQuantizationScales ( ) const

inline

Definition at line 201 of file Tensor.hpp.

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

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

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

HasPerAxisQuantization()

bool HasPerAxisQuantization

(

)

const

Definition at line 446 of file Tensor.cpp.

References TensorInfo::HasMultipleQuantizationScales().

Referenced by armnnTfLiteParser::AsFloatArray(), armnn::Convert1HWOtoMIHW(), armnn::GetBiasDataType(), armnnSerializer::GetFlatBufferArgMinMaxFunction(), armnnUtils::GetPerAxisParams(), armnn::MakeDecoder(), armnn::MakeEncoder(), and TypeNotPerAxisQuantized::TypeNotPerAxisQuantized().

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

IsConstant()

bool IsConstant

(

)

const

Definition at line 509 of file Tensor.cpp.

Referenced by armnn::ClConvolution2dWorkloadValidate(), armnn::ClDepthwiseConvolutionWorkloadValidate(), armnn::ClFullyConnectedWorkloadValidate(), LayerSupportHandle::IsConvolution2dSupported(), LayerSupportHandle::IsDepthwiseConvolutionSupported(), LayerSupportHandle::IsDilatedDepthwiseConvolutionSupported(), LayerSupportHandle::IsFullyConnectedSupported(), armnn::NeonConvolution2dWorkloadValidate(), armnn::NeonDepthwiseConvolutionWorkloadValidate(), armnn::NeonFullyConnectedWorkloadValidate(), ConvertConstDequantisationLayersToConstLayersImpl::Run(), ConvertConstPermuteLayersToConstLayers::Run(), and TensorInfo::SetConstant().

{
    return m_IsConstant;
}

IsQuantized()

bool IsQuantized

(

)

const

Definition at line 504 of file Tensor.cpp.

References armnn::IsQuantizedType().

Referenced by RefCastWorkload::ExecuteAsync(), armnn::GetBiasDataType(), armnn::optimizations::pad_fold::GetZeroElement(), 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 432 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 422 of file Tensor.cpp.

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

operator=()

TensorInfo & operator=

(

const TensorInfo &

other

)

Definition at line 405 of file Tensor.cpp.

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

operator==()

bool operator==

(

const TensorInfo &

other

)

const

Definition at line 414 of file Tensor.cpp.

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

SetConstant()

void SetConstant

(

const bool

IsConstant = true

)

Marks the data corresponding to this tensor info as constant.

: This can allow further optimization on execution : The user has to ensure that the underlying data actually is constant.

Examples:

AsyncExecutionSample.cpp, CustomMemoryAllocatorSample.cpp, DynamicSample.cpp, and SimpleSample.cpp.

Definition at line 514 of file Tensor.cpp.

References TensorInfo::IsConstant().

Referenced by armnn::experimental::AsyncEndToEndTestImpl(), armnn::experimental::AsyncThreadedEndToEndTestImpl(), armnnOnnxParser::CreateConstTensorImpl(), OnnxParserImpl::CreateNetworkFromString(), CreateTestNetwork(), TfLiteParserImpl::GetBuffer(), TfLiteParserImpl::GetNetworkInputBindingInfo(), TfLiteParserImpl::GetOutputTensorIds(), GetSoftmaxProfilerJson(), armnnUtils::MakeInputTensors(), armnn::PermuteTensor(), QuantizedLstmEndToEnd(), ConvertConstDequantisationLayersToConstLayersImpl::Run(), ConvertConstPermuteLayersToConstLayers::Run(), TEST_SUITE(), armnnDeserializer::ToConstTensor(), and VerifyPostOptimisationStructureTestImpl().

{
    m_IsConstant = IsConstant;
}

SetDataType()

void SetDataType ( DataType  type )

inline

Definition at line 199 of file Tensor.hpp.

Referenced by ConvertToDataType(), TfLiteParserImpl::GetOutputTensorIds(), armnn::InsertConvertBf16ToFp32LayersBefore(), armnn::InsertConvertFp16ToFp32LayersBefore(), armnn::InsertConvertFp32ToBf16LayersAfter(), armnn::InsertConvertFp32ToBf16LayersBefore(), armnn::InsertConvertFp32ToFp16LayersAfter(), and ConvertFp32NetworkToFp16Impl::Run().

{ m_DataType = type; }

SetQuantizationDim()

void SetQuantizationDim

(

const Optional< unsigned int > &

quantizationDim

)

Definition at line 499 of file Tensor.cpp.

Referenced by armnnUtils::Permuted(), and TensorInfo::TensorInfo().

{
    m_Quantization.m_QuantizationDim = quantizationDim;
}

SetQuantizationOffset()

void SetQuantizationOffset

(

int32_t

offset

)

Definition at line 489 of file Tensor.cpp.

Referenced by AdditionBroadcast1ElementTestImpl(), AdditionBroadcastTestImpl(), ArgMaxChannelTest(), ArgMaxHeightTest(), ArgMaxSimpleTest(), ArgMinChannelTest(), ArgMinSimpleTest(), ArgMinWidthTest(), armnn::CheckScaleSetOnQuantizedType(), CompareActivationTestImpl(), ConstantLinearActivationTestCommon(), Convolution2d3x3DilationTestCommon(), DepthwiseConvolution2d3x3DilationTestCommon(), DepthwiseConvolution2dAsymmetricTestImpl(), DepthwiseConvolution2dDepthMul1TestImpl(), DepthwiseConvolution2dTestImpl(), DetectionPostProcessFastNmsQuantizedTest(), DetectionPostProcessRegularNmsQuantizedTest(), RefCastWorkload::ExecuteAsync(), FullyConnectedLargeTestCommon(), GetSoftmaxProfilerJson(), PermuteValueSet1Test(), PermuteValueSet2Test(), PermuteValueSet3Test(), ReduceMaxNegativeAxisTest(), ReduceMaxSimpleTest(), ReduceMaxSimpleTest2(), ReduceMinNegativeAxisTest(), ReduceMinSimpleTest(), ReduceProdMultipleAxisTest(), ReduceProdSimpleTest(), ReduceProdSingleAxisTest1(), ReduceProdSingleAxisTest2(), ReduceProdSingleAxisTest3(), ReduceSumMultipleAxisTest(), ReduceSumSimpleTest(), ReduceSumSingleAxisTest1(), ReduceSumSingleAxisTest2(), ReduceSumSingleAxisTest3(), SimpleConvolution2dTestImpl(), SimpleGatherNd2dTest(), SimpleGatherNd3dTest(), SimpleGatherNd4dTest(), SimplePermuteTest(), SimpleTransposeTest(), TensorInfo::TensorInfo(), TEST_SUITE(), TransposeValueSet1Test(), TransposeValueSet2Test(), and TransposeValueSet3Test().

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

SetQuantizationScale()

void SetQuantizationScale

(

float

scale

)

Definition at line 473 of file Tensor.cpp.

Referenced by AdditionBroadcast1ElementTestImpl(), AdditionBroadcastTestImpl(), ArgMaxChannelTest(), ArgMaxHeightTest(), ArgMaxSimpleTest(), ArgMinChannelTest(), ArgMinSimpleTest(), ArgMinWidthTest(), BoundedReLuTestCommon(), CastTest(), ChannelShuffle2DTest(), ChannelShuffle4DTest(), armnn::CheckScaleSetOnQuantizedType(), CompareActivationTestImpl(), ConcatDifferentInputOutputQParamTest(), ConcatUint16Test(), ConcatUint8DifferentQParamsTest(), ConcatUint8Test(), ConstantLinearActivationTestCommon(), Convolution1dTestImpl(), Convolution2d3x3DilationTestCommon(), DepthwiseConvolution2d3x3DilationTestCommon(), DepthwiseConvolution2dAsymmetricTestImpl(), DepthwiseConvolution2dDepthMul1TestImpl(), DepthwiseConvolution2dTestImpl(), DetectionPostProcessFastNmsQuantizedTest(), DetectionPostProcessRegularNmsQuantizedTest(), ElementwiseTestHelper(), ElementwiseUnaryTestHelper(), RefCastWorkload::ExecuteAsync(), FullyConnectedLargeTestCommon(), FullyConnectedTest(), GetSoftmaxProfilerJson(), PermuteValueSet1Test(), PermuteValueSet2Test(), PermuteValueSet3Test(), PreluTest(), ReduceMaxNegativeAxisTest(), ReduceMaxSimpleTest(), ReduceMaxSimpleTest2(), ReduceMinNegativeAxisTest(), ReduceMinSimpleTest(), ReduceProdMultipleAxisTest(), ReduceProdSimpleTest(), ReduceProdSingleAxisTest1(), ReduceProdSingleAxisTest2(), ReduceProdSingleAxisTest3(), ReduceSumMultipleAxisTest(), ReduceSumSimpleTest(), ReduceSumSingleAxisTest1(), ReduceSumSingleAxisTest2(), ReduceSumSingleAxisTest3(), Reshape5dTest(), ReshapeBooleanTest(), SimpleActivationTest(), SimpleChannelShuffleTest(), SimpleConvolution2dTestImpl(), SimpleFloorTest(), SimpleGatherNd2dTest(), SimpleGatherNd3dTest(), SimpleGatherNd4dTest(), SimplePermuteTest(), SimpleReshapeTest(), SimpleTransposeTest(), TensorInfo::TensorInfo(), TEST_SUITE(), TransposeValueSet1Test(), TransposeValueSet2Test(), and TransposeValueSet3Test().

{
    m_Quantization.m_Scales = { scale };
}

SetQuantizationScales()

void SetQuantizationScales

(

const std::vector< float > &

scales

)

Definition at line 456 of file Tensor.cpp.

Referenced by TensorInfo::TensorInfo().

{
    m_Quantization.m_Scales = scales;
}

SetShape()

void SetShape ( const TensorShape &  newShape )

inline

Definition at line 193 of file Tensor.hpp.

Referenced by ClGatherNdWorkload::ClGatherNdWorkload(), armnn::ClGatherNdWorkloadValidate(), CompareDepthwiseConvolution2dTestImpl(), armnn::Convert1HWOtoMIHW(), OnnxParserImpl::CreateNetworkFromString(), DepthwiseConvolution2dPerAxisQuantTest(), RefGatherNdWorkload::ExecuteAsync(), RefUnidirectionalSequenceLstmWorkload::ExecuteAsync(), NeonGatherNdWorkload::NeonGatherNdWorkload(), armnn::NeonGatherNdWorkloadValidate(), IDeserializer::DeserializerImpl::OutputShapeOfReshape(), TfLiteParserImpl::OutputShapeOfReshape(), TfLiteParserImpl::OutputShapeOfSqueeze(), armnnUtils::Permuted(), PermuteInputsForConcat(), armnn::ReshapeWeightsForAcl(), AddBroadcastReshapeLayerImpl::Run(), TEST_SUITE(), 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