ICaffeParser Class Reference

#include <ICaffeParser.hpp>

Inheritance diagram for ICaffeParser:

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/testin.g. 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 ICaffeParser *	CreateRaw ()
static ICaffeParserPtr	Create ()
static void	Destroy (ICaffeParser *parser)

Protected Member Functions
virtual	~ICaffeParser ()

Detailed Description

Definition at line 24 of file ICaffeParser.hpp.

Constructor & Destructor Documentation

~ICaffeParser()

virtual ~ICaffeParser ( )

inlineprotectedvirtual

Definition at line 56 of file ICaffeParser.hpp.

{};

Member Function Documentation

Create()

ICaffeParserPtr Create ( )

static

Definition at line 264 of file CaffeParser.cpp.

References ICaffeParser::Destroy().

Referenced by BOOST_AUTO_TEST_CASE(), and main().

{
    return ICaffeParserPtr(CreateRaw(), &ICaffeParser::Destroy);
}

CreateNetworkFromBinaryFile()

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

pure virtual

Create the network from a protobuf binary file on the disk.

Implemented in CaffeParser, and RecordByRecordCaffeParser.

CreateNetworkFromString()

virtual armnn::INetworkPtr CreateNetworkFromString ( const char *  protoText, const std::map< std::string, armnn::TensorShape > &  inputShapes, const std::vector< std::string > &  requestedOutputs  )

pure virtual

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

Implemented in CaffeParserBase.

CreateNetworkFromTextFile()

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

pure virtual

Create the network from a protobuf text file on the disk.

Implemented in CaffeParserBase.

CreateRaw()

ICaffeParser * CreateRaw ( )

static

Definition at line 259 of file CaffeParser.cpp.

{
    return new RecordByRecordCaffeParser();
}

Destroy()

void Destroy ( ICaffeParser *  parser )

static

Definition at line 269 of file CaffeParser.cpp.

Referenced by ICaffeParser::Create().

{
    delete parser;
}

GetNetworkInputBindingInfo()

virtual BindingPointInfo GetNetworkInputBindingInfo ( const std::string &  name ) const

pure virtual

Retrieve binding info (layer id and tensor info) for the network input identified by the given layer name.

Implemented in CaffeParserBase.

GetNetworkOutputBindingInfo()

virtual BindingPointInfo GetNetworkOutputBindingInfo ( const std::string &  name ) const

pure virtual

Retrieve binding info (layer id and tensor info) for the network output identified by the given layer name.

Implemented in CaffeParserBase.

---

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

• include/armnnCaffeParser/ICaffeParser.hpp
• src/armnnCaffeParser/CaffeParser.cpp