Method LoadGraph

tengine-module/plugin/serializer/mxnet/mxnet_serializer.cpp:636–701

}

bool MxnetSerializer::LoadGraph(StaticGraph* graph, const std::vector<MxnetNode>& nodelist,
                                const std::vector<MxnetParam>& paramlist)
{
    SetGraphIdentity(graph, "mxnet", graph->model_name, "0");

    LoadConstTensor(graph, nodelist, paramlist);
    CreateInputNode(graph, nodelist, paramlist);

    unsigned int i;
    std::vector<std::string> no_supported_op;
    for(i = 0; i < nodelist.size(); i++)
    {
        MxnetNode mxnet_node = nodelist.at(i);
        if(mxnet_node.op == "null" || mxnet_node.op == "_zeros")
            continue;
        if(!FindOpLoadMethod(mxnet_node.op))
        {
            auto it = find(no_supported_op.begin(),no_supported_op.end(),mxnet_node.op);
            if(it != no_supported_op.end())
                no_supported_op.push_back(mxnet_node.op);
        }
    }
    if(no_supported_op.size())
    {
        LOG_ERROR() << "These" <<no_supported_op.size() <<"ops are not supported \n";
        LOG_ERROR() << "{";
        for(int j = 0; j < (int)no_supported_op.size(); j++)
        {
            LOG_ERROR() << no_supported_op[j] <<",";
        }
        LOG_ERROR() << "}\n";
        return false;
    }   
   
    for(i = 0; i < nodelist.size(); i++)
    {
        MxnetNode mxnet_node = nodelist.at(i);

        if(mxnet_node.op == "null" || mxnet_node.op == "_zeros")
            continue;

       // if(!FindOpLoadMethod(mxnet_node.op))
       // {
       //     LOG_ERROR() << "Cannot find load function for operator: " << mxnet_node.op << "\n";
       //     break;
       // }

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

        LoadNode(graph, node, mxnet_node, nodelist);

        op_load_t op_func = any_cast<op_load_t>(GetOpLoadMethod(mxnet_node.op));

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

    if(i < nodelist.size())