ArmNN
NotReleased
|
Parses a directed acyclic graph from a tensorflow protobuf file. More...
#include <ITfParser.hpp>
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 () |
Parses a directed acyclic graph from a tensorflow protobuf file.
Definition at line 25 of file ITfParser.hpp.
|
inlineprotectedvirtual |
Definition at line 57 of file ITfParser.hpp.
|
static |
Definition at line 393 of file TfParser.cpp.
Referenced by BOOST_AUTO_TEST_CASE().
|
pure virtual |
Create the network from a protobuf binary file on the disk.
Implemented in TfParser.
|
pure virtual |
Create the network directly from protobuf text in a string. Useful for debugging/testing.
Implemented in TfParser.
|
pure virtual |
Create the network from a protobuf text file on the disk.
Implemented in TfParser.
|
static |
Definition at line 388 of file TfParser.cpp.
|
static |
Definition at line 398 of file TfParser.cpp.
|
pure virtual |
Retrieve binding info (layer id and tensor info) for the network input identified by the given layer name.
Implemented in TfParser.
|
pure virtual |
Retrieve binding info (layer id and tensor info) for the network output identified by the given layer name.
Implemented in TfParser.