13 #include <unordered_map> 22 using ITfParserPtr = std::unique_ptr<ITfParser, void(*)(ITfParser* parser)>;
34 const char* graphFile,
35 const std::map<std::string, armnn::TensorShape>& inputShapes,
36 const std::vector<std::string>& requestedOutputs) = 0;
40 const char* graphFile,
41 const std::map<std::string, armnn::TensorShape>& inputShapes,
42 const std::vector<std::string>& requestedOutputs) = 0;
46 const char* protoText,
47 const std::map<std::string, armnn::TensorShape>& inputShapes,
48 const std::vector<std::string>& requestedOutputs) = 0;
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 ...
static ITfParser * CreateRaw()
std::unique_ptr< ITfParser, void(*)(ITfParser *parser)> ITfParserPtr
armnn::BindingPointInfo BindingPointInfo
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.
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.
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...
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.
static ITfParserPtr Create()
std::pair< armnn::LayerBindingId, armnn::TensorInfo > BindingPointInfo
Parses a directed acyclic graph from a tensorflow protobuf file.
std::unique_ptr< INetwork, void(*)(INetwork *network)> INetworkPtr
static void Destroy(ITfParser *parser)