Method LoadGraph

tengine-module/plugin/serializer/caffe/caffe_serializer.cpp:273–333

}

bool CaffeSingle::LoadGraph(te_caffe::NetParameter& caffe_net, StaticGraph* graph)
{
    SetGraphIdentity(graph, "caffe", caffe_net.name(), "0");

    name_map_t tensor_name_map;

    int layer_num = caffe_net.layer_size();
    int i;

    std::vector<std::string> no_supported_op;
    for(i =0; i < layer_num; i++)
    {
        const te_caffe::LayerParameter& layer_param = caffe_net.layer(i);
        const std::string& caffe_op_name = layer_param.type();

        if(!FindOpLoadMethod(caffe_op_name))
        {
            auto it = find(no_supported_op.begin(), no_supported_op.end(), caffe_op_name);
            if(it == no_supported_op.end())
                no_supported_op.push_back(caffe_op_name);
        }
    }
    if(no_supported_op.size() != 0)
    {
        LOG_ERROR() << "These "<< no_supported_op.size() << "ops are not supported:\n";
        LOG_ERROR() << "{";

        for(int j = 0 ; j < static_cast<int> (no_supported_op.size()); j++)
        {
            LOG_ERROR() << no_supported_op[j] << ",";
        }
        LOG_ERROR() << "}\n";
        return false;
    }
    for(i = 0; i < layer_num; i++)
    {
        const te_caffe::LayerParameter& layer_param = caffe_net.layer(i);
        const std::string& caffe_op_name = layer_param.type();

        //if(!FindOpLoadMethod(caffe_op_name))
        //{
        //    LOG_ERROR() << "cannot find load function for operator: " << caffe_op_name << "\n";
        //    break;
       // }

        StaticNode* node = CreateStaticNode(graph, layer_param.name());

        if(!LoadNode(graph, node, layer_param, tensor_name_map))
            break;

        op_load_t op_func = any_cast<op_load_t>(GetOpLoadMethod(caffe_op_name));

        if(!op_func(graph, node, layer_param))
            break;
    }

    if(i < layer_num)
        return false;