ArmNN
 21.08
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 37 of file OnnxParser.cpp.

Referenced by TEST_SUITE().

38 {
40 }
static IOnnxParser * CreateRaw()
Definition: OnnxParser.cpp:32
static void Destroy(IOnnxParser *parser)
Definition: OnnxParser.cpp:42
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 47 of file OnnxParser.cpp.

References IOnnxParser::CreateNetworkFromBinaryFile().

Referenced by IOnnxParser::CreateNetworkFromBinaryFile().

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

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

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

◆ CreateNetworkFromTextFile()

armnn::INetworkPtr CreateNetworkFromTextFile ( const char *  graphFile)

Create the network from a protobuf text file on disk.

Definition at line 52 of file OnnxParser.cpp.

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

◆ CreateRaw()

IOnnxParser * CreateRaw ( )
static

Definition at line 32 of file OnnxParser.cpp.

33 {
34  return new IOnnxParser();
35 }

◆ Destroy()

void Destroy ( IOnnxParser parser)
static

Definition at line 42 of file OnnxParser.cpp.

43 {
44  delete parser;
45 }

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

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

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

References CheckLocation::AsString().

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

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