22.11
|
Classes | |
struct | BindingPointInfo |
class | IDeserializer |
Typedefs | |
using | IDeserializerPtr = std::unique_ptr< IDeserializer, void(*)(IDeserializer *parser)> |
using | ConstTensorRawPtr = const armnnSerializer::ConstTensor * |
using | GraphPtr = const armnnSerializer::SerializedGraph * |
using | TensorRawPtr = const armnnSerializer::TensorInfo * |
using | Pooling2dDescriptor = const armnnSerializer::Pooling2dDescriptor * |
using | Pooling3dDescriptor = const armnnSerializer::Pooling3dDescriptor * |
using | NormalizationDescriptorPtr = const armnnSerializer::NormalizationDescriptor * |
using | LstmDescriptorPtr = const armnnSerializer::LstmDescriptor * |
using | LstmInputParamsPtr = const armnnSerializer::LstmInputParams * |
using | QLstmDescriptorPtr = const armnnSerializer::QLstmDescriptor * |
using | QunatizedLstmInputParamsPtr = const armnnSerializer::QuantizedLstmInputParams * |
using | TensorRawPtrVector = std::vector< TensorRawPtr > |
using | LayerRawPtr = const armnnSerializer::LayerBase * |
using | LayerBaseRawPtr = const armnnSerializer::LayerBase * |
using | LayerBaseRawPtrVector = std::vector< LayerBaseRawPtr > |
using | UnidirectionalSequenceLstmDescriptorPtr = const armnnSerializer::UnidirectionalSequenceLstmDescriptor * |
Functions | |
bool | CheckShape (const armnn::TensorShape &actual, const std::vector< uint32_t > &expected) |
armnn::DataLayout | ToDataLayout (armnnSerializer::DataLayout dataLayout) |
armnn::ActivationFunction | ToActivationFunction (armnnSerializer::ActivationFunction function) |
armnn::ArgMinMaxFunction | ToArgMinMaxFunction (armnnSerializer::ArgMinMaxFunction function) |
armnn::ComparisonOperation | ToComparisonOperation (armnnSerializer::ComparisonOperation operation) |
armnn::ReduceOperation | ToReduceOperation (armnnSerializer::ReduceOperation operation) |
armnn::LogicalBinaryOperation | ToLogicalBinaryOperation (armnnSerializer::LogicalBinaryOperation operation) |
armnn::UnaryOperation | ToUnaryOperation (armnnSerializer::UnaryOperation operation) |
armnn::PaddingMode | ToPaddingMode (armnnSerializer::PaddingMode paddingMode) |
armnn::ResizeMethod | ToResizeMethod (armnnSerializer::ResizeMethod method) |
armnn::TensorInfo | ToTensorInfo (TensorRawPtr tensorPtr) |
armnn::ConstTensor | ToConstTensor (ConstTensorRawPtr constTensorPtr) |
const armnnSerializer::OriginsDescriptor * | GetOriginsDescriptor (const armnnSerializer::SerializedGraph *graph, unsigned int layerIndex) |
using ConstTensorRawPtr = const armnnSerializer::ConstTensor * |
Definition at line 18 of file Deserializer.hpp.
using GraphPtr = const armnnSerializer::SerializedGraph * |
Definition at line 19 of file Deserializer.hpp.
using IDeserializerPtr = std::unique_ptr<IDeserializer, void(*)(IDeserializer* parser)> |
Definition at line 25 of file IDeserializer.hpp.
using LayerBaseRawPtr = const armnnSerializer::LayerBase * |
Definition at line 30 of file Deserializer.hpp.
using LayerBaseRawPtrVector = std::vector<LayerBaseRawPtr> |
Definition at line 31 of file Deserializer.hpp.
using LayerRawPtr = const armnnSerializer::LayerBase * |
Definition at line 29 of file Deserializer.hpp.
using LstmDescriptorPtr = const armnnSerializer::LstmDescriptor * |
Definition at line 24 of file Deserializer.hpp.
using LstmInputParamsPtr = const armnnSerializer::LstmInputParams * |
Definition at line 25 of file Deserializer.hpp.
using NormalizationDescriptorPtr = const armnnSerializer::NormalizationDescriptor * |
Definition at line 23 of file Deserializer.hpp.
using Pooling2dDescriptor = const armnnSerializer::Pooling2dDescriptor * |
Definition at line 21 of file Deserializer.hpp.
using Pooling3dDescriptor = const armnnSerializer::Pooling3dDescriptor * |
Definition at line 22 of file Deserializer.hpp.
using QLstmDescriptorPtr = const armnnSerializer::QLstmDescriptor * |
Definition at line 26 of file Deserializer.hpp.
using QunatizedLstmInputParamsPtr = const armnnSerializer::QuantizedLstmInputParams * |
Definition at line 27 of file Deserializer.hpp.
using TensorRawPtr = const armnnSerializer::TensorInfo * |
Definition at line 20 of file Deserializer.hpp.
using TensorRawPtrVector = std::vector<TensorRawPtr> |
Definition at line 28 of file Deserializer.hpp.
using UnidirectionalSequenceLstmDescriptorPtr = const armnnSerializer::UnidirectionalSequenceLstmDescriptor * |
Definition at line 32 of file Deserializer.hpp.
bool armnnDeserializer::CheckShape | ( | const armnn::TensorShape & | actual, |
const std::vector< uint32_t > & | expected | ||
) |
Definition at line 188 of file Deserializer.cpp.
References TensorShape::GetNumDimensions().
Referenced by IDeserializer::DeserializerImpl::OutputShapeOfReshape(), and TfLiteParserImpl::OutputShapeOfReshape().
const armnnSerializer::OriginsDescriptor* armnnDeserializer::GetOriginsDescriptor | ( | const armnnSerializer::SerializedGraph * | graph, |
unsigned int | layerIndex | ||
) |
Definition at line 1967 of file Deserializer.cpp.
References CHECK_LAYERS, CHECK_LOCATION, CHECK_VALID_SIZE, BaseTensor< MemoryType >::GetInfo(), IDeserializer::DeserializerImpl::GetInputs(), IDeserializer::DeserializerImpl::GetLayerName(), FullyConnectedDescriptor::GetNumInputs(), IDeserializer::DeserializerImpl::GetOutputs(), IConnectableLayer::GetOutputSlot(), ChannelShuffleDescriptor::m_Axis, FullyConnectedDescriptor::m_BiasEnabled, FullyConnectedDescriptor::m_ConstantWeights, ComparisonDescriptor::m_Operation, ElementwiseUnaryDescriptor::m_Operation, FullyConnectedDescriptor::m_TransposeWeightMatrix, IOutputSlot::SetTensorInfo(), ToComparisonOperation(), ToConstTensor(), ToPaddingMode(), ToTensorInfo(), and ToUnaryOperation().
armnn::ActivationFunction armnnDeserializer::ToActivationFunction | ( | armnnSerializer::ActivationFunction | function | ) |
Definition at line 470 of file Deserializer.cpp.
References armnn::Abs, armnn::BoundedReLu, armnn::Elu, armnn::HardSwish, armnn::LeakyReLu, armnn::Linear, armnn::ReLu, armnn::Sigmoid, armnn::Sqrt, armnn::Square, and armnn::TanH.
Referenced by IDeserializer::DeserializerImpl::GetNetworkOutputBindingInfo().
armnn::ArgMinMaxFunction armnnDeserializer::ToArgMinMaxFunction | ( | armnnSerializer::ArgMinMaxFunction | function | ) |
Definition at line 501 of file Deserializer.cpp.
References armnn::Max, and armnn::Min.
Referenced by IDeserializer::DeserializerImpl::GetNetworkOutputBindingInfo().
armnn::ComparisonOperation armnnDeserializer::ToComparisonOperation | ( | armnnSerializer::ComparisonOperation | operation | ) |
Definition at line 513 of file Deserializer.cpp.
References armnn::Equal, armnn::Greater, armnn::GreaterOrEqual, armnn::Less, armnn::LessOrEqual, and armnn::NotEqual.
Referenced by GetOriginsDescriptor().
armnn::ConstTensor armnnDeserializer::ToConstTensor | ( | ConstTensorRawPtr | constTensorPtr | ) |
Definition at line 722 of file Deserializer.cpp.
References CheckLocation::AsString(), CHECK_CONST_TENSOR_PTR, CHECK_CONST_TENSOR_SIZE, CHECK_LOCATION, TensorInfo::GetNumElements(), TensorInfo::SetConstant(), and ToTensorInfo().
Referenced by IDeserializer::DeserializerImpl::GetLstmDescriptor(), IDeserializer::DeserializerImpl::GetNetworkOutputBindingInfo(), GetOriginsDescriptor(), IDeserializer::DeserializerImpl::GetQLstmDescriptor(), and IDeserializer::DeserializerImpl::GetUnidirectionalSequenceLstmDescriptor().
armnn::DataLayout armnnDeserializer::ToDataLayout | ( | armnnSerializer::DataLayout | dataLayout | ) |
Definition at line 454 of file Deserializer.cpp.
References armnn::NCDHW, armnn::NCHW, armnn::NDHWC, and armnn::NHWC.
Referenced by IDeserializer::DeserializerImpl::GetNetworkOutputBindingInfo(), IDeserializer::DeserializerImpl::GetNormalizationDescriptor(), IDeserializer::DeserializerImpl::GetQLstmDescriptor(), and IDeserializer::DeserializerImpl::OutputShapeOfReshape().
armnn::LogicalBinaryOperation armnnDeserializer::ToLogicalBinaryOperation | ( | armnnSerializer::LogicalBinaryOperation | operation | ) |
Definition at line 552 of file Deserializer.cpp.
References armnn::LogicalAnd, and armnn::LogicalOr.
Referenced by IDeserializer::DeserializerImpl::GetNetworkOutputBindingInfo().
armnn::PaddingMode armnnDeserializer::ToPaddingMode | ( | armnnSerializer::PaddingMode | paddingMode | ) |
Definition at line 590 of file Deserializer.cpp.
References armnn::Constant, armnn::Reflect, and armnn::Symmetric.
Referenced by GetOriginsDescriptor().
armnn::ReduceOperation armnnDeserializer::ToReduceOperation | ( | armnnSerializer::ReduceOperation | operation | ) |
Definition at line 533 of file Deserializer.cpp.
References armnn::Max, armnn::Mean, armnn::Min, armnn::Prod, and armnn::Sum.
Referenced by IDeserializer::DeserializerImpl::OutputShapeOfReshape().
armnn::ResizeMethod armnnDeserializer::ToResizeMethod | ( | armnnSerializer::ResizeMethod | method | ) |
Definition at line 603 of file Deserializer.cpp.
References armnn::Bilinear, and armnn::NearestNeighbor.
Referenced by IDeserializer::DeserializerImpl::OutputShapeOfReshape().
armnn::TensorInfo armnnDeserializer::ToTensorInfo | ( | TensorRawPtr | tensorPtr | ) |
Definition at line 616 of file Deserializer.cpp.
References CheckLocation::AsString(), armnn::Boolean, CHECK_LOCATION, CHECK_TENSOR_PTR, armnn::Float16, armnn::Float32, armnn::MaxNumOfTensorDimensions, armnn::QAsymmS8, armnn::QAsymmU8, armnn::QSymmS16, armnn::QSymmS8, armnn::Scalar, armnn::Signed32, and armnn::Signed64.
Referenced by armnnTfLiteParser::AsFloatArray(), armnnTfLiteParser::ComputeWrappedIndex(), OnnxParserImpl::CreateNetworkFromString(), TfLiteParserImpl::GetBuffer(), IDeserializer::DeserializerImpl::GetLstmDescriptor(), TfLiteParserImpl::GetNetworkInputBindingInfo(), IDeserializer::DeserializerImpl::GetNetworkOutputBindingInfo(), TfLiteParserImpl::GetNetworkOutputBindingInfo(), IDeserializer::DeserializerImpl::GetNormalizationDescriptor(), GetOriginsDescriptor(), TfLiteParserImpl::GetOutputTensorIds(), IDeserializer::DeserializerImpl::GetPooling3dDescriptor(), IDeserializer::DeserializerImpl::GetQLstmDescriptor(), IDeserializer::DeserializerImpl::GetUnidirectionalSequenceLstmDescriptor(), IDeserializer::DeserializerImpl::OutputShapeOfReshape(), TfLiteParserImpl::OutputShapeOfReshape(), TfLiteParserImpl::OutputShapeOfSqueeze(), and ToConstTensor().
armnn::UnaryOperation armnnDeserializer::ToUnaryOperation | ( | armnnSerializer::UnaryOperation | operation | ) |
Definition at line 565 of file Deserializer.cpp.
References armnn::Abs, armnn::Exp, armnn::Log, armnn::LogicalNot, armnn::Neg, armnn::Rsqrt, armnn::Sin, and armnn::Sqrt.
Referenced by GetOriginsDescriptor().