ArmNN
 21.05
IOnnxParser Class Reference

#include <IOnnxParser.hpp>

Public Member Functions

armnn::INetworkPtr CreateNetworkFromBinaryFile (const char *graphFile)
 Create the network from a protobuf binary file on disk. More...
 
armnn::INetworkPtr CreateNetworkFromTextFile (const char *graphFile)
 Create the network from a protobuf text file on disk. More...
 
armnn::INetworkPtr CreateNetworkFromString (const std::string &protoText)
 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 IOnnxParserCreateRaw ()
 
static IOnnxParserPtr Create ()
 
static void Destroy (IOnnxParser *parser)
 

Detailed Description

Definition at line 23 of file IOnnxParser.hpp.

Member Function Documentation

◆ Create()

IOnnxParserPtr Create ( )
static

Definition at line 36 of file OnnxParser.cpp.

Referenced by BOOST_AUTO_TEST_CASE().

37 {
39 }
static IOnnxParser * CreateRaw()
Definition: OnnxParser.cpp:31
static void Destroy(IOnnxParser *parser)
Definition: OnnxParser.cpp:41
std::unique_ptr< IOnnxParser, void(*)(IOnnxParser *parser)> IOnnxParserPtr
Definition: IOnnxParser.hpp:21

◆ CreateNetworkFromBinaryFile()

armnn::INetworkPtr CreateNetworkFromBinaryFile ( const char *  graphFile)

Create the network from a protobuf binary file on disk.

Definition at line 46 of file OnnxParser.cpp.

References IOnnxParser::CreateNetworkFromBinaryFile().

Referenced by IOnnxParser::CreateNetworkFromBinaryFile().

47 {
48  return pOnnxParserImpl->CreateNetworkFromBinaryFile(graphFile);
49 }

◆ CreateNetworkFromString()

armnn::INetworkPtr CreateNetworkFromString ( const std::string &  protoText)

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

Definition at line 56 of file OnnxParser.cpp.

57 {
58  return pOnnxParserImpl->CreateNetworkFromString(protoText);
59 }

◆ CreateNetworkFromTextFile()

armnn::INetworkPtr CreateNetworkFromTextFile ( const char *  graphFile)

Create the network from a protobuf text file on disk.

Definition at line 51 of file OnnxParser.cpp.

52 {
53  return pOnnxParserImpl->CreateNetworkFromTextFile(graphFile);
54 }

◆ CreateRaw()

IOnnxParser * CreateRaw ( )
static

Definition at line 31 of file OnnxParser.cpp.

32 {
33  return new IOnnxParser();
34 }

◆ Destroy()

void Destroy ( IOnnxParser parser)
static

Definition at line 41 of file OnnxParser.cpp.

42 {
43  delete parser;
44 }

◆ GetNetworkInputBindingInfo()

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 61 of file OnnxParser.cpp.

62 {
63  return pOnnxParserImpl->GetNetworkInputBindingInfo(name);
64 }

◆ GetNetworkOutputBindingInfo()

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 66 of file OnnxParser.cpp.

References CheckLocation::AsString().

67 {
68  return pOnnxParserImpl->GetNetworkOutputBindingInfo(name);
69 }

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