ArmNN
NotReleased
|
Classes | |
struct | BindingPointInfo |
class | Deserializer |
class | IDeserializer |
Typedefs | |
using | IDeserializerPtr = std::unique_ptr< IDeserializer, void(*)(IDeserializer *parser)> |
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::UnaryOperation | ToUnaryOperation (armnnSerializer::UnaryOperation operation) |
armnn::ResizeMethod | ToResizeMethod (armnnSerializer::ResizeMethod method) |
armnn::TensorInfo | ToTensorInfo (Deserializer::TensorRawPtr tensorPtr) |
armnn::ConstTensor | ToConstTensor (Deserializer::ConstTensorRawPtr constTensorPtr) |
const armnnSerializer::OriginsDescriptor * | GetOriginsDescriptor (const armnnSerializer::SerializedGraph *graph, unsigned int layerIndex) |
using IDeserializerPtr = std::unique_ptr<IDeserializer, void(*)(IDeserializer* parser)> |
Definition at line 25 of file IDeserializer.hpp.
bool armnnDeserializer::CheckShape | ( | const armnn::TensorShape & | actual, |
const std::vector< uint32_t > & | expected | ||
) |
Definition at line 166 of file Deserializer.cpp.
References TensorShape::GetNumDimensions().
Referenced by Deserializer::OutputShapeOfReshape(), and TfLiteParser::OutputShapeOfReshape().
const armnnSerializer::OriginsDescriptor* armnnDeserializer::GetOriginsDescriptor | ( | const armnnSerializer::SerializedGraph * | graph, |
unsigned int | layerIndex | ||
) |
Definition at line 1543 of file Deserializer.cpp.
References CHECK_LAYERS, CHECK_LOCATION, CHECK_VALID_SIZE, Deserializer::GetInputs(), Deserializer::GetLayerName(), Deserializer::GetOutputs(), IConnectableLayer::GetOutputSlot(), FullyConnectedDescriptor::m_BiasEnabled, 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 402 of file Deserializer.cpp.
References armnn::Abs, armnn::BoundedReLu, armnn::LeakyReLu, armnn::Linear, armnn::ReLu, armnn::Sigmoid, armnn::Sqrt, armnn::Square, and armnn::TanH.
Referenced by Deserializer::GetNetworkOutputBindingInfo().
armnn::ArgMinMaxFunction armnnDeserializer::ToArgMinMaxFunction | ( | armnnSerializer::ArgMinMaxFunction | function | ) |
Definition at line 429 of file Deserializer.cpp.
References armnn::Max, and armnn::Min.
Referenced by Deserializer::GetNetworkOutputBindingInfo().
armnn::ComparisonOperation armnnDeserializer::ToComparisonOperation | ( | armnnSerializer::ComparisonOperation | operation | ) |
Definition at line 441 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 | ( | Deserializer::ConstTensorRawPtr | constTensorPtr | ) |
Definition at line 547 of file Deserializer.cpp.
References CheckLocation::AsString(), CHECK_CONST_TENSOR_PTR, CHECK_CONST_TENSOR_SIZE, CHECK_LOCATION, TensorInfo::GetNumElements(), and ToTensorInfo().
Referenced by Deserializer::GetLstmDescriptor(), Deserializer::GetNetworkOutputBindingInfo(), and GetOriginsDescriptor().
armnn::DataLayout armnnDeserializer::ToDataLayout | ( | armnnSerializer::DataLayout | dataLayout | ) |
Definition at line 390 of file Deserializer.cpp.
References armnn::NCHW, and armnn::NHWC.
Referenced by Deserializer::GetLstmDescriptor(), Deserializer::GetNetworkOutputBindingInfo(), Deserializer::GetNormalizationDescriptor(), and Deserializer::OutputShapeOfReshape().
armnn::ResizeMethod armnnDeserializer::ToResizeMethod | ( | armnnSerializer::ResizeMethod | method | ) |
Definition at line 480 of file Deserializer.cpp.
References armnn::Bilinear, and armnn::NearestNeighbor.
Referenced by Deserializer::OutputShapeOfReshape().
armnn::TensorInfo armnnDeserializer::ToTensorInfo | ( | Deserializer::TensorRawPtr | tensorPtr | ) |
Definition at line 493 of file Deserializer.cpp.
References CheckLocation::AsString(), armnn::Boolean, CHECK_LOCATION, CHECK_TENSOR_PTR, armnn::Float16, armnn::Float32, armnn::QAsymmU8, armnn::QSymmS16, and armnn::Signed32.
Referenced by TfLiteParser::CreateNetworkFromBinary(), OnnxParser::CreateNetworkFromString(), Deserializer::GetLstmDescriptor(), OnnxParser::GetNetworkInputBindingInfo(), TfLiteParser::GetNetworkInputBindingInfo(), Deserializer::GetNetworkOutputBindingInfo(), OnnxParser::GetNetworkOutputBindingInfo(), TfLiteParser::GetNetworkOutputBindingInfo(), Deserializer::GetNormalizationDescriptor(), GetOriginsDescriptor(), TfLiteParser::GetOutputTensorIds(), Deserializer::GetPoolingDescriptor(), Deserializer::OutputShapeOfReshape(), TfLiteParser::OutputShapeOfReshape(), TfLiteParser::OutputShapeOfSqueeze(), TfLiteParser::TfLiteParser(), and ToConstTensor().
armnn::UnaryOperation armnnDeserializer::ToUnaryOperation | ( | armnnSerializer::UnaryOperation | operation | ) |
Definition at line 461 of file Deserializer.cpp.
References armnn::Abs, armnn::Exp, armnn::Neg, armnn::Rsqrt, and armnn::Sqrt.
Referenced by GetOriginsDescriptor().