21.05
|
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 | PoolingDescriptor = const armnnSerializer::Pooling2dDescriptor * |
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 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 29 of file Deserializer.hpp.
using LayerBaseRawPtrVector = std::vector<LayerBaseRawPtr> |
Definition at line 30 of file Deserializer.hpp.
using LayerRawPtr = const armnnSerializer::LayerBase * |
Definition at line 28 of file Deserializer.hpp.
using LstmDescriptorPtr = const armnnSerializer::LstmDescriptor * |
Definition at line 23 of file Deserializer.hpp.
using LstmInputParamsPtr = const armnnSerializer::LstmInputParams * |
Definition at line 24 of file Deserializer.hpp.
using NormalizationDescriptorPtr = const armnnSerializer::NormalizationDescriptor * |
Definition at line 22 of file Deserializer.hpp.
using PoolingDescriptor = const armnnSerializer::Pooling2dDescriptor * |
Definition at line 21 of file Deserializer.hpp.
using QLstmDescriptorPtr = const armnnSerializer::QLstmDescriptor * |
Definition at line 25 of file Deserializer.hpp.
using QunatizedLstmInputParamsPtr = const armnnSerializer::QuantizedLstmInputParams * |
Definition at line 26 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 27 of file Deserializer.hpp.
bool armnnDeserializer::CheckShape | ( | const armnn::TensorShape & | actual, |
const std::vector< uint32_t > & | expected | ||
) |
Definition at line 187 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 1715 of file Deserializer.cpp.
References CHECK_LAYERS, CHECK_LOCATION, CHECK_VALID_SIZE, IDeserializer::DeserializerImpl::GetInputs(), IDeserializer::DeserializerImpl::GetLayerName(), IDeserializer::DeserializerImpl::GetOutputs(), IConnectableLayer::GetOutputSlot(), armnnSerializer::Layer_ConcatLayer, armnnSerializer::Layer_MergerLayer, FullyConnectedDescriptor::m_BiasEnabled, FullyConnectedDescriptor::m_ConstantWeights, ComparisonDescriptor::m_Operation, ElementwiseUnaryDescriptor::m_Operation, FullyConnectedDescriptor::m_TransposeWeightMatrix, IOutputSlot::SetTensorInfo(), ToComparisonOperation(), ToConstTensor(), ToTensorInfo(), and ToUnaryOperation().
armnn::ActivationFunction armnnDeserializer::ToActivationFunction | ( | armnnSerializer::ActivationFunction | function | ) |
Definition at line 444 of file Deserializer.cpp.
References armnn::Abs, armnnSerializer::ActivationFunction_Abs, armnnSerializer::ActivationFunction_BoundedReLu, armnnSerializer::ActivationFunction_Elu, armnnSerializer::ActivationFunction_HardSwish, armnnSerializer::ActivationFunction_LeakyReLu, armnnSerializer::ActivationFunction_Linear, armnnSerializer::ActivationFunction_ReLu, armnnSerializer::ActivationFunction_Sigmoid, armnnSerializer::ActivationFunction_Sqrt, armnnSerializer::ActivationFunction_Square, armnnSerializer::ActivationFunction_TanH, 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 475 of file Deserializer.cpp.
References armnnSerializer::ArgMinMaxFunction_Max, armnnSerializer::ArgMinMaxFunction_Min, armnn::Max, and armnn::Min.
Referenced by IDeserializer::DeserializerImpl::GetNetworkOutputBindingInfo().
armnn::ComparisonOperation armnnDeserializer::ToComparisonOperation | ( | armnnSerializer::ComparisonOperation | operation | ) |
Definition at line 487 of file Deserializer.cpp.
References armnnSerializer::ComparisonOperation_Equal, armnnSerializer::ComparisonOperation_Greater, armnnSerializer::ComparisonOperation_GreaterOrEqual, armnnSerializer::ComparisonOperation_Less, armnnSerializer::ComparisonOperation_LessOrEqual, armnnSerializer::ComparisonOperation_NotEqual, 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 677 of file Deserializer.cpp.
References CheckLocation::AsString(), CHECK_CONST_TENSOR_PTR, CHECK_CONST_TENSOR_SIZE, CHECK_LOCATION, armnnSerializer::ConstTensorData_ByteData, armnnSerializer::ConstTensorData_IntData, armnnSerializer::ConstTensorData_LongData, armnnSerializer::ConstTensorData_ShortData, armnnSerializer::EnumNameConstTensorData(), TensorInfo::GetNumElements(), and ToTensorInfo().
Referenced by IDeserializer::DeserializerImpl::GetLstmDescriptor(), IDeserializer::DeserializerImpl::GetNetworkOutputBindingInfo(), GetOriginsDescriptor(), and IDeserializer::DeserializerImpl::GetQLstmDescriptor().
armnn::DataLayout armnnDeserializer::ToDataLayout | ( | armnnSerializer::DataLayout | dataLayout | ) |
Definition at line 432 of file Deserializer.cpp.
References armnnSerializer::DataLayout_NCHW, armnnSerializer::DataLayout_NHWC, armnn::NCHW, 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 524 of file Deserializer.cpp.
References armnn::LogicalAnd, armnnSerializer::LogicalBinaryOperation_LogicalAnd, armnnSerializer::LogicalBinaryOperation_LogicalOr, and armnn::LogicalOr.
Referenced by IDeserializer::DeserializerImpl::GetNetworkOutputBindingInfo().
armnn::ReduceOperation armnnDeserializer::ToReduceOperation | ( | armnnSerializer::ReduceOperation | operation | ) |
Definition at line 507 of file Deserializer.cpp.
References armnn::Max, armnn::Mean, armnn::Min, armnnSerializer::ReduceOperation_Max, armnnSerializer::ReduceOperation_Mean, armnnSerializer::ReduceOperation_Min, armnnSerializer::ReduceOperation_Sum, and armnn::Sum.
Referenced by IDeserializer::DeserializerImpl::OutputShapeOfReshape().
armnn::ResizeMethod armnnDeserializer::ToResizeMethod | ( | armnnSerializer::ResizeMethod | method | ) |
Definition at line 558 of file Deserializer.cpp.
References armnn::Bilinear, armnn::NearestNeighbor, armnnSerializer::ResizeMethod_Bilinear, and armnnSerializer::ResizeMethod_NearestNeighbor.
Referenced by IDeserializer::DeserializerImpl::OutputShapeOfReshape().
armnn::TensorInfo armnnDeserializer::ToTensorInfo | ( | TensorRawPtr | tensorPtr | ) |
Definition at line 571 of file Deserializer.cpp.
References CheckLocation::AsString(), armnn::Boolean, CHECK_LOCATION, CHECK_TENSOR_PTR, armnnSerializer::DataType_Boolean, armnnSerializer::DataType_Float16, armnnSerializer::DataType_Float32, armnnSerializer::DataType_QAsymmS8, armnnSerializer::DataType_QAsymmU8, armnnSerializer::DataType_QSymmS16, armnnSerializer::DataType_QSymmS8, armnnSerializer::DataType_QuantisedAsymm8, armnnSerializer::DataType_QuantisedSymm16, armnnSerializer::DataType_Signed32, armnnSerializer::DataType_Signed64, armnnSerializer::EnumNameDataType(), armnn::Float16, armnn::Float32, armnn::MaxNumOfTensorDimensions, armnn::QAsymmS8, armnn::QAsymmU8, armnn::QSymmS16, armnn::QSymmS8, armnn::Scalar, armnn::Signed32, and armnn::Signed64.
Referenced by armnnTfLiteParser::ComputeWrappedIndex(), TfLiteParserImpl::CreateNetworkFromBinary(), OnnxParserImpl::CreateNetworkFromString(), IDeserializer::DeserializerImpl::GetLstmDescriptor(), OnnxParserImpl::GetNetworkInputBindingInfo(), TfLiteParserImpl::GetNetworkInputBindingInfo(), IDeserializer::DeserializerImpl::GetNetworkOutputBindingInfo(), OnnxParserImpl::GetNetworkOutputBindingInfo(), TfLiteParserImpl::GetNetworkOutputBindingInfo(), IDeserializer::DeserializerImpl::GetNormalizationDescriptor(), GetOriginsDescriptor(), TfLiteParserImpl::GetOutputTensorIds(), IDeserializer::DeserializerImpl::GetPoolingDescriptor(), IDeserializer::DeserializerImpl::GetQLstmDescriptor(), IDeserializer::DeserializerImpl::OutputShapeOfReshape(), TfLiteParserImpl::OutputShapeOfReshape(), TfLiteParserImpl::OutputShapeOfSqueeze(), and ToConstTensor().
armnn::UnaryOperation armnnDeserializer::ToUnaryOperation | ( | armnnSerializer::UnaryOperation | operation | ) |
Definition at line 537 of file Deserializer.cpp.
References armnn::Abs, armnn::Exp, armnn::LogicalNot, armnn::Neg, armnn::Rsqrt, armnn::Sqrt, armnnSerializer::UnaryOperation_Abs, armnnSerializer::UnaryOperation_Exp, armnnSerializer::UnaryOperation_LogicalNot, armnnSerializer::UnaryOperation_Neg, armnnSerializer::UnaryOperation_Rsqrt, and armnnSerializer::UnaryOperation_Sqrt.
Referenced by GetOriginsDescriptor().