21.11
|
#include "TfLiteParser.hpp"
#include "armnnTfLiteParser/Version.hpp"
#include <armnn/BackendOptions.hpp>
#include <armnn/Descriptors.hpp>
#include <armnn/Exceptions.hpp>
#include <armnn/Logging.hpp>
#include <armnn/Tensor.hpp>
#include <armnnUtils/TensorUtils.hpp>
#include <armnn/TypesUtils.hpp>
#include <armnn/utility/Assert.hpp>
#include <armnn/utility/IgnoreUnused.hpp>
#include <armnn/utility/NumericCast.hpp>
#include <armnnUtils/Permute.hpp>
#include <armnnUtils/Filesystem.hpp>
#include <ParserHelper.hpp>
#include <VerificationHelpers.hpp>
#include <schema_generated.h>
#include <flatbuffers/flexbuffers.h>
#include <fmt/format.h>
#include <algorithm>
#include <fstream>
#include <iostream>
#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) |
Functions | |
unsigned int | ComputeWrappedIndex (int idx, unsigned int numDimsIn) |
#define ARMNN_THROW_PARSE_EXCEPTION | ( | msg | ) |
Definition at line 42 of file TfLiteParser.cpp.
Referenced by TfLiteParserImpl::OutputShapeOfReshape(), and TfLiteParserImpl::OutputShapeOfSqueeze().
#define CHECK_BUFFER | ( | MODEL, | |
BUFFER_INDEX | |||
) | CheckBuffer(MODEL, BUFFER_INDEX, CHECK_LOCATION()) |
Definition at line 260 of file TfLiteParser.cpp.
Referenced by TfLiteParserImpl::GetBuffer().
#define CHECK_BUFFER_SIZE | ( | BUFFER_PTR, | |
TENSOR_INFO, | |||
BUFFER_ID | |||
) | CheckBufferSize(BUFFER_PTR, TENSOR_INFO, BUFFER_ID, CHECK_LOCATION()) |
Definition at line 287 of file TfLiteParser.cpp.
Referenced by TfLiteParserImpl::GetBuffer().
#define CHECK_MODEL | ( | MODEL, | |
SUBGRAPH_INDEX, | |||
OPERATOR_INDEX | |||
) | CheckModel(MODEL, SUBGRAPH_INDEX, OPERATOR_INDEX, CHECK_LOCATION()) |
Definition at line 183 of file TfLiteParser.cpp.
Referenced by armnnTfLiteParser::ComputeWrappedIndex(), TfLiteParserImpl::GetInputs(), TfLiteParserImpl::GetInputTensorIds(), TfLiteParserImpl::GetOutputs(), TfLiteParserImpl::GetOutputTensorIds(), TfLiteParserImpl::LoadModel(), TfLiteParserImpl::OutputShapeOfReshape(), and TfLiteParserImpl::OutputShapeOfSqueeze().
#define CHECK_SUBGRAPH | ( | MODEL, | |
SUBGRAPH_INDEX | |||
) | CheckSubgraph(MODEL, SUBGRAPH_INDEX, CHECK_LOCATION()) |
Definition at line 141 of file TfLiteParser.cpp.
Referenced by TfLiteParserImpl::GetNetworkInputBindingInfo(), TfLiteParserImpl::GetNetworkOutputBindingInfo(), TfLiteParserImpl::GetOutputTensorIds(), TfLiteParserImpl::GetSubgraphInputs(), TfLiteParserImpl::GetSubgraphInputTensorNames(), TfLiteParserImpl::GetSubgraphOutputs(), and TfLiteParserImpl::GetSubgraphOutputTensorNames().
#define CHECK_SUPPORTED_FUSED_ACTIVATION | ( | OPTION, | |
SUBGRAPH_INDEX, | |||
OPERATOR_INDEX | |||
) |
Definition at line 308 of file TfLiteParser.cpp.
Referenced by TfLiteParserImpl::LoadModel(), and TfLiteParserImpl::OutputShapeOfReshape().
#define CHECK_TENSOR | ( | MODEL, | |
SUBGRAPH_INDEX, | |||
TENSOR_INDEX | |||
) | CheckTensor(MODEL, SUBGRAPH_INDEX, TENSOR_INDEX, CHECK_LOCATION()) |
Definition at line 212 of file TfLiteParser.cpp.
Referenced by TfLiteParserImpl::GetOutputs(), TfLiteParserImpl::GetSubgraphInputs(), and TfLiteParserImpl::LoadModel().
#define CHECK_TENSOR_PTR | ( | TENSOR_PTR | ) | CheckTensorPtr(TENSOR_PTR, CHECK_LOCATION()) |
Definition at line 225 of file TfLiteParser.cpp.
Referenced by TfLiteParserImpl::GetBuffer().