ITfParser Class Reference

Parses a directed acyclic graph from a tensorflow protobuf file. More...

#include <ITfParser.hpp>

Inheritance diagram for ITfParser:

Public Member Functions
virtual armnn::INetworkPtr	CreateNetworkFromTextFile (const char *graphFile, const std::map< std::string, armnn::TensorShape > &inputShapes, const std::vector< std::string > &requestedOutputs)=0
	Create the network from a protobuf text file on the disk. More...
virtual armnn::INetworkPtr	CreateNetworkFromBinaryFile (const char *graphFile, const std::map< std::string, armnn::TensorShape > &inputShapes, const std::vector< std::string > &requestedOutputs)=0
	Create the network from a protobuf binary file on the disk. More...
virtual armnn::INetworkPtr	CreateNetworkFromString (const char *protoText, const std::map< std::string, armnn::TensorShape > &inputShapes, const std::vector< std::string > &requestedOutputs)=0
	Create the network directly from protobuf text in a string. Useful for debugging/testing. More...
virtual BindingPointInfo	GetNetworkInputBindingInfo (const std::string &name) const =0
	Retrieve binding info (layer id and tensor info) for the network input identified by the given layer name. More...
virtual BindingPointInfo	GetNetworkOutputBindingInfo (const std::string &name) const =0
	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)

Protected Member Functions
virtual	~ITfParser ()

Detailed Description

Parses a directed acyclic graph from a tensorflow protobuf file.

Definition at line 25 of file ITfParser.hpp.

Constructor & Destructor Documentation

~ITfParser()

virtual ~ITfParser ( )

inlineprotectedvirtual

Definition at line 57 of file ITfParser.hpp.

{};

Member Function Documentation

Create()

ITfParserPtr Create ( )

static

Definition at line 393 of file TfParser.cpp.

Referenced by BOOST_AUTO_TEST_CASE().

{
    return ITfParserPtr(CreateRaw(), &ITfParser::Destroy);
}

CreateNetworkFromBinaryFile()

virtual armnn::INetworkPtr CreateNetworkFromBinaryFile ( const char *  graphFile, const std::map< std::string, armnn::TensorShape > &  inputShapes, const std::vector< std::string > &  requestedOutputs  )

pure virtual

Create the network from a protobuf binary file on the disk.

Implemented in TfParser.

CreateNetworkFromString()

virtual armnn::INetworkPtr CreateNetworkFromString ( const char *  protoText, const std::map< std::string, armnn::TensorShape > &  inputShapes, const std::vector< std::string > &  requestedOutputs  )

pure virtual

Create the network directly from protobuf text in a string. Useful for debugging/testing.

Implemented in TfParser.

CreateNetworkFromTextFile()

virtual armnn::INetworkPtr CreateNetworkFromTextFile ( const char *  graphFile, const std::map< std::string, armnn::TensorShape > &  inputShapes, const std::vector< std::string > &  requestedOutputs  )

pure virtual

Create the network from a protobuf text file on the disk.

Implemented in TfParser.

CreateRaw()

ITfParser * CreateRaw ( )

static

Definition at line 388 of file TfParser.cpp.

{
    return new TfParser();
}

Destroy()

void Destroy ( ITfParser *  parser )

static

Definition at line 398 of file TfParser.cpp.

{
    delete parser;
}

GetNetworkInputBindingInfo()

virtual BindingPointInfo GetNetworkInputBindingInfo ( const std::string &  name ) const

pure virtual

Retrieve binding info (layer id and tensor info) for the network input identified by the given layer name.

Implemented in TfParser.

GetNetworkOutputBindingInfo()

virtual BindingPointInfo GetNetworkOutputBindingInfo ( const std::string &  name ) const

pure virtual

Retrieve binding info (layer id and tensor info) for the network output identified by the given layer name.

Implemented in TfParser.

---

The documentation for this class was generated from the following files:

• include/armnnTfParser/ITfParser.hpp
• src/armnnTfParser/TfParser.cpp