Method LoadTensor

serializer/tengine/v2/tm2_serializer.cpp:531–610

}

bool TmSerializer2::LoadTensor(StaticGraph* graph, const TM2_Tensor* tm_tensor, const TM2_Buffer* tm_buf,
                               void* mmap_buf)
{
    /* Set the tensor name */
    int idx = tm_tensor->tensor_id;
    std::string tm_tensor_name;
    if(tm_tensor->offset_s_tname == TM2_NOT_SET)
        tm_tensor_name = "tensor_" + std::to_string(idx);
    else
    {
        const TM2_String* tm_str = GetTmPtr<TM2_String>(mmap_buf, tm_tensor->offset_s_tname);
        tm_tensor_name.assign(GetTmPtr<char>(mmap_buf, tm_str->offset_data), tm_str->size - 1);
    }

    /* Create the static tensor */
    StaticTensor* tensor;
    if(tm_tensor->type == kConstTensor)
        tensor = CreateStaticConstTensor(graph, tm_tensor_name);
    else
        tensor = CreateStaticTensor(graph, tm_tensor_name);
    if(!tensor)
    {
        LOG_ERROR() << "Create static const tensor failed: " << tm_tensor_name << "\n";
        return false;
    }

    /* Set the dims */
    if(tm_tensor->offset_vd_dims != TM2_NOT_SET)
    {
        const TM2_Vector_dims* v_dims = GetTmPtr<TM2_Vector_dims>(mmap_buf, tm_tensor->offset_vd_dims);
        if(!v_dims || !(v_dims->v_num))
        {
            LOG_ERROR() << "Get tensor dims failed\n";
            return false;
        }
        std::vector<int> dims;
        for(unsigned int i = 0; i < v_dims->v_num; i++)
            dims.push_back(v_dims->dims[i]);
        SetTensorDim(tensor, dims);
    }

    /* Set the tensor type and the data type */
    SetTensorType(tensor, tm_tensor->type);
    SetTensorDataType(tensor, tm_tensor->data_type);

    /* Set the memory size and pointer */
    if(tm_tensor->type == kConstTensor)
    {
        SetTensorSize(tensor, tm_buf->size);
        void* buf = malloc(tm_buf->size + 128);
        if(tm_buf->offset_data != TM2_NOT_SET)
        {
            memcpy(buf, GetTmPtr<void>(mmap_buf, tm_buf->offset_data), tm_buf->size);
        }

        SetConstTensorBuffer(tensor, buf);
        SetConstTensorFileLocation(tensor, -1, 0);
    }