CaffeParser Class Reference

#include <CaffeParser.hpp>

Inheritance diagram for CaffeParser:

Public Member Functions
virtual armnn::INetworkPtr	CreateNetworkFromBinaryFile (const char *graphFile, const std::map< std::string, armnn::TensorShape > &inputShapes, const std::vector< std::string > &requestedOutputs) override
	Create the network from a protobuf binary file on disk. More...
	CaffeParser ()
Public Member Functions inherited from CaffeParserBase
virtual armnn::INetworkPtr	CreateNetworkFromTextFile (const char *graphFile, const std::map< std::string, armnn::TensorShape > &inputShapes, const std::vector< std::string > &requestedOutputs) override
	Create the network from a protobuf text file on disk. More...
virtual armnn::INetworkPtr	CreateNetworkFromString (const char *protoText, const std::map< std::string, armnn::TensorShape > &inputShapes, const std::vector< std::string > &requestedOutputs) override
	Creates the network directly from protobuf text in a string. Useful for debugging/testing. More...
virtual BindingPointInfo	GetNetworkInputBindingInfo (const std::string &name) const override
	Retrieves 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 override
	Retrieves binding info (layer id and tensor info) for the network output identified by the given layer name. More...
	CaffeParserBase ()

Additional Inherited Members
Static Public Member Functions inherited from ICaffeParser
static ICaffeParser *	CreateRaw ()
static ICaffeParserPtr	Create ()
static void	Destroy (ICaffeParser *parser)
Protected Types inherited from CaffeParserBase
using	OperationParsingFunction = void(CaffeParserBase::*)(const caffe::LayerParameter &layerParam)
Protected Member Functions inherited from CaffeParserBase
armnn::TensorInfo	BlobShapeToTensorInfo (const caffe::BlobShape &blobShape) const
	Converts Caffe's protobuf tensor shape format to ArmNN's. More...
void	TrackInputBinding (armnn::IConnectableLayer *layer, armnn::LayerBindingId id, const armnn::TensorInfo &tensorInfo)
void	TrackOutputBinding (armnn::IConnectableLayer *layer, armnn::LayerBindingId id, const armnn::TensorInfo &tensorInfo)
void	SetArmnnOutputSlotForCaffeTop (const std::string &caffeTopName, armnn::IOutputSlot &armnnOutputSlot)
armnn::IOutputSlot &	GetArmnnOutputSlotForCaffeTop (const std::string &caffeTopName) const
	Retrieves the Armnn IOutputSlot representing the given Caffe top. More...
void	Cleanup ()
armnn::INetworkPtr	CreateNetworkFromNetParameter (caffe::NetParameter &netParam, const std::map< std::string, armnn::TensorShape > &inputShapes, const std::vector< std::string > &requestedOutputs)
	Parses a NetParameter loaded into memory from one of the other CreateNetwork*. More...
void	LoadNetParam (caffe::NetParameter &netParameter)
	does the actual conversion from caffe::NetParameter to armnn::INetwork More...
std::vector< const caffe::LayerParameter * >	GetInputs (const caffe::LayerParameter &layerParam)
	Find the Caffe layers listed as inputs (bottoms) for a given layer. More...
void	ResolveInPlaceLayers (caffe::NetParameter &netParameter)
	Modifies the Caffe network to replace "in-place" layers (whose top() and bottom() are both the same) with regular layers. More...
void	ParseInputLayer (const caffe::LayerParameter &layerParam)
	Adds an armnn layer to m_Network given a Caffe LayerParameter of the correct type and is responsible for recording any newly created IOutputSlots using SetArmnnOutputSlotForCaffeTop(). More...
void	ParseConvLayer (const caffe::LayerParameter &layerParam)
void	ParsePoolingLayer (const caffe::LayerParameter &layerParam)
void	ParseReluLayer (const caffe::LayerParameter &layerParam)
void	ParseLRNLayer (const caffe::LayerParameter &layerParam)
void	ParseInnerProductLayer (const caffe::LayerParameter &layerParam)
void	ParseSoftmaxLayer (const caffe::LayerParameter &layerParam)
void	ParseEltwiseLayer (const caffe::LayerParameter &layerParam)
void	ParseConcatLayer (const caffe::LayerParameter &layerParam)
void	ParseBatchNormLayer (const caffe::LayerParameter &layerParam)
void	ParseScaleLayer (const caffe::LayerParameter &layerParam)
void	ParseSplitLayer (const caffe::LayerParameter &layerParam)
void	ParseDropoutLayer (const caffe::LayerParameter &layerParam)
void	AddConvLayerWithSplits (const caffe::LayerParameter &layerParam, const armnn::Convolution2dDescriptor &desc, unsigned int kernelW, unsigned int kernelH)
	ParseConv may use these helpers depending on the group parameter. More...
void	AddConvLayerWithDepthwiseConv (const caffe::LayerParameter &layerParam, const armnn::Convolution2dDescriptor &desc, unsigned int kernelW, unsigned int kernelH)
Protected Member Functions inherited from ICaffeParser
virtual	~ICaffeParser ()
Static Protected Member Functions inherited from CaffeParserBase
static void	TrackBindingPoint (armnn::IConnectableLayer *layer, armnn::LayerBindingId id, const armnn::TensorInfo &tensorInfo, const char *bindingPointDesc, std::unordered_map< std::string, BindingPointInfo > &nameToBindingInfo)
static std::pair< armnn::LayerBindingId, armnn::TensorInfo >	GetBindingInfo (const std::string &layerName, const char *bindingPointDesc, const std::unordered_map< std::string, BindingPointInfo > &bindingInfos)
Protected Attributes inherited from CaffeParserBase
std::unordered_map< std::string, BindingPointInfo >	m_NetworkInputsBindingInfo
	maps input layer names to their corresponding ids and tensor infos More...
std::unordered_map< std::string, BindingPointInfo >	m_NetworkOutputsBindingInfo
	maps output layer names to their corresponding ids and tensor infos More...
armnn::INetworkPtr	m_Network
std::map< std::string, armnn::TensorShape >	m_InputShapes
std::unordered_map< std::string, armnn::IOutputSlot * >	m_ArmnnOutputSlotForCaffeTop
	As we add armnn layers we store the armnn IOutputSlot which corresponds to the Caffe tops. More...
std::vector< std::string >	m_RequestedOutputs
std::map< std::string, const caffe::LayerParameter * >	m_CaffeLayersByTopName
Static Protected Attributes inherited from CaffeParserBase
static const std::map< std::string, OperationParsingFunction >	ms_CaffeLayerNameToParsingFunctions
	Maps Caffe layer names to parsing member functions. More...

Detailed Description

Definition at line 159 of file CaffeParser.hpp.

Constructor & Destructor Documentation

CaffeParser()

CaffeParser

(

)

Definition at line 273 of file CaffeParser.cpp.

: CaffeParserBase()
{

}

Member Function Documentation

CreateNetworkFromBinaryFile()

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

overridevirtual

Create the network from a protobuf binary file on disk.

Implements ICaffeParser.

Definition at line 1715 of file CaffeParser.cpp.

References CHECK_LOCATION, and CaffeParserBase::CreateNetworkFromNetParameter().

{
    FILE* fd = fopen(graphFile, "rb");

    if (fd == nullptr)
    {
        throw FileNotFoundException(
            fmt::format("Failed to open graph file at: {} {}",
                        graphFile,
                        CHECK_LOCATION().AsString()));
    }

    // Parses the file into a message.
    NetParameter netParam;

    FileInputStream  inStream(fileno(fd));
    CodedInputStream codedStream(&inStream);
    codedStream.SetTotalBytesLimit(INT_MAX);
    bool success = netParam.ParseFromCodedStream(&codedStream);
    fclose(fd);

    if (!success)
    {
        throw ParseException(
            fmt::format("Failed to parse protobuf file: {} {}",
                        graphFile,
                        CHECK_LOCATION().AsString()));
    }

    return CreateNetworkFromNetParameter(netParam, inputShapes, requestedOutputs);
}

---

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

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