21.02
|
Classes | |
class | ITfParser |
Parses a directed acyclic graph from a tensorflow protobuf file. More... | |
struct | WithOutputTensorIndex |
WithOutputTensorIndex wraps a value and an index. More... | |
Typedefs | |
using | BindingPointInfo = armnn::BindingPointInfo |
using | ITfParserPtr = std::unique_ptr< ITfParser, void(*)(ITfParser *parser)> |
using | ParsedTfOperationPtr = std::unique_ptr< ParsedTfOperation > |
using | OutputOfParsedTfOperation = WithOutputTensorIndex< ParsedTfOperation * > |
using | OutputOfConstNodeDef = WithOutputTensorIndex< const tensorflow::NodeDef * > |
using | OutputId = WithOutputTensorIndex< std::string > |
Functions | |
void | CalcPadding (uint32_t inputSize, uint32_t filterSize, uint32_t stride, uint32_t dilation, uint32_t &paddingFront, uint32_t &paddingBack, bool samePadding) |
DataType | ConvertTfTensorDataType (const tensorflow::DataType tfDataType, const tensorflow::NodeDef &nodeDef) |
TensorInfo | OutputShapeOfExpandDims (const tensorflow::NodeDef &nodeDef, TensorInfo inputTensorInfo, std::int32_t expandDim) |
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) |
Definition at line 19 of file ITfParser.hpp.
using ITfParserPtr = std::unique_ptr<ITfParser, void(*)(ITfParser* parser)> |
Definition at line 22 of file ITfParser.hpp.
using OutputId = WithOutputTensorIndex<std::string> |
Definition at line 62 of file TfParser.hpp.
using OutputOfConstNodeDef = WithOutputTensorIndex<const tensorflow::NodeDef*> |
Definition at line 61 of file TfParser.hpp.
using OutputOfParsedTfOperation = WithOutputTensorIndex<ParsedTfOperation *> |
Definition at line 60 of file TfParser.hpp.
using ParsedTfOperationPtr = std::unique_ptr<ParsedTfOperation> |
Definition at line 35 of file TfParser.hpp.
void armnnTfParser::CalcPadding | ( | uint32_t | inputSize, |
uint32_t | filterSize, | ||
uint32_t | stride, | ||
uint32_t | dilation, | ||
uint32_t & | paddingFront, | ||
uint32_t & | paddingBack, | ||
bool | samePadding | ||
) |
Definition at line 429 of file TfParser.cpp.
References ARMNN_ASSERT, CHECK_LOCATION, Layer::GetName(), Layer::GetNumOutputSlots(), Layer::GetOutputSlot(), and m_Layer.
Referenced by TfLiteParserImpl::CreateNetworkFromBinary(), OnnxParserImpl::CreateNetworkFromString(), ITfParser::TfParserImpl::ParseConv2D(), ITfParser::TfParserImpl::ParseDepthwiseConv2D(), and ITfParser::TfParserImpl::ParsePooling2d().
TensorInfo armnnTfParser::CalculatePaddedOutputTensorInfo | ( | const TensorInfo & | inputTensorInfo, |
const std::vector< std::pair< unsigned int, unsigned int >> & | padList | ||
) |
Definition at line 2176 of file TfParser.cpp.
References TensorInfo::GetNumDimensions(), TensorInfo::GetShape(), and TensorInfo::SetShape().
Referenced by ITfParser::TfParserImpl::ParsePad().
unsigned int armnnTfParser::CheckPaddingTensor | ( | const ConstTensor & | paddingTensor, |
const TensorInfo & | inputTensorInfo, | ||
const std::string & | nodeName | ||
) |
Definition at line 2147 of file TfParser.cpp.
References CHECK_LOCATION, TensorInfo::GetNumDimensions(), and BaseTensor< MemoryType >::GetShape().
Referenced by ITfParser::TfParserImpl::ParsePad().
DataType armnnTfParser::ConvertTfTensorDataType | ( | const tensorflow::DataType | tfDataType, |
const tensorflow::NodeDef & | nodeDef | ||
) |
Definition at line 952 of file TfParser.cpp.
References CHECK_LOCATION, and ITfParser::MakeTfOperation.
Referenced by ITfParser::TfParserImpl::ParseConst().
TensorInfo armnnTfParser::OutputShapeOfExpandDims | ( | const tensorflow::NodeDef & | nodeDef, |
TensorInfo | inputTensorInfo, | ||
std::int32_t | expandDim | ||
) |
Definition at line 1466 of file TfParser.cpp.
References ARMNN_ASSERT, CHECK_LOCATION, TensorInfo::GetNumDimensions(), TensorInfo::GetShape(), armnn::numeric_cast(), and TensorInfo::SetShape().
Referenced by ITfParser::TfParserImpl::ParseExpandDims().
TensorInfo armnnTfParser::OutputShapeOfSqueeze | ( | const tensorflow::NodeDef & | nodeDef, |
TensorInfo | inputTensorInfo | ||
) |
Definition at line 2485 of file TfParser.cpp.
References ARMNN_ASSERT, CHECK_LOCATION, TensorInfo::GetNumDimensions(), TensorInfo::GetShape(), and TensorInfo::SetShape().
Referenced by ITfParser::TfParserImpl::ParseSqueeze().