12 #include "caffe/proto/caffe.pb.h" 21 class NetParameterInfo;
22 class LayerParameterInfo;
31 const char* graphFile,
32 const std::map<std::string, armnn::TensorShape>& inputShapes,
33 const std::vector<std::string>& requestedOutputs)
override;
38 void ProcessLayers(
const NetParameterInfo& netParameterInfo,
39 std::vector<LayerParameterInfo>& layerInfo,
40 const std::vector<std::string>& m_RequestedOutputs,
41 std::vector<const LayerParameterInfo*>& sortedNodes);
43 std::vector<const LayerParameterInfo *>& sortedNodes,
44 const NetParameterInfo& netParameterInfo);
45 std::vector<const LayerParameterInfo*> GetInputs(
46 const LayerParameterInfo& layerParam);
48 std::map<std::string, const LayerParameterInfo*> m_CaffeLayersByTopName;
49 std::vector<std::string> m_RequestedOutputs;
Caffe networks are loaded from protobuf files (binary or text) using the protobuf library and the gen...
RecordByRecordCaffeParser()
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.
std::unique_ptr< INetwork, void(*)(INetwork *network)> INetworkPtr