21.02
|
Parses a directed acyclic graph from a tensorflow protobuf file. More...
#include <ITfParser.hpp>
Classes | |
struct | TfParserImpl |
Public Member Functions | |
armnn::INetworkPtr | CreateNetworkFromTextFile (const char *graphFile, const std::map< std::string, armnn::TensorShape > &inputShapes, const std::vector< std::string > &requestedOutputs) |
Create the network from a protobuf text file on the disk. More... | |
armnn::INetworkPtr | CreateNetworkFromBinaryFile (const char *graphFile, const std::map< std::string, armnn::TensorShape > &inputShapes, const std::vector< std::string > &requestedOutputs) |
Create the network from a protobuf binary file on the disk. More... | |
armnn::INetworkPtr | CreateNetworkFromString (const char *protoText, const std::map< std::string, armnn::TensorShape > &inputShapes, const std::vector< std::string > &requestedOutputs) |
Create the network directly from protobuf text in a string. Useful for debugging/testing. More... | |
BindingPointInfo | GetNetworkInputBindingInfo (const std::string &name) const |
Retrieve binding info (layer id and tensor info) for the network input identified by the given layer name. More... | |
BindingPointInfo | GetNetworkOutputBindingInfo (const std::string &name) const |
Retrieve binding info (layer id and tensor info) for the network output identified by the given layer name. More... | |
Static Public Member Functions | |
static ITfParser * | CreateRaw () |
static ITfParserPtr | Create () |
static void | Destroy (ITfParser *parser) |
Friends | |
template<typename T > | |
class | ParsedConstTfOperation |
class | ParsedMatMulTfOperation |
class | ParsedMulTfOperation |
class | ParsedTfOperation |
class | SingleLayerParsedTfOperation |
class | DeferredSingleLayerParsedTfOperation |
class | ParsedIdentityTfOperation |
template<template< typename > class OperatorType, typename T > | |
struct | MakeTfOperation |
Parses a directed acyclic graph from a tensorflow protobuf file.
Definition at line 25 of file ITfParser.hpp.
|
static |
Definition at line 48 of file TfParser.cpp.
Referenced by BOOST_AUTO_TEST_CASE().
armnn::INetworkPtr CreateNetworkFromBinaryFile | ( | const char * | graphFile, |
const std::map< std::string, armnn::TensorShape > & | inputShapes, | ||
const std::vector< std::string > & | requestedOutputs | ||
) |
Create the network from a protobuf binary file on the disk.
Definition at line 65 of file TfParser.cpp.
armnn::INetworkPtr CreateNetworkFromString | ( | const char * | protoText, |
const std::map< std::string, armnn::TensorShape > & | inputShapes, | ||
const std::vector< std::string > & | requestedOutputs | ||
) |
Create the network directly from protobuf text in a string. Useful for debugging/testing.
Definition at line 72 of file TfParser.cpp.
armnn::INetworkPtr CreateNetworkFromTextFile | ( | const char * | graphFile, |
const std::map< std::string, armnn::TensorShape > & | inputShapes, | ||
const std::vector< std::string > & | requestedOutputs | ||
) |
Create the network from a protobuf text file on the disk.
Definition at line 58 of file TfParser.cpp.
References ITfParser::CreateNetworkFromTextFile().
Referenced by ITfParser::CreateNetworkFromTextFile().
|
static |
Definition at line 43 of file TfParser.cpp.
|
static |
Definition at line 53 of file TfParser.cpp.
BindingPointInfo GetNetworkInputBindingInfo | ( | const std::string & | name | ) | const |
Retrieve binding info (layer id and tensor info) for the network input identified by the given layer name.
Definition at line 79 of file TfParser.cpp.
BindingPointInfo GetNetworkOutputBindingInfo | ( | const std::string & | name | ) | const |
Retrieve binding info (layer id and tensor info) for the network output identified by the given layer name.
Definition at line 84 of file TfParser.cpp.
References INetwork::AddReshapeLayer(), armnnUtils::ArmNNToNHWC, CHECK_LOCATION, IOutputSlot::Connect(), IConnectableLayer::GetInputSlot(), TensorInfo::GetNumDimensions(), TensorInfo::GetNumElements(), IConnectableLayer::GetOutputSlot(), TensorInfo::GetShape(), IOutputSlot::GetTensorInfo(), ReshapeDescriptor::m_TargetShape, armnnUtils::NHWCToArmNN, armnn::numeric_cast(), TensorInfo::SetShape(), and IOutputSlot::SetTensorInfo().
|
friend |
Definition at line 63 of file ITfParser.hpp.
Referenced by ITfParser::TfParserImpl::ParseIdentity(), ITfParser::TfParserImpl::ParseLrn(), and ITfParser::TfParserImpl::ParseMean().
|
friend |
Definition at line 67 of file ITfParser.hpp.
Referenced by armnnTfParser::ConvertTfTensorDataType().
|
friend |
Definition at line 58 of file ITfParser.hpp.
Referenced by ITfParser::TfParserImpl::ParseIdentity().
|
friend |
Definition at line 64 of file ITfParser.hpp.
Referenced by ITfParser::TfParserImpl::ParseBiasAdd().
|
friend |
Definition at line 59 of file ITfParser.hpp.
Referenced by ITfParser::TfParserImpl::ParseLrn().
|
friend |
Definition at line 60 of file ITfParser.hpp.
Referenced by ITfParser::TfParserImpl::ParseMean().
|
friend |
|
friend |
Definition at line 62 of file ITfParser.hpp.