Method SaveTmSubgraph

serializer/tengine/v2/tm2_serializer.cpp:303–414

}

tm_uoffset_t TmSerializer2::SaveTmSubgraph(void* const start_ptr, tm_uoffset_t* cur_pos, Graph* graph)
{
    TM2_Subgraph tm_subgraph;
    tm_subgraph.subgraph_id = 0; /* subgraph_id starts from 0 */
    tm_subgraph.offset_s_sname = TM2_NOT_SET;

    tm_subgraph.graph_layout = graph->GetLayout();
    tm_subgraph.model_layout = graph->GetModelLayout();

    unsigned int tensor_num = 0;
    unsigned int buffer_num = 0;
    std::vector<Tensor*> tensor_ptrs;
    std::vector<void*> buf_ptrs;
    std::vector<unsigned int> buf_sizes;
    name_map_t tensor_name_map; /* map of tensor name and tensor index */
    bool tm_no_data = !IsSaveData();

    /* Write the nodes */
    size_t vector_size = sizeof(tm_size_t) + sizeof(tm_uoffset_t) * graph->seq_nodes.size();
    TM2_Vector_offsets* v_nodes = ( TM2_Vector_offsets* )malloc(vector_size);
    v_nodes->v_num = graph->seq_nodes.size();
    for(unsigned int i = 0; i < graph->seq_nodes.size(); i++)
    {
        Node* p_node = graph->seq_nodes[i];
        for(unsigned int k = 0; k < p_node->GetOutputNum(); k++)
        {
            Tensor* p_tensor = p_node->GetOutputTensor(k);
            tensor_ptrs.push_back(p_tensor);
            tensor_name_map[p_tensor->GetName()] = tensor_num;
            tensor_num++;
        }
        v_nodes->offsets[i] = SaveTmNode(start_ptr, cur_pos, p_node, tensor_name_map);
    }
    /* Write the vector of nodes */
    tm_subgraph.offset_vo_seq_nodes = WriteTmObject(start_ptr, cur_pos, v_nodes, vector_size);

    /* Write the tensors */
    vector_size = sizeof(tm_size_t) + sizeof(tm_uoffset_t) * tensor_num;
    TM2_Vector_offsets* v_tensors = ( TM2_Vector_offsets* )malloc(vector_size);
    v_tensors->v_num = tensor_num;
    for(unsigned int i = 0; i < tensor_num; i++)
    {
        Tensor* p_tensor = tensor_ptrs[i];
        if(p_tensor->GetType() == kConstTensor)
        {
            buf_ptrs.push_back(p_tensor->GetMemAddr());
            buf_sizes.push_back(p_tensor->GetTotalSize());
            buffer_num++;
        }

        v_tensors->offsets[i] = SaveTmTensor(start_ptr, cur_pos, p_tensor, i, buffer_num - 1);
    }
    /* Write the vector of tensors */
    tm_subgraph.offset_vo_tensors = WriteTmObject(start_ptr, cur_pos, v_tensors, vector_size);

    /* Write the buffers */
    vector_size = sizeof(tm_size_t) + sizeof(tm_uoffset_t) * buffer_num;
    TM2_Vector_offsets* v_buffers = ( TM2_Vector_offsets* )malloc(vector_size);