TensorInfo Class Reference

#include <Tensor.hpp>

Public Types
using	DifferenceType = std::vector< TensorInfo >::difference_type

Public Member Functions
	TensorInfo ()
	Empty (invalid) constructor. More...
	TensorInfo (const TensorShape &shape, DataType dataType, float quantizationScale=1.0f, int32_t quantizationOffset=0, bool isConstant=false)
	TensorInfo (unsigned int numDimensions, const unsigned int *dimensionSizes, DataType dataType, float quantizationScale=1.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.

Member Typedef Documentation

DifferenceType

using DifferenceType = std::vector<TensorInfo>::difference_type

Definition at line 191 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)
{
}

References armnn::Float32.

TensorInfo() [2/6]

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

Definition at line 346 of file Tensor.cpp.

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

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

TensorInfo() [3/6]

TensorInfo	(	unsigned int	numDimensions,
		const unsigned int *	dimensionSizes,
		DataType	dataType,
		float	quantizationScale = 1.0f,
		int32_t	quantizationOffset = 0,
		bool	isConstant = false
	)

Definition at line 359 of file Tensor.cpp.

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

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

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.

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

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

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.

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

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

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 200 of file Tensor.hpp.

{ return m_DataType; }

Referenced by BiasAndWeightsTypesMatch::BiasAndWeightsTypesMatch(), armnnUtils::CalculateReducedOutputTensoInfo(), armnnUtils::CalculateStridedSliceOutputTensorInfo(), TosaRefTensorHandle::CanBeImported(), RefTensorHandle::CanBeImported(), armnn::ClConvertFp16ToFp32WorkloadValidate(), armnn::ClConvertFp32ToFp16WorkloadValidate(), ConvertQuantizeToTosaOperator(), armnn_driver::ConvertToLayerInputHandle(), Layer::GetDataType(), BaseTensor< void * >::GetDataType(), armnn::GetLayerInOutDatatype(), armnn::optimizations::pad_fold::GetLowestElement(), armnn::GetUnpaddedTensorStrides(), armnn::InitializeArmComputeTensorData(), armnn::InsertConvertFp16ToFp32LayersBefore(), armnn::InsertConvertFp32ToFp16LayersAfter(), SampleDynamicLayerSupport::IsAdditionSupported(), RefLayerSupport::IsConvertFp16ToFp32Supported(), RefLayerSupport::IsConvertFp32ToFp16Supported(), RefLayerSupport::IsConvolution2dSupported(), RefLayerSupport::IsConvolution3dSupported(), RefLayerSupport::IsDepthwiseConvolutionSupported(), NeonLayerSupport::IsFloorSupported(), NeonLayerSupport::IsQLstmSupported(), ClLayerSupport::IsQLstmSupported(), RefLayerSupport::IsTransposeConvolution2dSupported(), NeonLayerSupport::IsUnidirectionalSequenceLstmSupported(), armnn::LstmImpl(), armnn::Pad(), armnn::PermuteTensor(), ConvertFp32NetworkToFp16Impl::Run(), armnn_driver::SwizzleAndroidNn4dTensorToArmNn(), armnnUtils::ToFloatArray(), ArgMinMaxQueueDescriptor::Validate(), FullyConnectedQueueDescriptor::Validate(), Convolution2dQueueDescriptor::Validate(), Convolution3dQueueDescriptor::Validate(), DepthwiseConvolution2dQueueDescriptor::Validate(), QuantizeQueueDescriptor::Validate(), EqualQueueDescriptor::Validate(), ConvertFp16ToFp32QueueDescriptor::Validate(), ConvertFp32ToFp16QueueDescriptor::Validate(), GreaterQueueDescriptor::Validate(), GatherNdQueueDescriptor::Validate(), GatherQueueDescriptor::Validate(), TransposeConvolution2dQueueDescriptor::Validate(), ComparisonQueueDescriptor::Validate(), and LogicalBinaryQueueDescriptor::Validate().

GetNumBytes()

unsigned int GetNumBytes

(

)

const

Definition at line 427 of file Tensor.cpp.

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

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

Referenced by TosaRefTensorHandle::Allocate(), SampleTensorHandle::Allocate(), RefTensorHandle::Allocate(), ConstTensorPin::ConstTensorPin(), armnnOnnxParser::CreateConstTensorImpl(), ArmnnPreparedModel::ExecuteWithDummyInputs(), BaseTensor< void * >::GetNumBytes(), TosaRefTensorHandle::Manage(), SampleTensorHandle::Manage(), RefTensorHandle::Manage(), armnn::PermuteTensor(), armnn::ReorderWeightChannelsForAcl(), and ScopedTensorHandle::ScopedTensorHandle().

GetNumDimensions()

unsigned int GetNumDimensions ( ) const

inline

Definition at line 197 of file Tensor.hpp.

{ return m_Shape.GetNumDimensions(); }

References TensorShape::GetNumDimensions().

Referenced by armnn::ArgMinMax(), armnn::BatchToSpaceNd(), armnn::CalculateGatherNdKeyIndices(), armnnUtils::CalculateReducedOutputTensoInfo(), armnnUtils::CalculateStridedSliceOutputTensorInfo(), armnn::CollapseLeadingUnitDimensions(), armnn::ComputeAclAxis(), armnn::ComputeReductionTensorShape(), armnn::ComputeSoftmaxAclAxis(), armnn::Concatenate(), armnn::ConnectedToLayerType(), armnn_driver::ConvertReduce(), armnn::Gather(), armnn::GetNumActivations(), BaseTensor< void * >::GetNumDimensions(), armnn_driver::IsDynamicTensor(), RefLayerSupport::IsMeanSupported(), armnn::LogSoftmax(), TfLiteParserImpl::OutputShapeOfSqueeze(), armnn::PrintOutput(), armnnUtils::ProcessConcatInputTensorInfo(), armnn::Reduce(), armnn::ReverseV2(), PermuteAndBatchToSpaceAsDepthToSpaceImpl< PermuteType >::Run(), AddBroadcastReshapeLayerImpl::Run(), armnn::Softmax(), armnn::SpaceToBatchNd(), armnn::Split(), armnn::Splitter(), armnn::Stack(), armnn_driver::SwizzleAndroidNn4dTensorToArmNn(), armnn::Tile(), FullyConnectedQueueDescriptor::Validate(), MeanQueueDescriptor::Validate(), PadQueueDescriptor::Validate(), InstanceNormalizationQueueDescriptor::Validate(), L2NormalizationQueueDescriptor::Validate(), SpaceToBatchNdQueueDescriptor::Validate(), StridedSliceQueueDescriptor::Validate(), GatherNdQueueDescriptor::Validate(), GatherQueueDescriptor::Validate(), SliceQueueDescriptor::Validate(), BatchMatMulQueueDescriptor::Validate(), ReverseV2QueueDescriptor::Validate(), TileQueueDescriptor::Validate(), QueueDescriptor::ValidateTensorNumDimensions(), and ReduceLayer::ValidateTensorShapesFromInputs().

GetNumElements()

unsigned int GetNumElements ( ) const

inline

Definition at line 198 of file Tensor.hpp.

{ return m_Shape.GetNumElements(); }

References TensorShape::GetNumElements().

Referenced by armnn::Activation(), armnnUtils::CheckSizes(), armnn::Concatenate(), armnnOnnxParser::CreateConstTensorImpl(), armnn::Dequantize(), armnn::DetectionPostProcess(), armnn::Gather(), TensorInfo::GetNumBytes(), BaseTensor< void * >::GetNumElements(), armnn::MirrorPad(), IDeserializer::DeserializerImpl::OutputShapeOfReshape(), TfLiteParserImpl::OutputShapeOfReshape(), armnn::Pad(), armnn::PrintOutput(), armnn::Reduce(), armnn::ReverseV2(), ShapesAreSameTotalSize::ShapesAreSameTotalSize(), armnn::Split(), armnn::Splitter(), armnn::Stack(), armnn::Tile(), armnnDeserializer::ToConstTensor(), armnnUtils::ToFloatArray(), and SpaceToBatchNdQueueDescriptor::Validate().

GetQuantizationDim()

Optional< unsigned int > GetQuantizationDim

(

)

const

Definition at line 494 of file Tensor.cpp.

{
    return m_Quantization.m_QuantizationDim;
}

Referenced by armnnUtils::ToFloatArray().

GetQuantizationOffset()

int32_t GetQuantizationOffset

(

)

const

Definition at line 478 of file Tensor.cpp.

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

Referenced by ConvertQuantizeToTosaOperator(), armnn::optimizations::pad_fold::GetLowestElement(), armnn::optimizations::pad_fold::GetZeroElement(), TensorInfo::IsTypeSpaceMatch(), QuantizationParametersAreEqual::QuantizationParametersAreEqual(), armnnUtils::ToFloatArray(), and QLstmQueueDescriptor::Validate().

GetQuantizationScale()

float GetQuantizationScale

(

)

const

Definition at line 461 of file Tensor.cpp.

{
    if (m_Quantization.m_Scales.empty())
    {
        // NOTE: old default for backward compatibility
        return 1.0f;
    }
 
    ARMNN_ASSERT(!HasMultipleQuantizationScales());
    return m_Quantization.m_Scales[0];
}

References ARMNN_ASSERT, and TensorInfo::HasMultipleQuantizationScales().

Referenced by ConvertQuantizeToTosaOperator(), armnn::optimizations::pad_fold::GetLowestElement(), TensorInfo::IsTypeSpaceMatch(), QuantizationParametersAreEqual::QuantizationParametersAreEqual(), armnnUtils::ToFloatArray(), and QLstmQueueDescriptor::Validate().

GetQuantizationScales()

std::vector< float > GetQuantizationScales

(

)

const

Definition at line 451 of file Tensor.cpp.

{
    return m_Quantization.m_Scales;
}

Referenced by armnnUtils::ToFloatArray().

GetShape() [1/2]

TensorShape& GetShape ( )

inline

Definition at line 194 of file Tensor.hpp.

{ return m_Shape; }

GetShape() [2/2]

const TensorShape& GetShape ( ) const

inline

Definition at line 193 of file Tensor.hpp.

{ return m_Shape; }

Referenced by armnn::ArgMinMax(), BatchMatMul::BatchMatMul(), armnn::BatchNormImpl(), armnn::BatchToSpaceNd(), armnn::CalculateGatherNdKeyIndices(), armnnUtils::CalculateReducedOutputTensoInfo(), armnnUtils::CalculateStridedSliceOutputTensorInfo(), armnn::ClUnidirectionalSequenceLstmFloatWorkloadValidate(), armnn::CollapseLeadingUnitDimensions(), armnn::Concatenate(), armnn::Convert1HWOTensorInfoToAcl(), armnn::Convert1HWOTensorToAcl(), armnn::Convert1HWOtoMIHW(), armnn_driver::ConvertPooling2d(), ConvertQuantizeToTosaOperator(), armnn::ConvertWeightTensorFromArmnnToAcl(), armnn::CreateAclNormalizationLayerInfoForL2Normalization(), armnnOnnxParser::CreateConstTensorImpl(), armnn::DepthToSpace(), armnn::DetectionPostProcess(), armnn::ExecuteFunction(), armnn::Gather(), armnn::GetNumActivations(), TosaRefTensorHandle::GetShape(), SampleTensorHandle::GetShape(), RefTensorHandle::GetShape(), ConstTensorHandle::GetShape(), RefTensorHandleDecorator::GetShape(), BaseTensor< void * >::GetShape(), armnn::GetUnpaddedTensorStrides(), armnn::InstanceNorm(), armnn_driver::IsDynamicTensor(), ClLayerSupport::IsSplitterSupported(), NeonLayerSupport::IsSplitterSupported(), armnn::LogSoftmax(), armnn::LstmImpl(), armnn::MirrorPad(), armnn::NeonUnidirectionalSequenceLstmFloatWorkloadValidate(), armnn::NeonUnidirectionalSequenceLstmWorkloadValidate(), TfLiteParserImpl::OutputShapeOfSqueeze(), armnn::Pad(), armnn::PermuteTensor(), armnn::Pooling2d(), armnn::Pooling3d(), armnn::PreluImpl(), armnn::PrintOutput(), armnnUtils::ProcessConcatInputTensorInfo(), armnn::Reduce(), armnnUtils::ReduceDims(), RefTransposeConvolution2dWorkload::RefTransposeConvolution2dWorkload(), armnn::ReshapeWeightsForAcl(), armnn::Resize(), armnn::ReverseV2(), PermuteAsReshapeImpl::Run(), TransposeAsReshapeImpl::Run(), OptimizeConsecutiveReshapesImpl::Run(), AddBroadcastReshapeLayerImpl::Run(), FuseBatchNorm< ConvLayer, ArmnnType, T >::Run(), Convolution2dLayer::SerializeLayerParameters(), Convolution3dLayer::SerializeLayerParameters(), DepthwiseConvolution2dLayer::SerializeLayerParameters(), Graph::SerializeToDot(), ShapesAreBroadcastCompatible::ShapesAreBroadcastCompatible(), ShapesAreSameRank::ShapesAreSameRank(), armnn::Slice(), armnn::Softmax(), armnn::SpaceToBatchNd(), armnn::SpaceToDepth(), armnn::Split(), armnn::Splitter(), armnn::Stack(), armnn::StridedSlice(), armnn_driver::SwizzleAndroidNn4dTensorToArmNn(), armnn::Tile(), armnnUtils::ToFloatArray(), ArgMinMaxQueueDescriptor::Validate(), PermuteQueueDescriptor::Validate(), DepthwiseConvolution2dQueueDescriptor::Validate(), DetectionPostProcessQueueDescriptor::Validate(), ResizeQueueDescriptor::Validate(), SpaceToBatchNdQueueDescriptor::Validate(), SpaceToDepthQueueDescriptor::Validate(), BatchToSpaceNdQueueDescriptor::Validate(), TransposeQueueDescriptor::Validate(), QLstmQueueDescriptor::Validate(), QuantizedLstmQueueDescriptor::Validate(), SliceQueueDescriptor::Validate(), DepthToSpaceQueueDescriptor::Validate(), BatchMatMulQueueDescriptor::Validate(), QueueDescriptor::ValidateTensorNumDimensions(), OutputSlot::ValidateTensorShape(), ElementwiseBaseLayer::ValidateTensorShapesFromInputs(), QuantizeLayer::ValidateTensorShapesFromInputs(), ChannelShuffleLayer::ValidateTensorShapesFromInputs(), ActivationLayer::ValidateTensorShapesFromInputs(), FillLayer::ValidateTensorShapesFromInputs(), ConvertFp32ToFp16Layer::ValidateTensorShapesFromInputs(), RankLayer::ValidateTensorShapesFromInputs(), CastLayer::ValidateTensorShapesFromInputs(), SoftmaxLayer::ValidateTensorShapesFromInputs(), AbsLayer::ValidateTensorShapesFromInputs(), StackLayer::ValidateTensorShapesFromInputs(), TransposeLayer::ValidateTensorShapesFromInputs(), FakeQuantizationLayer::ValidateTensorShapesFromInputs(), BatchToSpaceNdLayer::ValidateTensorShapesFromInputs(), FloorLayer::ValidateTensorShapesFromInputs(), SwitchLayer::ValidateTensorShapesFromInputs(), ResizeLayer::ValidateTensorShapesFromInputs(), Pooling3dLayer::ValidateTensorShapesFromInputs(), InstanceNormalizationLayer::ValidateTensorShapesFromInputs(), DequantizeLayer::ValidateTensorShapesFromInputs(), Pooling2dLayer::ValidateTensorShapesFromInputs(), SliceLayer::ValidateTensorShapesFromInputs(), MemImportLayer::ValidateTensorShapesFromInputs(), L2NormalizationLayer::ValidateTensorShapesFromInputs(), DebugLayer::ValidateTensorShapesFromInputs(), NormalizationLayer::ValidateTensorShapesFromInputs(), PermuteLayer::ValidateTensorShapesFromInputs(), ShapeLayer::ValidateTensorShapesFromInputs(), RsqrtLayer::ValidateTensorShapesFromInputs(), ConvertFp16ToFp32Layer::ValidateTensorShapesFromInputs(), MemCopyLayer::ValidateTensorShapesFromInputs(), MergeLayer::ValidateTensorShapesFromInputs(), MeanLayer::ValidateTensorShapesFromInputs(), PadLayer::ValidateTensorShapesFromInputs(), LogSoftmaxLayer::ValidateTensorShapesFromInputs(), ReshapeLayer::ValidateTensorShapesFromInputs(), BroadcastToLayer::ValidateTensorShapesFromInputs(), DepthwiseConvolution2dLayer::ValidateTensorShapesFromInputs(), Convolution2dLayer::ValidateTensorShapesFromInputs(), FullyConnectedLayer::ValidateTensorShapesFromInputs(), Convolution3dLayer::ValidateTensorShapesFromInputs(), GatherNdLayer::ValidateTensorShapesFromInputs(), ElementwiseBinaryLayer::ValidateTensorShapesFromInputs(), TileLayer::ValidateTensorShapesFromInputs(), BatchMatMulLayer::ValidateTensorShapesFromInputs(), GatherLayer::ValidateTensorShapesFromInputs(), DetectionPostProcessLayer::ValidateTensorShapesFromInputs(), ArgMinMaxLayer::ValidateTensorShapesFromInputs(), StridedSliceLayer::ValidateTensorShapesFromInputs(), ReduceLayer::ValidateTensorShapesFromInputs(), ElementwiseUnaryLayer::ValidateTensorShapesFromInputs(), ReverseV2Layer::ValidateTensorShapesFromInputs(), PreluLayer::ValidateTensorShapesFromInputs(), LogicalBinaryLayer::ValidateTensorShapesFromInputs(), ComparisonLayer::ValidateTensorShapesFromInputs(), TransposeConvolution2dLayer::ValidateTensorShapesFromInputs(), SpaceToBatchNdLayer::ValidateTensorShapesFromInputs(), SpaceToDepthLayer::ValidateTensorShapesFromInputs(), DepthToSpaceLayer::ValidateTensorShapesFromInputs(), SplitterLayer::ValidateTensorShapesFromInputs(), UnidirectionalSequenceLstmLayer::ValidateTensorShapesFromInputs(), LstmLayer::ValidateTensorShapesFromInputs(), ConcatLayer::ValidateTensorShapesFromInputs(), BatchNormalizationLayer::ValidateTensorShapesFromInputs(), QuantizedLstmLayer::ValidateTensorShapesFromInputs(), and QLstmLayer::ValidateTensorShapesFromInputs().

HasMultipleQuantizationScales()

bool HasMultipleQuantizationScales ( ) const

inline

Definition at line 203 of file Tensor.hpp.

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

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

HasPerAxisQuantization()

bool HasPerAxisQuantization

(

)

const

Definition at line 446 of file Tensor.cpp.

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

References TensorInfo::HasMultipleQuantizationScales().

Referenced by armnn::Convert1HWOtoMIHW(), and armnnUtils::ToFloatArray().

IsConstant()

bool IsConstant

(

)

const

Definition at line 509 of file Tensor.cpp.

{
    return m_IsConstant;
}

Referenced by NeonFullyConnectedWorkload::Execute(), LayerWithParameters< StridedSliceDescriptor >::GetConnectedConstantAsInputTensors(), LayerSupportHandle::IsConvolution2dSupported(), LayerSupportHandle::IsDepthwiseConvolutionSupported(), LayerSupportHandle::IsDilatedDepthwiseConvolutionSupported(), LayerSupportHandle::IsFullyConnectedSupported(), NeonFullyConnectedWorkload::NeonFullyConnectedWorkload(), and TensorInfo::SetConstant().

IsQuantized()

bool IsQuantized

(

)

const

Definition at line 504 of file Tensor.cpp.

{
    return IsQuantizedType(m_DataType);
}

References armnn::IsQuantizedType().

Referenced by armnn::optimizations::pad_fold::GetZeroElement(), TensorInfo::IsTypeSpaceMatch(), armnn::Pad(), and armnn::optimizations::pad_fold::TryFoldPadIntoLayer2d().

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.

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

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

Referenced by ClLayerSupport::IsConcatSupported(), NeonLayerSupport::IsConcatSupported(), NeonLayerSupport::IsSplitterSupported(), and ClLayerSupport::IsSplitterSupported().

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 @Note: The user has to ensure that the underlying data actually is constant.

Examples

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

Definition at line 514 of file Tensor.cpp.

{
    m_IsConstant = IsConstant;
}

References TensorInfo::IsConstant().

Referenced by ConstTensor::ConstTensor(), armnn_driver::ConvertOperandToConstTensorPin(), ArmnnPreparedModel::ExecuteWithDummyInputs(), TfLiteParserImpl::GetNetworkInputBindingInfo(), armnnUtils::MakeInputTensors(), armnn::PermuteTensor(), and armnnDeserializer::ToConstTensor().

SetDataType()

void SetDataType ( DataType  type )

inline

Definition at line 201 of file Tensor.hpp.

{ m_DataType = type; }

Referenced by armnn::AttemptBackendAssignment(), armnn_driver::ConvertOperandToConstTensorPin(), ConvertToDataType(), armnn::InsertConvertFp16ToFp32LayersBefore(), armnn::InsertConvertFp32ToFp16LayersAfter(), and ConvertFp32NetworkToFp16Impl::Run().

SetQuantizationDim()

void SetQuantizationDim

(

const Optional< unsigned int > &

quantizationDim

)

Definition at line 499 of file Tensor.cpp.

{
    m_Quantization.m_QuantizationDim = quantizationDim;
}

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

SetQuantizationOffset()

void SetQuantizationOffset

(

int32_t

offset

)

Definition at line 489 of file Tensor.cpp.

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

Referenced by TensorInfo::TensorInfo().

SetQuantizationScale()

void SetQuantizationScale

(

float

scale

)

Definition at line 473 of file Tensor.cpp.

{
    m_Quantization.m_Scales = { scale };
}

Referenced by TensorInfo::TensorInfo().

SetQuantizationScales()

void SetQuantizationScales

(

const std::vector< float > &

scales

)

Definition at line 456 of file Tensor.cpp.

{
    m_Quantization.m_Scales = scales;
}

Referenced by TensorInfo::TensorInfo().

SetShape()

void SetShape ( const TensorShape &  newShape )

inline

Definition at line 195 of file Tensor.hpp.

{ m_Shape = newShape; }

Referenced by ClGatherNdWorkload::ClGatherNdWorkload(), armnn::ClGatherNdWorkloadValidate(), armnn::CollapseLeadingUnitDimensions(), armnn::Convert1HWOtoMIHW(), armnn_driver::ConvertOperandToConstTensorPin(), NeonGatherNdWorkload::NeonGatherNdWorkload(), armnn::NeonGatherNdWorkloadValidate(), IDeserializer::DeserializerImpl::OutputShapeOfReshape(), TfLiteParserImpl::OutputShapeOfReshape(), TfLiteParserImpl::OutputShapeOfSqueeze(), armnnUtils::Permuted(), armnnUtils::ReduceDims(), armnn::RemoveReshapeLayer(), armnn::ReshapeWeightsForAcl(), AddBroadcastReshapeLayerImpl::Run(), and armnnUtils::TransposeTensorShape().

---

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

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