22 using ICaffeParserPtr = std::unique_ptr<ICaffeParser, void(*)(ICaffeParser* parser)>;
33 const char* graphFile,
34 const std::map<std::string, armnn::TensorShape>& inputShapes,
35 const std::vector<std::string>& requestedOutputs);
39 const char* graphFile,
40 const std::map<std::string, armnn::TensorShape>& inputShapes,
41 const std::vector<std::string>& requestedOutputs);
45 const char* protoText,
46 const std::map<std::string, armnn::TensorShape>& inputShapes,
47 const std::vector<std::string>& requestedOutputs);
63 std::unique_ptr<CaffeParserImpl> pCaffeParserImpl;
static void Destroy(ICaffeParser *parser)
Caffe networks are loaded from protobuf files (binary or text) using the protobuf library and the gen...
armnn::INetworkPtr CreateNetworkFromBinaryFile(const char *graphFile, const std::map< std::string, armnn::TensorShape > &inputShapes, const std::vector< std::string > &requestedOutputs)
Create the network from a protobuf binary file on the disk.
armnn::INetworkPtr CreateNetworkFromTextFile(const char *graphFile, const std::map< std::string, armnn::TensorShape > &inputShapes, const std::vector< std::string > &requestedOutputs)
Create the network from a protobuf text file on the disk.
static ICaffeParserPtr Create()
std::unique_ptr< ICaffeParser, void(*)(ICaffeParser *parser)> ICaffeParserPtr
BindingPointInfo GetNetworkInputBindingInfo(const std::string &name) const
Retrieve binding info (layer id and tensor info) for the network input identified by the given layer ...
std::pair< armnn::LayerBindingId, armnn::TensorInfo > BindingPointInfo
static ICaffeParser * CreateRaw()
armnn::INetworkPtr CreateNetworkFromString(const char *protoText, const std::map< std::string, armnn::TensorShape > &inputShapes, const std::vector< std::string > &requestedOutputs)
Create the network directly from protobuf text in a string. Useful for debugging/testin.g.
BindingPointInfo GetNetworkOutputBindingInfo(const std::string &name) const
Retrieve binding info (layer id and tensor info) for the network output identified by the given layer...
std::unique_ptr< INetwork, void(*)(INetwork *network)> INetworkPtr