Method SaveModelIntoMem

serializer/tengine/v2/tm2_serializer.cpp:416–465

}

bool TmSerializer2::SaveModelIntoMem(void* start_ptr, Graph* graph, uint32_t* tm_model_size)
{
    bool tm_with_string = IsSaveString();

    tm_uoffset_t cur_pos = sizeof(TM2_Header);

    /* Define the TM2_Header object */
    TM2_Header header;
    header.ver_main = TM2_FILE_VER_MAIN;
    header.ver_sub = TM2_FILE_VER_SUB;
    header.ver_compile = TM2_FILE_VER_COMPILE;

    /* Define the TM2_Model object */
    TM2_Model tm_model;
    tm_model.orig_format = graph->GetModelFormat();
    tm_model.sub_format = 0;

    if(tm_with_string)
    {
        const std::string& fname = graph->GetName();
        TM2_String model_name;
        model_name.size = fname.size() + 1;    // including trailing \0
        model_name.offset_data = WriteTmFileAlign1(start_ptr, &cur_pos, fname.c_str(), model_name.size);
        tm_model.offset_s_mname = WriteTmObject(start_ptr, &cur_pos, &model_name, sizeof(TM2_String));
    }
    else
        tm_model.offset_s_mname = TM2_NOT_SET;

    /* Write the subgraphs */
    /* Only 1 subgraph is supported currently */
    size_t vector_size = sizeof(tm_size_t) + sizeof(tm_uoffset_t) * 1;
    TM2_Vector_offsets* v_subgraphs = ( TM2_Vector_offsets* )malloc(vector_size);
    v_subgraphs->v_num = 1;
    v_subgraphs->offsets[0] = SaveTmSubgraph(start_ptr, &cur_pos, graph);

    /* Write the vector of subgraphs */
    tm_model.offset_vo_subgraphs = WriteTmObject(start_ptr, &cur_pos, v_subgraphs, vector_size);

    /* Write the model */
    header.offset_root = WriteTmObject(start_ptr, &cur_pos, &tm_model, sizeof(TM2_Model));
    *tm_model_size = cur_pos;

    /* Write the header */
    cur_pos = 0;
    WriteTmObject(start_ptr, &cur_pos, &header, sizeof(TM2_Header));

    free(v_subgraphs);

    return true;
}

bool TmSerializer2::LoadNode(StaticGraph* graph, StaticNode* node, const TM2_Node* tm_node, void* mmap_buf)
{