IOnnxParser Class Reference

#include <IOnnxParser.hpp>

Public Member Functions
armnn::INetworkPtr	CreateNetworkFromBinary (const std::vector< uint8_t > &binaryContent)
	Create the network from a protobuf binary vector. More...
armnn::INetworkPtr	CreateNetworkFromBinary (const std::vector< uint8_t > &binaryContent, const std::map< std::string, armnn::TensorShape > &inputShapes)
	Create the network from a protobuf binary vector, with inputShapes specified. More...
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...
armnn::INetworkPtr	CreateNetworkFromBinaryFile (const char *graphFile, const std::map< std::string, armnn::TensorShape > &inputShapes)
	Create the network from a protobuf binary file on disk, with inputShapes specified. More...
armnn::INetworkPtr	CreateNetworkFromTextFile (const char *graphFile, const std::map< std::string, armnn::TensorShape > &inputShapes)
	Create the network from a protobuf text file on disk, with inputShapes specified. More...
armnn::INetworkPtr	CreateNetworkFromString (const std::string &protoText, const std::map< std::string, armnn::TensorShape > &inputShapes)
	Create the network directly from protobuf text in a string, with inputShapes specified. 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 IOnnxParser *	CreateRaw ()
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 38 of file OnnxParser.cpp.

{
    return IOnnxParserPtr(CreateRaw(), &IOnnxParser::Destroy);
}

References IDeserializer::CreateRaw().

CreateNetworkFromBinary() [1/2]

armnn::INetworkPtr CreateNetworkFromBinary

(

const std::vector< uint8_t > &

binaryContent

)

Create the network from a protobuf binary vector.

Definition at line 53 of file OnnxParser.cpp.

{
    return pOnnxParserImpl->CreateNetworkFromBinary(binaryContent);
}

CreateNetworkFromBinary() [2/2]

armnn::INetworkPtr CreateNetworkFromBinary	(	const std::vector< uint8_t > &	binaryContent,
		const std::map< std::string, armnn::TensorShape > &	inputShapes
	)

Create the network from a protobuf binary vector, with inputShapes specified.

Definition at line 58 of file OnnxParser.cpp.

{
    return pOnnxParserImpl->CreateNetworkFromBinary(binaryContent, inputShapes);
}

CreateNetworkFromBinaryFile() [1/2]

armnn::INetworkPtr CreateNetworkFromBinaryFile

(

const char *

graphFile

)

Create the network from a protobuf binary file on disk.

Definition at line 48 of file OnnxParser.cpp.

{
    return pOnnxParserImpl->CreateNetworkFromBinaryFile(graphFile);
}

References IOnnxParser::CreateNetworkFromBinaryFile().

Referenced by IOnnxParser::CreateNetworkFromBinaryFile().

CreateNetworkFromBinaryFile() [2/2]

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

Create the network from a protobuf binary file on disk, with inputShapes specified.

Definition at line 74 of file OnnxParser.cpp.

{
    return pOnnxParserImpl->CreateNetworkFromBinaryFile(graphFile, inputShapes);
}

CreateNetworkFromString() [1/2]

armnn::INetworkPtr CreateNetworkFromString

(

const std::string &

protoText

)

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

Definition at line 69 of file OnnxParser.cpp.

{
    return pOnnxParserImpl->CreateNetworkFromString(protoText);
}

CreateNetworkFromString() [2/2]

armnn::INetworkPtr CreateNetworkFromString	(	const std::string &	protoText,
		const std::map< std::string, armnn::TensorShape > &	inputShapes
	)

Create the network directly from protobuf text in a string, with inputShapes specified.

Useful for debugging/testing

Definition at line 87 of file OnnxParser.cpp.

{
    return pOnnxParserImpl->CreateNetworkFromString(protoText, inputShapes);
}

CreateNetworkFromTextFile() [1/2]

armnn::INetworkPtr CreateNetworkFromTextFile

(

const char *

graphFile

)

Create the network from a protobuf text file on disk.

Definition at line 64 of file OnnxParser.cpp.

{
    return pOnnxParserImpl->CreateNetworkFromTextFile(graphFile);
}

CreateNetworkFromTextFile() [2/2]

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

Create the network from a protobuf text file on disk, with inputShapes specified.

Definition at line 81 of file OnnxParser.cpp.

{
    return pOnnxParserImpl->CreateNetworkFromTextFile(graphFile, inputShapes);
}

CreateRaw()

IOnnxParser * CreateRaw ( )

static

Definition at line 33 of file OnnxParser.cpp.

{
    return new IOnnxParser();
}

Destroy()

void Destroy ( IOnnxParser *  parser )

static

Definition at line 43 of file OnnxParser.cpp.

{
    delete parser;
}

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

{
    return pOnnxParserImpl->GetNetworkInputBindingInfo(name);
}

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

{
    return pOnnxParserImpl->GetNetworkOutputBindingInfo(name);
}

---

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

• include/armnnOnnxParser/IOnnxParser.hpp
• src/armnnOnnxParser/OnnxParser.cpp