22.08
|
Classes | |
class | ITfLiteParser |
class | TfLiteParserImpl |
Typedefs | |
using | BindingPointInfo = armnn::BindingPointInfo |
using | ITfLiteParserPtr = std::unique_ptr< ITfLiteParser, void(*)(ITfLiteParser *parser)> |
Functions | |
std::unique_ptr< float[]> | AsFloatArray (TfLiteParserImpl::BufferRawPtr bufferPtr, const TensorInfo &tensorInfo) |
unsigned int | ComputeWrappedIndex (int idx, unsigned int numDimsIn) |
Definition at line 20 of file ITfLiteParser.hpp.
using ITfLiteParserPtr = std::unique_ptr<ITfLiteParser, void(*)(ITfLiteParser* parser)> |
Definition at line 24 of file ITfLiteParser.hpp.
std::unique_ptr<float[]> armnnTfLiteParser::AsFloatArray | ( | TfLiteParserImpl::BufferRawPtr | bufferPtr, |
const TensorInfo & | tensorInfo | ||
) |
Definition at line 903 of file TfLiteParser.cpp.
References TensorShape::AreAllDimensionsSpecified(), ARMNN_ASSERT, ARMNN_ASSERT_MSG, CHECK_LOCATION, CHECK_MODEL, CHECK_SUPPORTED_FUSED_ACTIVATION, CHECK_TENSOR, CHECK_VALID_SIZE, CHECKED_NON_NEGATIVE, TfLiteParserImpl::GetBuffer(), TensorInfo::GetDataType(), armnn::GetDataTypeName(), TfLiteParserImpl::GetInputs(), TfLiteParserImpl::GetInputTensorIds(), TensorInfo::GetNumBytes(), TensorShape::GetNumDimensions(), TensorInfo::GetNumDimensions(), TensorInfo::GetNumElements(), armnnUtils::GetNumElementsAfter(), TfLiteParserImpl::GetOutputs(), IConnectableLayer::GetOutputSlot(), TfLiteParserImpl::GetOutputTensorIds(), IOutputSlot::GetOwningIConnectableLayer(), TensorInfo::GetQuantizationDim(), TensorInfo::GetQuantizationOffset(), TensorInfo::GetQuantizationScale(), TensorInfo::GetQuantizationScales(), TensorInfo::GetShape(), IConnectableLayer::GetType(), TensorInfo::HasPerAxisQuantization(), SoftmaxDescriptor::m_Beta, Convolution2dDescriptor::m_BiasEnabled, Convolution3dDescriptor::m_BiasEnabled, DepthwiseConvolution2dDescriptor::m_BiasEnabled, TransposeConvolution2dDescriptor::m_BiasEnabled, BatchToSpaceNdDescriptor::m_BlockShape, SpaceToBatchNdDescriptor::m_BlockShape, BatchToSpaceNdDescriptor::m_Crops, Pooling2dDescriptor::m_DataLayout, Convolution3dDescriptor::m_DataLayout, DepthwiseConvolution2dDescriptor::m_DataLayout, L2NormalizationDescriptor::m_DataLayout, BatchToSpaceNdDescriptor::m_DataLayout, SpaceToBatchNdDescriptor::m_DataLayout, TransposeConvolution2dDescriptor::m_DataLayout, Convolution3dDescriptor::m_DilationX, DepthwiseConvolution2dDescriptor::m_DilationX, Convolution3dDescriptor::m_DilationY, DepthwiseConvolution2dDescriptor::m_DilationY, Convolution3dDescriptor::m_DilationZ, TransposeConvolution2dDescriptor::m_OutputShape, TransposeConvolution2dDescriptor::m_OutputShapeEnabled, Pooling2dDescriptor::m_OutputShapeRounding, Convolution3dDescriptor::m_PadBack, Pooling2dDescriptor::m_PadBottom, Convolution3dDescriptor::m_PadBottom, DepthwiseConvolution2dDescriptor::m_PadBottom, TransposeConvolution2dDescriptor::m_PadBottom, Pooling2dDescriptor::m_PaddingMethod, Convolution3dDescriptor::m_PadFront, Pooling2dDescriptor::m_PadLeft, Convolution3dDescriptor::m_PadLeft, DepthwiseConvolution2dDescriptor::m_PadLeft, TransposeConvolution2dDescriptor::m_PadLeft, SpaceToBatchNdDescriptor::m_PadList, Pooling2dDescriptor::m_PadRight, Convolution3dDescriptor::m_PadRight, DepthwiseConvolution2dDescriptor::m_PadRight, TransposeConvolution2dDescriptor::m_PadRight, Pooling2dDescriptor::m_PadTop, Convolution3dDescriptor::m_PadTop, DepthwiseConvolution2dDescriptor::m_PadTop, TransposeConvolution2dDescriptor::m_PadTop, Pooling2dDescriptor::m_PoolHeight, Pooling2dDescriptor::m_PoolType, Pooling2dDescriptor::m_PoolWidth, Pooling2dDescriptor::m_StrideX, Convolution3dDescriptor::m_StrideX, DepthwiseConvolution2dDescriptor::m_StrideX, TransposeConvolution2dDescriptor::m_StrideX, Pooling2dDescriptor::m_StrideY, Convolution3dDescriptor::m_StrideY, DepthwiseConvolution2dDescriptor::m_StrideY, TransposeConvolution2dDescriptor::m_StrideY, Convolution3dDescriptor::m_StrideZ, ReshapeDescriptor::m_TargetShape, armnn::NDHWC, armnn::NHWC, armnn::numeric_cast(), IOutputSlot::SetTensorInfo(), armnnDeserializer::ToTensorInfo(), and OptionalReferenceSwitch< IsReference, T >::value().
Referenced by TfLiteParserImpl::GetBuffer().
unsigned int armnnTfLiteParser::ComputeWrappedIndex | ( | int | idx, |
unsigned int | numDimsIn | ||
) |
Definition at line 3727 of file TfLiteParser.cpp.
References armnn::Abs, armnn::Across, ARMNN_ASSERT, CHECK_LOCATION, CHECK_MODEL, CHECK_VALID_SIZE, CHECKED_NON_NEGATIVE, IOutputSlot::Connect(), armnn::Equal, armnn::Exp, TfLiteParserImpl::GetBuffer(), armnn::GetComparisonOperationAsCString(), TensorInfo::GetDataType(), TfLiteParserImpl::GetInputs(), IConnectableLayer::GetInputSlot(), TfLiteParserImpl::GetInputTensorIds(), IConnectableLayer::GetName(), TensorInfo::GetNumBytes(), TensorInfo::GetNumDimensions(), TensorInfo::GetNumElements(), IConnectableLayer::GetNumOutputSlots(), TfLiteParserImpl::GetOutputs(), IConnectableLayer::GetOutputSlot(), TfLiteParserImpl::GetOutputTensorIds(), TensorInfo::GetShape(), IOutputSlot::GetTensorInfo(), armnn::GetUnaryOperationAsCString(), armnn::Greater, armnn::GreaterOrEqual, armnn::Less, armnn::LessOrEqual, armnn::LocalBrightness, armnn::Log, armnn::LogicalNot, ActivationDescriptor::m_A, NormalizationDescriptor::m_Alpha, ArgMinMaxDescriptor::m_Axis, GatherDescriptor::m_Axis, ActivationDescriptor::m_B, NormalizationDescriptor::m_Beta, NormalizationDescriptor::m_DataLayout, ActivationDescriptor::m_Function, ArgMinMaxDescriptor::m_Function, NormalizationDescriptor::m_K, ReduceDescriptor::m_KeepDims, NormalizationDescriptor::m_NormChannelType, NormalizationDescriptor::m_NormMethodType, NormalizationDescriptor::m_NormSize, ComparisonDescriptor::m_Operation, ElementwiseUnaryDescriptor::m_Operation, ReduceDescriptor::m_ReduceOperation, ReduceDescriptor::m_vAxis, armnn::Max, armnn::MaxNumOfTensorDimensions, armnn::Min, armnn::Neg, armnn::NHWC, armnn::NotEqual, armnn::numeric_cast(), armnn::Prod, armnn::Rsqrt, IOutputSlot::SetTensorInfo(), armnn::Signed32, armnn::Signed64, armnn::Sin, armnn::Sqrt, armnn::Sum, and armnnDeserializer::ToTensorInfo().