Function parse_tensor

runtime/src/parse.c:109–154

}

TinyNNStatus parse_tensor(
        Tensor* tensor, ns(Tensor_table_t) fbs_tensor, int tensor_id) {
    //! dtype
    ns(DType_table_t) fbs_dtype = ns(Tensor_dtype(fbs_tensor));
    tensor->dtype.type_enum = dtype_from_fbs(ns(DType_type(fbs_dtype)));
    tensor->dtype.param = dtype_param_from_fbs(fbs_dtype);

    //! name
    const char* name = ns(Tensor_name(fbs_tensor));
    if (strlen(name) > 0) {
        tensor->name = get_string(name);
    } else {
        char name[20] = "tensor:";
        sprintf(name, "tensor:%d", tensor_id);
        tensor->name = get_string(name);
    }
    //! use_count
    tensor->use_count = ns(Tensor_use_count(fbs_tensor));
    tensor->is_dynamic = ns(Tensor_dynamic(fbs_tensor));
    //! offset
    tensor->offset = ns(Tensor_offset(fbs_tensor));
    LOG_DEBUG(
            "Tensor info: name=%s, use_count=%d, offset=%zu, is_dynamic=%d\n",
            tensor->name, tensor->use_count, tensor->offset, tensor->is_dynamic);
    //! layout
    ns(Layout_table_t) fbs_layout = ns(Tensor_layout(fbs_tensor));
    flatbuffers_int32_vec_t fbs_dims = ns(Layout_dims(fbs_layout));
    flatbuffers_int32_vec_t fbs_stride = ns(Layout_stride(fbs_layout));
    tensor->layout.nr_dim = flatbuffers_int32_vec_len(fbs_dims);
    LOG_DEBUG("Tensor layout: [");
    for (int i = 0; i < tensor->layout.nr_dim; i++) {
        tensor->layout.dims[i] = flatbuffers_int32_vec_at(fbs_dims, i);
        LOG_DEBUG_NO_PREFIX("%d,", tensor->layout.dims[i]);
        tensor->layout.stride[i] = flatbuffers_int32_vec_at(fbs_stride, i);
    }
    //! format
    tensor->layout.format = format_from_fbs(ns(Layout_format(fbs_layout)));
    LOG_DEBUG_NO_PREFIX(
            "], format=%s, Dtype=%s\n", format2string(tensor->layout.format),
            dtype2string(tensor->dtype.type_enum));

    tensor->is_weight = 0;

    tensor->size = tensor_length_in_byte(tensor);
    return TinyNN_SUCCESS;
}

TinyNNStatus parse_weight(
        Tensor* weight, ns(Weight_table_t) fbs_weight, Device* host_dev) {