Method LoadModel

tengine-module/plugin/serializer/mxnet/mxnet_serializer.cpp:452–485

}

bool MxnetSerializer::LoadModel(const std::vector<std::string>& file_list, StaticGraph* graph)
{
    if(file_list.size() != GetFileNum())
        return false;

    std::vector<MxnetNode> nodelist;
    if(!LoadTextFile(file_list[0].c_str(), nodelist))
    {
        LOG_ERROR() << "Parse text file " << file_list[0].c_str() << " failed\n";
        return false;
    }

    std::vector<MxnetParam> paramlist;
    if(!LoadBinaryFile(file_list[1].c_str(), paramlist))
    {
        LOG_ERROR() << "Parse binary file " << file_list[1].c_str() << " failed\n";
        return false;
    }

    SetGraphSource(graph, file_list[1]);
    SetGraphSourceFormat(graph, "mxnet");
    SetGraphConstTensorFile(graph, file_list[1]);
    SetGraphLayout(graph, TENGINE_LAYOUT_NCHW);
    SetModelLayout(graph, TENGINE_LAYOUT_NCHW);
    SetModelFormat(graph, MODEL_FORMAT_MXNET);

    bool res = LoadGraph(graph, nodelist, paramlist);
    for(std::size_t ii = 0; ii < paramlist.size(); ++ii)
    {
        std::free(paramlist[ii].raw_data);
    }

    return res;
}

bool MxnetSerializer::LoadConstTensor(StaticGraph* graph, const std::vector<MxnetNode>& nodelist,
                                      const std::vector<MxnetParam>& paramlist)