Method LoadNode

tengine-module/plugin/serializer/caffe/caffe_serializer.cpp:235–271

}

bool CaffeSingle::LoadNode(StaticGraph* graph, StaticNode* node, const te_caffe::LayerParameter& layer_param,
                           name_map_t& tensor_name_map)
{
    for(int i = 0; i < layer_param.bottom_size(); i++)
    {
        const std::string& orig_name = layer_param.bottom(i);

        std::string& tensor_name = tensor_name_map[orig_name];

        StaticTensor* tensor = FindTensor(graph, tensor_name);

        AddNodeInputTensor(node, tensor);
    }

    for(int i = 0; i < layer_param.top_size(); i++)
    {
        const std::string& orig_name = layer_param.top(i);
        std::string tensor_name;

        if(tensor_name_map.count(orig_name))
            tensor_name = GetNodeName(node) + "/" + std::to_string(i);
        else
            tensor_name = orig_name;

        StaticTensor* tensor = CreateStaticTensor(graph, tensor_name);

        SetTensorDataType(tensor, DataType::GetTypeID("float32"));

        AddNodeOutputTensor(node, tensor);

        // record the name mapping

        tensor_name_map[orig_name] = tensor_name;
    }

    return true;
}

bool CaffeSingle::LoadGraph(te_caffe::NetParameter& caffe_net, StaticGraph* graph)
{