24.08
|
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::ScatterNdFunction | ToScatterNdFunction (armnnSerializer::ScatterNdFunction function) |
armnn::ComparisonOperation | ToComparisonOperation (armnnSerializer::ComparisonOperation operation) |
armnn::ReduceOperation | ToReduceOperation (armnnSerializer::ReduceOperation operation) |
armnn::LogicalBinaryOperation | ToLogicalBinaryOperation (armnnSerializer::LogicalBinaryOperation operation) |
armnn::BinaryOperation | ToElementwiseBinaryOperation (armnnSerializer::BinaryOperation operation) |
armnn::UnaryOperation | ToElementwiseUnaryOperation (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 | ||
) |
const armnnSerializer::OriginsDescriptor* armnnDeserializer::GetOriginsDescriptor | ( | const armnnSerializer::SerializedGraph * | graph, |
unsigned int | layerIndex | ||
) |
Definition at line 2055 of file Deserializer.cpp.
armnn::ActivationFunction armnnDeserializer::ToActivationFunction | ( | armnnSerializer::ActivationFunction | function | ) |
Definition at line 482 of file Deserializer.cpp.
References armnn::Abs, armnn::BoundedReLu, armnn::Elu, armnn::Gelu, armnn::HardSwish, armnn::LeakyReLu, armnn::Linear, armnn::ReLu, armnn::Sigmoid, armnn::Sqrt, armnn::Square, and armnn::TanH.
armnn::ArgMinMaxFunction armnnDeserializer::ToArgMinMaxFunction | ( | armnnSerializer::ArgMinMaxFunction | function | ) |
armnn::ComparisonOperation armnnDeserializer::ToComparisonOperation | ( | armnnSerializer::ComparisonOperation | operation | ) |
Definition at line 546 of file Deserializer.cpp.
References armnn::Equal, armnn::Greater, armnn::GreaterOrEqual, armnn::Less, armnn::LessOrEqual, and armnn::NotEqual.
armnn::ConstTensor armnnDeserializer::ToConstTensor | ( | ConstTensorRawPtr | constTensorPtr | ) |
Definition at line 782 of file Deserializer.cpp.
References CheckLocation::AsString(), CHECK_CONST_TENSOR_PTR, CHECK_CONST_TENSOR_SIZE, CHECK_LOCATION, TensorInfo::GetNumElements(), TensorInfo::SetConstant(), and ToTensorInfo().
armnn::DataLayout armnnDeserializer::ToDataLayout | ( | armnnSerializer::DataLayout | dataLayout | ) |
Definition at line 466 of file Deserializer.cpp.
References armnn::NCDHW, armnn::NCHW, armnn::NDHWC, and armnn::NHWC.
armnn::BinaryOperation armnnDeserializer::ToElementwiseBinaryOperation | ( | armnnSerializer::BinaryOperation | operation | ) |
Definition at line 598 of file Deserializer.cpp.
References armnn::Add, armnn::Div, armnn::Maximum, armnn::Minimum, armnn::Mul, armnn::Power, armnn::SqDiff, and armnn::Sub.
armnn::UnaryOperation armnnDeserializer::ToElementwiseUnaryOperation | ( | armnnSerializer::UnaryOperation | operation | ) |
Definition at line 623 of file Deserializer.cpp.
References armnn::Abs, armnn::Ceil, armnn::Exp, armnn::Log, armnn::LogicalNot, armnn::Neg, armnn::Rsqrt, armnn::Sin, and armnn::Sqrt.
armnn::LogicalBinaryOperation armnnDeserializer::ToLogicalBinaryOperation | ( | armnnSerializer::LogicalBinaryOperation | operation | ) |
Definition at line 585 of file Deserializer.cpp.
References armnn::LogicalAnd, and armnn::LogicalOr.
armnn::PaddingMode armnnDeserializer::ToPaddingMode | ( | armnnSerializer::PaddingMode | paddingMode | ) |
Definition at line 650 of file Deserializer.cpp.
References armnn::Constant, armnn::Reflect, and armnn::Symmetric.
armnn::ReduceOperation armnnDeserializer::ToReduceOperation | ( | armnnSerializer::ReduceOperation | operation | ) |
Definition at line 566 of file Deserializer.cpp.
References armnn::Max, armnn::Mean, armnn::Min, armnn::Prod, and armnn::Sum.
armnn::ResizeMethod armnnDeserializer::ToResizeMethod | ( | armnnSerializer::ResizeMethod | method | ) |
Definition at line 663 of file Deserializer.cpp.
References armnn::Bilinear, and armnn::NearestNeighbor.
armnn::ScatterNdFunction armnnDeserializer::ToScatterNdFunction | ( | armnnSerializer::ScatterNdFunction | function | ) |
Definition at line 527 of file Deserializer.cpp.
References armnn::Add, armnn::Max, armnn::Min, armnn::Sub, and armnn::Update.
armnn::TensorInfo armnnDeserializer::ToTensorInfo | ( | TensorRawPtr | tensorPtr | ) |
Definition at line 676 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 TfLiteParserImpl::GetNetworkInputBindingInfo(), TfLiteParserImpl::GetNetworkOutputBindingInfo(), and ToConstTensor().