20.02
|
#include "TfLiteParser.hpp"
#include <armnn/Descriptors.hpp>
#include <armnn/Exceptions.hpp>
#include <armnn/Logging.hpp>
#include <armnn/TypesUtils.hpp>
#include <armnn/utility/IgnoreUnused.hpp>
#include <armnnUtils/Permute.hpp>
#include <ParserHelper.hpp>
#include <VerificationHelpers.hpp>
#include <schema_generated.h>
#include <flatbuffers/flexbuffers.h>
#include <boost/assert.hpp>
#include <boost/format.hpp>
#include <boost/numeric/conversion/cast.hpp>
#include <boost/filesystem.hpp>
#include <fstream>
#include <algorithm>
#include <limits>
#include <numeric>
#include <sstream>
Go to the source code of this file.
Namespaces | |
armnnTfLiteParser | |
Macros | |
#define | ARMNN_THROW_PARSE_EXCEPTION(msg) |
#define | CHECK_SUBGRAPH(MODEL, SUBGRAPH_INDEX) CheckSubgraph(MODEL, SUBGRAPH_INDEX, CHECK_LOCATION()) |
#define | CHECK_MODEL(MODEL, SUBGRAPH_INDEX, OPERATOR_INDEX) CheckModel(MODEL, SUBGRAPH_INDEX, OPERATOR_INDEX, CHECK_LOCATION()) |
#define | CHECK_TENSOR(MODEL, SUBGRAPH_INDEX, TENSOR_INDEX) CheckTensor(MODEL, SUBGRAPH_INDEX, TENSOR_INDEX, CHECK_LOCATION()) |
#define | CHECK_TENSOR_PTR(TENSOR_PTR) CheckTensorPtr(TENSOR_PTR, CHECK_LOCATION()) |
#define | CHECK_BUFFER(MODEL, BUFFER_INDEX) CheckBuffer(MODEL, BUFFER_INDEX, CHECK_LOCATION()) |
#define | CHECK_BUFFER_SIZE(BUFFER_PTR, TENSOR_INFO, BUFFER_ID) CheckBufferSize(BUFFER_PTR, TENSOR_INFO, BUFFER_ID, CHECK_LOCATION()) |
#define | CHECK_SUPPORTED_FUSED_ACTIVATION(OPTION, SUBGRAPH_INDEX, OPERATOR_INDEX) |
#define ARMNN_THROW_PARSE_EXCEPTION | ( | msg | ) |
Definition at line 36 of file TfLiteParser.cpp.
Referenced by TfLiteParser::OutputShapeOfReshape().
#define CHECK_BUFFER | ( | MODEL, | |
BUFFER_INDEX | |||
) | CheckBuffer(MODEL, BUFFER_INDEX, CHECK_LOCATION()) |
Definition at line 210 of file TfLiteParser.cpp.
Referenced by TfLiteParser::GetBuffer().
#define CHECK_BUFFER_SIZE | ( | BUFFER_PTR, | |
TENSOR_INFO, | |||
BUFFER_ID | |||
) | CheckBufferSize(BUFFER_PTR, TENSOR_INFO, BUFFER_ID, CHECK_LOCATION()) |
Definition at line 238 of file TfLiteParser.cpp.
Referenced by TfLiteParser::GetBuffer().
#define CHECK_MODEL | ( | MODEL, | |
SUBGRAPH_INDEX, | |||
OPERATOR_INDEX | |||
) | CheckModel(MODEL, SUBGRAPH_INDEX, OPERATOR_INDEX, CHECK_LOCATION()) |
Definition at line 124 of file TfLiteParser.cpp.
Referenced by TfLiteParser::CreateNetworkFromBinary(), TfLiteParser::GetInputs(), TfLiteParser::GetInputTensorIds(), TfLiteParser::GetOutputs(), TfLiteParser::GetOutputTensorIds(), TfLiteParser::OutputShapeOfReshape(), TfLiteParser::OutputShapeOfSqueeze(), and TfLiteParser::TfLiteParser().
#define CHECK_SUBGRAPH | ( | MODEL, | |
SUBGRAPH_INDEX | |||
) | CheckSubgraph(MODEL, SUBGRAPH_INDEX, CHECK_LOCATION()) |
Definition at line 79 of file TfLiteParser.cpp.
Referenced by TfLiteParser::GetNetworkInputBindingInfo(), TfLiteParser::GetNetworkOutputBindingInfo(), TfLiteParser::GetOutputTensorIds(), TfLiteParser::GetSubgraphInputs(), TfLiteParser::GetSubgraphInputTensorNames(), TfLiteParser::GetSubgraphOutputs(), and TfLiteParser::GetSubgraphOutputTensorNames().
#define CHECK_SUPPORTED_FUSED_ACTIVATION | ( | OPTION, | |
SUBGRAPH_INDEX, | |||
OPERATOR_INDEX | |||
) |
Definition at line 259 of file TfLiteParser.cpp.
Referenced by TfLiteParser::CreateNetworkFromBinary(), and TfLiteParser::OutputShapeOfReshape().
#define CHECK_TENSOR | ( | MODEL, | |
SUBGRAPH_INDEX, | |||
TENSOR_INDEX | |||
) | CheckTensor(MODEL, SUBGRAPH_INDEX, TENSOR_INDEX, CHECK_LOCATION()) |
Definition at line 154 of file TfLiteParser.cpp.
Referenced by TfLiteParser::CreateNetworkFromBinary(), TfLiteParser::GetOutputs(), and TfLiteParser::GetSubgraphInputs().
#define CHECK_TENSOR_PTR | ( | TENSOR_PTR | ) | CheckTensorPtr(TENSOR_PTR, CHECK_LOCATION()) |
Definition at line 172 of file TfLiteParser.cpp.
Referenced by TfLiteParser::GetBuffer().