|
|
@ -568,7 +568,7 @@ private: |
|
|
|
typedef std::map<std::string, TFImporterNodeParser> DispatchMap; |
|
|
|
typedef std::map<std::string, TFImporterNodeParser> DispatchMap; |
|
|
|
|
|
|
|
|
|
|
|
const DispatchMap dispatch; |
|
|
|
const DispatchMap dispatch; |
|
|
|
static const DispatchMap buildDispatchMap(); |
|
|
|
static DispatchMap buildDispatchMap(); |
|
|
|
|
|
|
|
|
|
|
|
void parseConvolution (tensorflow::GraphDef& net, const tensorflow::NodeDef& layer, LayerParams& layerParams); |
|
|
|
void parseConvolution (tensorflow::GraphDef& net, const tensorflow::NodeDef& layer, LayerParams& layerParams); |
|
|
|
void parseBias (tensorflow::GraphDef& net, const tensorflow::NodeDef& layer, LayerParams& layerParams); |
|
|
|
void parseBias (tensorflow::GraphDef& net, const tensorflow::NodeDef& layer, LayerParams& layerParams); |
|
|
@ -645,7 +645,7 @@ protected: |
|
|
|
TFImporter* importer; |
|
|
|
TFImporter* importer; |
|
|
|
}; |
|
|
|
}; |
|
|
|
|
|
|
|
|
|
|
|
const TFImporter::DispatchMap TFImporter::buildDispatchMap() |
|
|
|
TFImporter::DispatchMap TFImporter::buildDispatchMap() |
|
|
|
{ |
|
|
|
{ |
|
|
|
static DispatchMap dispatch; |
|
|
|
static DispatchMap dispatch; |
|
|
|
dispatch["Conv2D"] = dispatch["SpaceToBatchND"] = dispatch["DepthwiseConv2dNative"] = |
|
|
|
dispatch["Conv2D"] = dispatch["SpaceToBatchND"] = dispatch["DepthwiseConv2dNative"] = |
|
|
|