Method LoadNode

serializer/tengine/v2/tm2_serializer.cpp:467–529

}

bool TmSerializer2::LoadNode(StaticGraph* graph, StaticNode* node, const TM2_Node* tm_node, void* mmap_buf)
{
    if(tm_node->offset_vi_input_tensors != TM2_NOT_SET)
    {
        const TM2_Vector_indices* v_input_tensors =
            GetTmPtr<TM2_Vector_indices>(mmap_buf, tm_node->offset_vi_input_tensors);

        /* Set the input tensors to the node */
        for(unsigned int i = 0; i < v_input_tensors->v_num; i++)
        {
            StaticTensor* tensor = graph->tensor_list[v_input_tensors->indices[i]].get();
            if(!tensor)
            {
                LOG_ERROR() << "The input tensor not exist: " << v_input_tensors->indices[i] << "\n";
                return false;
            }
            AddNodeInputTensor(node, tensor);
        }
    }

    if(tm_node->offset_vi_output_tensors != TM2_NOT_SET)
    {
        const TM2_Vector_indices* v_output_tensors =
            GetTmPtr<TM2_Vector_indices>(mmap_buf, tm_node->offset_vi_output_tensors);

        /* Set the output tensors to the node */
        for(unsigned int i = 0; i < v_output_tensors->v_num; i++)
        {
            StaticTensor* tensor = graph->tensor_list[v_output_tensors->indices[i]].get();
            if(!tensor)
            {
                LOG_ERROR() << "The output tensor not exist: " << v_output_tensors->indices[i] << "\n";
                return false;
            }
            AddNodeOutputTensor(node, tensor);
        }
    }

    /* set the custom attributes into static node */
    if(tm_node->offset_vo_attrs == TM2_NOT_SET)
        return true;

    const TM2_Vector_offsets* v_attrs = GetTmPtr<TM2_Vector_offsets>(mmap_buf, tm_node->offset_vo_attrs);
    for(unsigned int i = 0; i < v_attrs->v_num; i++)
    {
        const TM2_Attr* tm_attr = GetTmPtr<TM2_Attr>(mmap_buf, v_attrs->offsets[i]);
        const TM2_String* tm_attr_name = GetTmPtr<TM2_String>(mmap_buf, tm_attr->offset_s_attrname);
        const TM2_String* tm_attr_val = GetTmPtr<TM2_String>(mmap_buf, tm_attr->offset_s_attrval);

        const char* attr_name = GetTmPtr<char>(mmap_buf, tm_attr_name->offset_data);
        const char* attr_val = GetTmPtr<char>(mmap_buf, tm_attr_val->offset_data);
        const char* type_name = int_to_typename(tm_attr->attr_type);

        if(NodeAddParamGeneric(node, attr_name, type_name, tm_attr_val->size) < 0 ||
           NodeSetParamGeneric(node, attr_name, type_name, attr_val, tm_attr_val->size) < 0)
        {
            LOG_ERROR() << "Add and set node param failed\n";
            return false;