20.02
|
#include "TfParser.hpp"
#include <armnn/TypesUtils.hpp>
#include <armnn/Descriptors.hpp>
#include <armnnUtils/Permute.hpp>
#include <armnnUtils/DataLayoutIndexed.hpp>
#include <armnnUtils/Transpose.hpp>
#include <armnn/utility/IgnoreUnused.hpp>
#include <GraphTopologicalSort.hpp>
#include <ParserHelper.hpp>
#include <google/protobuf/io/zero_copy_stream_impl.h>
#include <google/protobuf/text_format.h>
#include <tensorflow/core/framework/graph.pb.h>
#include <boost/format.hpp>
#include <boost/numeric/conversion/cast.hpp>
#include <boost/polymorphic_cast.hpp>
#include <numeric>
Go to the source code of this file.
Namespaces | |
armnnTfParser | |
Macros | |
#define | CHECK_DATA_FORMAT(NODE_DEF, FORMAT, NODE_TYPE) |
#define | CHECK_PADDING_TYPE(NODE_DEF, PADDING) |
Functions | |
void | CalculateSamePadding (uint32_t inputSize, uint32_t stride, uint32_t filterSize, bool samePadding, uint32_t *paddingFront, uint32_t *paddingBack) |
void | CalcPadding (uint32_t input, uint32_t kernel, uint32_t stride, uint32_t &outPadHead, uint32_t &outPadTail, bool samePadding) |
DataType | ConvertTfTensorDataType (const tensorflow::DataType tfDataType, const tensorflow::NodeDef &nodeDef) |
TensorInfo | OutputShapeOfExpandDims (const tensorflow::NodeDef &nodeDef, TensorInfo inputTensorInfo) |
unsigned int | CheckPaddingTensor (const ConstTensor &paddingTensor, const TensorInfo &inputTensorInfo, const std::string &nodeName) |
TensorInfo | CalculatePaddedOutputTensorInfo (const TensorInfo &inputTensorInfo, const std::vector< std::pair< unsigned int, unsigned int >> &padList) |
TensorInfo | OutputShapeOfSqueeze (const tensorflow::NodeDef &nodeDef, TensorInfo inputTensorInfo) |
#define CHECK_DATA_FORMAT | ( | NODE_DEF, | |
FORMAT, | |||
NODE_TYPE | |||
) |
Definition at line 314 of file TfParser.cpp.
Referenced by armnnTfParser::ConvertTfTensorDataType(), armnnTfParser::OutputShapeOfExpandDims(), and armnnTfParser::OutputShapeOfSqueeze().
#define CHECK_PADDING_TYPE | ( | NODE_DEF, | |
PADDING | |||
) |
Definition at line 328 of file TfParser.cpp.
Referenced by armnnTfParser::ConvertTfTensorDataType(), and armnnTfParser::OutputShapeOfSqueeze().