Method parse_tensor

src/onnx/onnx_parser.cpp:696–793

}

literal onnx_parser::parse_tensor(const onnx::TensorProto& t) const
{
    auto tensor_shape  = parse_tensor_shape(t);
    const auto& dims   = tensor_shape.lens();
    auto type          = tensor_shape.type();
    const auto& external_data = t.external_data();

    if(not external_data.empty())
    {
        const std::string& data_file = external_data.at(0).value();
        size_t num_data_fields       = external_data.size();
        size_t offset                = 0;
        size_t nbytes                = tensor_shape.bytes();

        if(num_data_fields > 1) // if offset field is present
        {
            offset = std::stoull(t.external_data().at(1).value());
        }
        if(num_data_fields > 2) // if nbytes field is present
        {
            nbytes = std::stoull(t.external_data().at(2).value());
        }
        std::vector<char> raw_buffer;
        if(not external_data_path.empty())
        {
            raw_buffer = read_buffer(fs::path{external_data_path} / data_file, offset, nbytes);
        }
        else
        {
            raw_buffer = read_buffer(path / data_file, offset, nbytes);
        }
        std::string s(raw_buffer.begin(), raw_buffer.end());
        return create_literal(type, dims, s.data());
    }

    if(t.has_raw_data())
    {
        const std::string& s = t.raw_data();
        return create_literal(type, dims, s.data());
    }

    switch(t.data_type())
    {
    case onnx::TensorProto::BOOL: return create_literal(shape::bool_type, dims, t.int32_data());

    // INT4 or UINT4 operate as 8-bit buffers:
    case onnx::TensorProto::INT4: return create_literal(shape::int8_type, dims, t.int32_data());
    case onnx::TensorProto::UINT4: return create_literal(shape::uint8_type, dims, t.int32_data());

    case onnx::TensorProto::INT8: return create_literal(shape::int8_type, dims, t.int32_data());
    case onnx::TensorProto::UINT8: return create_literal(shape::uint8_type, dims, t.int32_data());

    case onnx::TensorProto::INT16: return create_literal(shape::int16_type, dims, t.int32_data());
    case onnx::TensorProto::UINT16: return create_literal(shape::uint16_type, dims, t.int32_data());

    case onnx::TensorProto::INT32: return create_literal(shape::int32_type, dims, t.int32_data());
    case onnx::TensorProto::UINT32:
        return create_literal(shape::uint32_type, dims, t.uint64_data());