ArmNN
NotReleased
|
#include <CaffeParser.hpp>
Public Member Functions | |
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 () | |
Public Member Functions inherited from ICaffeParser | |
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... | |
Protected Types | |
using | OperationParsingFunction = void(CaffeParserBase::*)(const caffe::LayerParameter &layerParam) |
Protected Member Functions | |
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 |
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) |
void | ParseInputLayer (const caffe::LayerParameter &layerParam) |
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) |
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 | |
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 | |
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 | |
static const std::map< std::string, OperationParsingFunction > | ms_CaffeLayerNameToParsingFunctions |
Maps Caffe layer names to parsing member functions. More... | |
Additional Inherited Members | |
Static Public Member Functions inherited from ICaffeParser | |
static ICaffeParser * | CreateRaw () |
static ICaffeParserPtr | Create () |
static void | Destroy (ICaffeParser *parser) |
Definition at line 26 of file CaffeParser.hpp.
|
protected |
Definition at line 115 of file CaffeParser.hpp.
CaffeParserBase | ( | ) |
Definition at line 274 of file CaffeParser.cpp.
|
protected |
Definition at line 611 of file CaffeParser.cpp.
References CaffeParserBase::BlobShapeToTensorInfo(), CHECK_LOCATION, IOutputSlot::Connect(), CaffeParserBase::GetArmnnOutputSlotForCaffeTop(), armnnUtils::GetTensorInfo(), Convolution2dDescriptor::m_BiasEnabled, CaffeParserBase::m_Network, Convolution2dDescriptor::m_PadBottom, Convolution2dDescriptor::m_PadLeft, DepthwiseConvolution2dDescriptor::m_PadLeft, Convolution2dDescriptor::m_PadRight, Convolution2dDescriptor::m_PadTop, Convolution2dDescriptor::m_StrideX, Convolution2dDescriptor::m_StrideY, CaffeParserBase::SetArmnnOutputSlotForCaffeTop(), and armnnCaffeParser::TensorDescToBlobShape().
Referenced by CaffeParserBase::ParseConvLayer().
|
protected |
ParseConv may use these helpers depending on the group parameter
Definition at line 419 of file CaffeParser.cpp.
References CaffeParserBase::BlobShapeToTensorInfo(), CHECK_LOCATION, IOutputSlot::Connect(), CaffeParserBase::GetArmnnOutputSlotForCaffeTop(), IConnectableLayer::GetInputSlot(), IConnectableLayer::GetNumOutputSlots(), IConnectableLayer::GetOutputSlot(), armnnUtils::GetTensorInfo(), Convolution2dDescriptor::m_BiasEnabled, CaffeParserBase::m_Network, Convolution2dDescriptor::m_PadBottom, Convolution2dDescriptor::m_PadRight, Convolution2dDescriptor::m_StrideX, Convolution2dDescriptor::m_StrideY, CaffeParserBase::SetArmnnOutputSlotForCaffeTop(), IOutputSlot::SetTensorInfo(), OriginsDescriptor::SetViewOriginCoord(), ViewsDescriptor::SetViewOriginCoord(), ViewsDescriptor::SetViewSize(), and armnnCaffeParser::TensorDescToBlobShape().
Referenced by CaffeParserBase::ParseConvLayer().
|
protected |
Converts Caffe's protobuf tensor shape format to ArmNN's.
Definition at line 314 of file CaffeParser.cpp.
Referenced by CaffeParserBase::AddConvLayerWithDepthwiseConv(), CaffeParserBase::AddConvLayerWithSplits(), CaffeParserBase::ParseConvLayer(), and CaffeParserBase::ParseInputLayer().
|
protected |
Definition at line 1860 of file CaffeParser.cpp.
References CaffeParserBase::m_ArmnnOutputSlotForCaffeTop, CaffeParserBase::m_CaffeLayersByTopName, CaffeParserBase::m_InputShapes, and CaffeParserBase::m_RequestedOutputs.
Referenced by RecordByRecordCaffeParser::CreateNetworkFromBinaryFile(), and CaffeParserBase::CreateNetworkFromNetParameter().
|
protected |
Parses a NetParameter loaded into memory from one of the other CreateNetwork*.
Definition at line 1829 of file CaffeParser.cpp.
References CaffeParserBase::Cleanup(), INetwork::Create(), CaffeParserBase::LoadNetParam(), CaffeParserBase::m_InputShapes, CaffeParserBase::m_Network, CaffeParserBase::m_NetworkInputsBindingInfo, CaffeParserBase::m_NetworkOutputsBindingInfo, and CaffeParserBase::m_RequestedOutputs.
Referenced by CaffeParser::CreateNetworkFromBinaryFile(), CaffeParserBase::CreateNetworkFromString(), and CaffeParserBase::CreateNetworkFromTextFile().
|
overridevirtual |
Creates the network directly from protobuf text in a string. Useful for debugging/testing.
Implements ICaffeParser.
Definition at line 1769 of file CaffeParser.cpp.
References CHECK_LOCATION, and CaffeParserBase::CreateNetworkFromNetParameter().
|
overridevirtual |
Create the network from a protobuf text file on disk.
Implements ICaffeParser.
Definition at line 1733 of file CaffeParser.cpp.
References CHECK_LOCATION, and CaffeParserBase::CreateNetworkFromNetParameter().
|
protected |
Retrieves the Armnn IOutputSlot representing the given Caffe top. Throws if it cannot be found (e.g. not parsed yet).
Definition at line 1533 of file CaffeParser.cpp.
References CHECK_LOCATION, and CaffeParserBase::m_ArmnnOutputSlotForCaffeTop.
Referenced by CaffeParserBase::AddConvLayerWithDepthwiseConv(), CaffeParserBase::AddConvLayerWithSplits(), RecordByRecordCaffeParser::CreateNetworkFromBinaryFile(), CaffeParserBase::LoadNetParam(), CaffeParserBase::ParseBatchNormLayer(), CaffeParserBase::ParseConcatLayer(), CaffeParserBase::ParseConvLayer(), CaffeParserBase::ParseDropoutLayer(), CaffeParserBase::ParseEltwiseLayer(), CaffeParserBase::ParseInnerProductLayer(), CaffeParserBase::ParseLRNLayer(), CaffeParserBase::ParsePoolingLayer(), CaffeParserBase::ParseReluLayer(), CaffeParserBase::ParseScaleLayer(), CaffeParserBase::ParseSoftmaxLayer(), and CaffeParserBase::ParseSplitLayer().
|
staticprotected |
Definition at line 296 of file CaffeParser.cpp.
References CHECK_LOCATION.
Referenced by CaffeParserBase::GetNetworkInputBindingInfo(), and CaffeParserBase::GetNetworkOutputBindingInfo().
|
protected |
Find the Caffe layers listed as inputs (bottoms) for a given layer.
Definition at line 339 of file CaffeParser.cpp.
References CHECK_LOCATION, and CaffeParserBase::m_CaffeLayersByTopName.
Referenced by CaffeParserBase::LoadNetParam().
|
overridevirtual |
Retrieves binding info (layer id and tensor info) for the network input identified by the given layer name.
Implements ICaffeParser.
Definition at line 286 of file CaffeParser.cpp.
References CaffeParserBase::GetBindingInfo(), and CaffeParserBase::m_NetworkInputsBindingInfo.
|
overridevirtual |
Retrieves binding info (layer id and tensor info) for the network output identified by the given layer name.
Implements ICaffeParser.
Definition at line 291 of file CaffeParser.cpp.
References CaffeParserBase::GetBindingInfo(), and CaffeParserBase::m_NetworkOutputsBindingInfo.
|
protected |
does the actual conversion from caffe::NetParameter to armnn::INetwork
Definition at line 1632 of file CaffeParser.cpp.
References CHECK_LOCATION, IOutputSlot::Connect(), CaffeParserBase::GetArmnnOutputSlotForCaffeTop(), CaffeParserBase::GetInputs(), CaffeParserBase::m_CaffeLayersByTopName, CaffeParserBase::m_Network, CaffeParserBase::m_NetworkOutputsBindingInfo, CaffeParserBase::m_RequestedOutputs, CaffeParserBase::ms_CaffeLayerNameToParsingFunctions, CaffeParserBase::ResolveInPlaceLayers(), and CaffeParserBase::TrackOutputBinding().
Referenced by CaffeParserBase::CreateNetworkFromNetParameter().
|
protected |
Definition at line 1338 of file CaffeParser.cpp.
References CHECK_LOCATION, Connect(), armnn::Float32, CaffeParserBase::GetArmnnOutputSlotForCaffeTop(), IConnectableLayer::GetInputSlot(), IConnectableLayer::GetOutputSlot(), TensorInfo::GetShape(), armnnUtils::GetTensorInfo(), BatchNormalizationDescriptor::m_Eps, CaffeParserBase::m_Network, CaffeParserBase::SetArmnnOutputSlotForCaffeTop(), and IOutputSlot::SetTensorInfo().
|
protected |
Definition at line 1277 of file CaffeParser.cpp.
References CHECK_LOCATION, IOutputSlot::Connect(), CaffeParserBase::GetArmnnOutputSlotForCaffeTop(), IConnectableLayer::GetInputSlot(), TensorInfo::GetNumDimensions(), IConnectableLayer::GetOutputSlot(), TensorInfo::GetShape(), armnnUtils::GetTensorInfo(), CaffeParserBase::m_Network, CaffeParserBase::SetArmnnOutputSlotForCaffeTop(), IOutputSlot::SetTensorInfo(), and OriginsDescriptor::SetViewOriginCoord().
|
protected |
Definition at line 701 of file CaffeParser.cpp.
References CaffeParserBase::AddConvLayerWithDepthwiseConv(), CaffeParserBase::AddConvLayerWithSplits(), CaffeParserBase::BlobShapeToTensorInfo(), CHECK_LOCATION, IOutputSlot::Connect(), GET_OPTIONAL_WITH_VECTOR_FALLBACK, CaffeParserBase::GetArmnnOutputSlotForCaffeTop(), armnnUtils::GetTensorInfo(), Convolution2dDescriptor::m_BiasEnabled, CaffeParserBase::m_Network, Convolution2dDescriptor::m_PadBottom, Convolution2dDescriptor::m_PadLeft, Convolution2dDescriptor::m_PadRight, Convolution2dDescriptor::m_PadTop, Convolution2dDescriptor::m_StrideX, Convolution2dDescriptor::m_StrideY, CaffeParserBase::SetArmnnOutputSlotForCaffeTop(), and armnnCaffeParser::TensorDescToBlobShape().
|
protected |
Definition at line 1477 of file CaffeParser.cpp.
References CHECK_LOCATION, CaffeParserBase::GetArmnnOutputSlotForCaffeTop(), and CaffeParserBase::SetArmnnOutputSlotForCaffeTop().
|
protected |
Definition at line 1230 of file CaffeParser.cpp.
References CHECK_LOCATION, Connect(), CaffeParserBase::GetArmnnOutputSlotForCaffeTop(), IConnectableLayer::GetInputSlot(), IConnectableLayer::GetOutputSlot(), armnnUtils::GetTensorInfo(), CaffeParserBase::m_Network, CaffeParserBase::SetArmnnOutputSlotForCaffeTop(), and IOutputSlot::SetTensorInfo().
|
protected |
Definition at line 1134 of file CaffeParser.cpp.
References Connect(), CaffeParserBase::GetArmnnOutputSlotForCaffeTop(), IConnectableLayer::GetInputSlot(), TensorInfo::GetNumDimensions(), IConnectableLayer::GetOutputSlot(), TensorInfo::GetShape(), armnnUtils::GetTensorInfo(), FullyConnectedDescriptor::m_BiasEnabled, CaffeParserBase::m_Network, FullyConnectedDescriptor::m_TransposeWeightMatrix, CaffeParserBase::SetArmnnOutputSlotForCaffeTop(), and IOutputSlot::SetTensorInfo().
|
protected |
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().
Definition at line 364 of file CaffeParser.cpp.
References CaffeParserBase::BlobShapeToTensorInfo(), CHECK_LOCATION, CaffeParserBase::m_InputShapes, CaffeParserBase::m_Network, CaffeParserBase::m_NetworkInputsBindingInfo, CaffeParserBase::SetArmnnOutputSlotForCaffeTop(), TensorInfo::SetShape(), and CaffeParserBase::TrackInputBinding().
|
protected |
Definition at line 1027 of file CaffeParser.cpp.
References CHECK_LOCATION, Connect(), CaffeParserBase::GetArmnnOutputSlotForCaffeTop(), IConnectableLayer::GetInputSlot(), IConnectableLayer::GetOutputSlot(), armnnUtils::GetTensorInfo(), NormalizationDescriptor::m_Alpha, NormalizationDescriptor::m_Beta, NormalizationDescriptor::m_K, CaffeParserBase::m_Network, NormalizationDescriptor::m_NormChannelType, NormalizationDescriptor::m_NormMethodType, NormalizationDescriptor::m_NormSize, CaffeParserBase::SetArmnnOutputSlotForCaffeTop(), and IOutputSlot::SetTensorInfo().
|
protected |
Definition at line 883 of file CaffeParser.cpp.
References CHECK_LOCATION, Connect(), GET_OPTIONAL_WITH_FALLBACK, CaffeParserBase::GetArmnnOutputSlotForCaffeTop(), IConnectableLayer::GetInputSlot(), IConnectableLayer::GetOutputSlot(), TensorInfo::GetShape(), armnnUtils::GetTensorInfo(), CaffeParserBase::m_Network, Pooling2dDescriptor::m_OutputShapeRounding, Pooling2dDescriptor::m_PadBottom, Pooling2dDescriptor::m_PaddingMethod, Pooling2dDescriptor::m_PadLeft, Pooling2dDescriptor::m_PadRight, Pooling2dDescriptor::m_PadTop, Pooling2dDescriptor::m_PoolHeight, Pooling2dDescriptor::m_PoolType, Pooling2dDescriptor::m_PoolWidth, Pooling2dDescriptor::m_StrideX, Pooling2dDescriptor::m_StrideY, CaffeParserBase::SetArmnnOutputSlotForCaffeTop(), and IOutputSlot::SetTensorInfo().
|
protected |
Definition at line 1001 of file CaffeParser.cpp.
References Connect(), CaffeParserBase::GetArmnnOutputSlotForCaffeTop(), IConnectableLayer::GetInputSlot(), IConnectableLayer::GetOutputSlot(), armnnUtils::GetTensorInfo(), ActivationDescriptor::m_A, ActivationDescriptor::m_Function, CaffeParserBase::m_Network, CaffeParserBase::SetArmnnOutputSlotForCaffeTop(), and IOutputSlot::SetTensorInfo().
|
protected |
Definition at line 1404 of file CaffeParser.cpp.
References CHECK_LOCATION, Connect(), armnn::Float32, CaffeParserBase::GetArmnnOutputSlotForCaffeTop(), IConnectableLayer::GetInputSlot(), IConnectableLayer::GetOutputSlot(), TensorInfo::GetShape(), armnnUtils::GetTensorInfo(), BatchNormalizationDescriptor::m_Eps, CaffeParserBase::m_Network, CaffeParserBase::SetArmnnOutputSlotForCaffeTop(), and IOutputSlot::SetTensorInfo().
|
protected |
Definition at line 1208 of file CaffeParser.cpp.
References Connect(), CaffeParserBase::GetArmnnOutputSlotForCaffeTop(), IConnectableLayer::GetInputSlot(), IConnectableLayer::GetOutputSlot(), armnnUtils::GetTensorInfo(), SoftmaxDescriptor::m_Axis, CaffeParserBase::m_Network, CaffeParserBase::SetArmnnOutputSlotForCaffeTop(), and IOutputSlot::SetTensorInfo().
|
protected |
Definition at line 1456 of file CaffeParser.cpp.
References CHECK_LOCATION, CaffeParserBase::GetArmnnOutputSlotForCaffeTop(), and CaffeParserBase::SetArmnnOutputSlotForCaffeTop().
|
protected |
Modifies the Caffe network to replace "in-place" layers (whose top() and bottom() are both the same) with regular layers. This simplifies further parsing.
Definition at line 1572 of file CaffeParser.cpp.
References CHECK_LOCATION.
Referenced by RecordByRecordCaffeParser::CreateNetworkFromBinaryFile(), and CaffeParserBase::LoadNetParam().
|
protected |
Definition at line 1551 of file CaffeParser.cpp.
References CHECK_LOCATION, and CaffeParserBase::m_ArmnnOutputSlotForCaffeTop.
Referenced by CaffeParserBase::AddConvLayerWithDepthwiseConv(), CaffeParserBase::AddConvLayerWithSplits(), CaffeParserBase::ParseBatchNormLayer(), CaffeParserBase::ParseConcatLayer(), CaffeParserBase::ParseConvLayer(), CaffeParserBase::ParseDropoutLayer(), CaffeParserBase::ParseEltwiseLayer(), CaffeParserBase::ParseInnerProductLayer(), CaffeParserBase::ParseInputLayer(), CaffeParserBase::ParseLRNLayer(), CaffeParserBase::ParsePoolingLayer(), CaffeParserBase::ParseReluLayer(), CaffeParserBase::ParseScaleLayer(), CaffeParserBase::ParseSoftmaxLayer(), and CaffeParserBase::ParseSplitLayer().
|
staticprotected |
Definition at line 1509 of file CaffeParser.cpp.
References CHECK_LOCATION, and IConnectableLayer::GetName().
Referenced by CaffeParserBase::TrackInputBinding(), and CaffeParserBase::TrackOutputBinding().
|
protected |
Definition at line 1495 of file CaffeParser.cpp.
References IConnectableLayer::GetName(), CaffeParserBase::m_NetworkInputsBindingInfo, and CaffeParserBase::TrackBindingPoint().
Referenced by CaffeParserBase::ParseInputLayer().
|
protected |
Definition at line 1502 of file CaffeParser.cpp.
References IConnectableLayer::GetName(), CaffeParserBase::m_NetworkOutputsBindingInfo, and CaffeParserBase::TrackBindingPoint().
Referenced by RecordByRecordCaffeParser::CreateNetworkFromBinaryFile(), and CaffeParserBase::LoadNetParam().
|
protected |
As we add armnn layers we store the armnn IOutputSlot which corresponds to the Caffe tops.
Definition at line 131 of file CaffeParser.hpp.
Referenced by CaffeParserBase::Cleanup(), CaffeParserBase::GetArmnnOutputSlotForCaffeTop(), and CaffeParserBase::SetArmnnOutputSlotForCaffeTop().
|
protected |
Definition at line 139 of file CaffeParser.hpp.
Referenced by CaffeParserBase::Cleanup(), CaffeParserBase::GetInputs(), and CaffeParserBase::LoadNetParam().
|
protected |
Definition at line 128 of file CaffeParser.hpp.
Referenced by CaffeParserBase::Cleanup(), RecordByRecordCaffeParser::CreateNetworkFromBinaryFile(), CaffeParserBase::CreateNetworkFromNetParameter(), and CaffeParserBase::ParseInputLayer().
|
protected |
Definition at line 126 of file CaffeParser.hpp.
Referenced by CaffeParserBase::AddConvLayerWithDepthwiseConv(), CaffeParserBase::AddConvLayerWithSplits(), RecordByRecordCaffeParser::CreateNetworkFromBinaryFile(), CaffeParserBase::CreateNetworkFromNetParameter(), CaffeParserBase::LoadNetParam(), CaffeParserBase::ParseBatchNormLayer(), CaffeParserBase::ParseConcatLayer(), CaffeParserBase::ParseConvLayer(), CaffeParserBase::ParseEltwiseLayer(), CaffeParserBase::ParseInnerProductLayer(), CaffeParserBase::ParseInputLayer(), CaffeParserBase::ParseLRNLayer(), CaffeParserBase::ParsePoolingLayer(), CaffeParserBase::ParseReluLayer(), CaffeParserBase::ParseScaleLayer(), and CaffeParserBase::ParseSoftmaxLayer().
|
protected |
maps input layer names to their corresponding ids and tensor infos
Definition at line 121 of file CaffeParser.hpp.
Referenced by RecordByRecordCaffeParser::CreateNetworkFromBinaryFile(), CaffeParserBase::CreateNetworkFromNetParameter(), CaffeParserBase::GetNetworkInputBindingInfo(), CaffeParserBase::ParseInputLayer(), and CaffeParserBase::TrackInputBinding().
|
protected |
maps output layer names to their corresponding ids and tensor infos
Definition at line 124 of file CaffeParser.hpp.
Referenced by RecordByRecordCaffeParser::CreateNetworkFromBinaryFile(), CaffeParserBase::CreateNetworkFromNetParameter(), CaffeParserBase::GetNetworkOutputBindingInfo(), CaffeParserBase::LoadNetParam(), and CaffeParserBase::TrackOutputBinding().
|
protected |
Definition at line 133 of file CaffeParser.hpp.
Referenced by CaffeParserBase::Cleanup(), CaffeParserBase::CreateNetworkFromNetParameter(), and CaffeParserBase::LoadNetParam().
|
staticprotected |
Maps Caffe layer names to parsing member functions.
Definition at line 118 of file CaffeParser.hpp.
Referenced by RecordByRecordCaffeParser::CreateNetworkFromBinaryFile(), and CaffeParserBase::LoadNetParam().