ArmNN
NotReleased
|
#include "Deserializer.hpp"
#include <armnn/Descriptors.hpp>
#include <armnn/Exceptions.hpp>
#include <armnn/TypesUtils.hpp>
#include <armnn/LstmParams.hpp>
#include <armnn/QuantizedLstmParams.hpp>
#include <armnnUtils/Permute.hpp>
#include <ParserHelper.hpp>
#include <VerificationHelpers.hpp>
#include <boost/filesystem.hpp>
#include <boost/format.hpp>
#include <boost/core/ignore_unused.hpp>
#include <boost/assert.hpp>
#include <boost/numeric/conversion/cast.hpp>
#include <boost/polymorphic_cast.hpp>
#include <fstream>
#include <algorithm>
#include <limits>
#include <numeric>
Go to the source code of this file.
Namespaces | |
armnnDeserializer | |
Macros | |
#define | CHECK_TENSOR_PTR(TENSOR_PTR) CheckTensorPtr(TENSOR_PTR, CHECK_LOCATION()) |
#define | CHECK_CONST_TENSOR_SIZE(CONST_TENSOR_SIZE, TENSOR_SIZE) CheckConstTensorSize(CONST_TENSOR_SIZE, TENSOR_SIZE, CHECK_LOCATION()) |
#define | CHECK_CONST_TENSOR_PTR(TENSOR_PTR) CheckConstTensorPtr(TENSOR_PTR, CHECK_LOCATION()) |
#define | CHECK_LAYERS(GRAPH, LAYERS_INDEX, LAYER_INDEX) CheckLayers(GRAPH, LAYERS_INDEX, LAYER_INDEX, CHECK_LOCATION()) |
#define | CHECK_GRAPH(GRAPH, LAYERS_INDEX) CheckGraph(GRAPH, LAYERS_INDEX, CHECK_LOCATION()) |
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) |
#define CHECK_CONST_TENSOR_PTR | ( | TENSOR_PTR | ) | CheckConstTensorPtr(TENSOR_PTR, CHECK_LOCATION()) |
Definition at line 156 of file Deserializer.cpp.
Referenced by armnnDeserializer::ToConstTensor().
#define CHECK_CONST_TENSOR_SIZE | ( | CONST_TENSOR_SIZE, | |
TENSOR_SIZE | |||
) | CheckConstTensorSize(CONST_TENSOR_SIZE, TENSOR_SIZE, CHECK_LOCATION()) |
Definition at line 153 of file Deserializer.cpp.
Referenced by armnnDeserializer::ToConstTensor().
#define CHECK_GRAPH | ( | GRAPH, | |
LAYERS_INDEX | |||
) | CheckGraph(GRAPH, LAYERS_INDEX, CHECK_LOCATION()) |
Definition at line 162 of file Deserializer.cpp.
Referenced by Deserializer::GetNetworkOutputBindingInfo().
#define CHECK_LAYERS | ( | GRAPH, | |
LAYERS_INDEX, | |||
LAYER_INDEX | |||
) | CheckLayers(GRAPH, LAYERS_INDEX, LAYER_INDEX, CHECK_LOCATION()) |
Definition at line 159 of file Deserializer.cpp.
Referenced by Deserializer::GetInputs(), Deserializer::GetLstmDescriptor(), Deserializer::GetNetworkOutputBindingInfo(), Deserializer::GetNormalizationDescriptor(), armnnDeserializer::GetOriginsDescriptor(), Deserializer::GetOutputs(), Deserializer::GetPoolingDescriptor(), and Deserializer::OutputShapeOfReshape().
#define CHECK_TENSOR_PTR | ( | TENSOR_PTR | ) | CheckTensorPtr(TENSOR_PTR, CHECK_LOCATION()) |
Definition at line 150 of file Deserializer.cpp.
Referenced by armnnDeserializer::ToTensorInfo().