Method LoadModelFromMem

serializer/tengine/v1/tm1_serializer.cpp:541–573

}

bool TmSerializer1::LoadModelFromMem(void* mmap_buf, StaticGraph* graph)
{
    const TM_Header* tm_header = reinterpret_cast<const TM_Header*>(mmap_buf);
    /* Check the version of tm file format */
    if(tm_header->ver_main != TM_FILE_VER_MAIN || tm_header->ver_sub != TM_FILE_VER_SUB ||
       tm_header->ver_compile != TM_FILE_VER_COMPILE)
    {
        LOG_ERROR() << "Wrong version of tm file\n";
        return false;
    }

    const TM_Model* tm_model = GetTmPtr<TM_Model>(mmap_buf, tm_header->offset_root);
    if(tm_model->offset_s_mname == NOT_SET)
    {
        SetGraphIdentity(graph, "tengine", "tengine_model", "0");
    }
    else
    {
        std::string tm_model_name;
        const TM_String* tm_string = GetTmPtr<TM_String>(mmap_buf, tm_model->offset_s_mname);
        tm_model_name.assign(GetTmPtr<char>(mmap_buf, tm_string->offset_data), tm_string->size);
        SetGraphIdentity(graph, "tengine", tm_model_name, "0");
    }

    SetModelFormat(graph, MODEL_FORMAT_TENGINE);
    SetGraphLayout(graph, TENGINE_LAYOUT_NCHW);
    SetModelLayout(graph, TENGINE_LAYOUT_NCHW);

    if(LoadGraph(graph, tm_model, mmap_buf))
        return true;
    else
        return false;
}

bool TmSerializerRegisterOpLoader1(void)
{