From 4c8618abb1d985e5943b0457ef70cb3c03aad0ca Mon Sep 17 00:00:00 2001
From: Julien Girard <julien.girard2@cea.fr>
Date: Mon, 13 Sep 2021 16:34:16 +0200
Subject: [PATCH] ONNX protobuf interface and dependencies
---
caisar.opam | 6 +
dune-project | 7 +-
onnx_piqi.ml | 1320 +++++++++++++++++++++++++++++++++++++++++++++++
onnx_piqi.mli | 718 ++++++++++++++++++++++++++
standalone/dune | 6 +
5 files changed, 2056 insertions(+), 1 deletion(-)
create mode 100644 onnx_piqi.ml
create mode 100644 onnx_piqi.mli
diff --git a/caisar.opam b/caisar.opam
index 7a4b2ab0..81686d25 100644
--- a/caisar.opam
+++ b/caisar.opam
@@ -7,7 +7,13 @@ depends: [
"dune" {>= "2.9" & >= "2.9.0"}
"dune-site" {>= "2.9.0"}
"why3"
+ "piqi" {>= "0.7.6"}
+ "piqilib" {>= "0.6.14"}
+ "zarith" {>= "1.7"}
+ "ocplib-endian" {>= "1.0"}
+ "dune" {>= "2.7" & >= "2.7.1"}
"base" {>= "v0.14.0"}
+ "stdio" {>= "v0.14.0"}
"cmdliner" {>= "1.0.4"}
"fmt" {>= "0.8.9"}
"logs" {>= "0.7.0"}
diff --git a/dune-project b/dune-project
index 6d4d8060..21ab8e86 100644
--- a/dune-project
+++ b/dune-project
@@ -16,8 +16,13 @@
(ocaml (>= 4.10))
(dune (>= 2.9.0))
(dune-site (>= 2.9.0))
- why3
+ (piqi (>= 0.7.6))
+ (piqilib (>= 0.6.14))
+ (zarith (>= 1.7))
+ (ocplib-endian (>= 1.0))
+ (dune (>= 2.7.1))
(base (>= v0.14.0))
+ (stdio (>= v0.14.0))
(cmdliner (>= 1.0.4))
(fmt (>= 0.8.9))
(logs (>= 0.7.0))
diff --git a/onnx_piqi.ml b/onnx_piqi.ml
new file mode 100644
index 00000000..2a0636f1
--- /dev/null
+++ b/onnx_piqi.ml
@@ -0,0 +1,1320 @@
+module rec Onnx_piqi:
+ sig
+ type float32 = float
+ type float64 = float
+ type uint64 = int64
+ type protobuf_int64 = int64
+ type binary = string
+ type protobuf_int32 = int32
+ type attribute_proto_attribute_type =
+ [
+ | `undefined
+ | `float
+ | `int
+ | `string
+ | `tensor
+ | `graph
+ | `sparse_tensor
+ | `type_proto
+ | `floats
+ | `ints
+ | `strings
+ | `tensors
+ | `graphs
+ | `sparse_tensors
+ | `type_protos
+ ]
+ type tensor_proto_data_type =
+ [
+ | `undefined
+ | `float
+ | `uint8
+ | `int8
+ | `uint16
+ | `int16
+ | `int32
+ | `int64
+ | `string
+ | `bool
+ | `float16
+ | `double
+ | `uint32
+ | `uint64
+ | `complex64
+ | `complex128
+ | `bfloat16
+ ]
+ type tensor_proto_data_location =
+ [
+ | `default
+ | `externall
+ ]
+ type version =
+ [
+ | `start_version
+ | `ir_version_2017_10_10
+ | `ir_version_2017_10_30
+ | `ir_version_2017_11_3
+ | `ir_version_2019_1_22
+ | `ir_version_2019_3_18
+ | `ir_version_2019_9_19
+ | `ir_version_2020_5_8
+ | `ir_version
+ ]
+ type operator_status =
+ [
+ | `experimental
+ | `stable
+ ]
+ type attribute_proto = Attribute_proto.t
+ type value_info_proto = Value_info_proto.t
+ type node_proto = Node_proto.t
+ type training_info_proto = Training_info_proto.t
+ type model_proto = Model_proto.t
+ type string_string_entry_proto = String_string_entry_proto.t
+ type tensor_annotation = Tensor_annotation.t
+ type graph_proto = Graph_proto.t
+ type tensor_proto = Tensor_proto.t
+ type tensor_proto_segment = Tensor_proto_segment.t
+ type sparse_tensor_proto = Sparse_tensor_proto.t
+ type tensor_shape_proto = Tensor_shape_proto.t
+ type tensor_shape_proto_dimension = Tensor_shape_proto_dimension.t
+ type type_proto = Type_proto.t
+ type type_proto_tensor = Type_proto_tensor.t
+ type type_proto_sequence = Type_proto_sequence.t
+ type type_proto_map = Type_proto_map.t
+ type type_proto_optional = Type_proto_optional.t
+ type type_proto_sparse_tensor = Type_proto_sparse_tensor.t
+ type operator_set_id_proto = Operator_set_id_proto.t
+ type function_proto = Function_proto.t
+ end = Onnx_piqi
+and Attribute_proto:
+ sig
+ type t = {
+ mutable name: string option;
+ mutable ref_attr_name: string option;
+ mutable doc_string: string option;
+ mutable type_: Onnx_piqi.attribute_proto_attribute_type option;
+ mutable f: Onnx_piqi.float32 option;
+ mutable i: Onnx_piqi.protobuf_int64 option;
+ mutable s: Onnx_piqi.binary option;
+ mutable t: Onnx_piqi.tensor_proto option;
+ mutable g: Onnx_piqi.graph_proto option;
+ mutable sparse_tensor: Onnx_piqi.sparse_tensor_proto option;
+ mutable tp: Onnx_piqi.type_proto option;
+ mutable floats: Onnx_piqi.float32 list;
+ mutable ints: Onnx_piqi.protobuf_int64 list;
+ mutable strings: Onnx_piqi.binary list;
+ mutable tensors: Onnx_piqi.tensor_proto list;
+ mutable graphs: Onnx_piqi.graph_proto list;
+ mutable sparse_tensors: Onnx_piqi.sparse_tensor_proto list;
+ mutable type_protos: Onnx_piqi.type_proto list;
+ }
+ end = Attribute_proto
+and Value_info_proto:
+ sig
+ type t = {
+ mutable name: string option;
+ mutable type_: Onnx_piqi.type_proto option;
+ mutable doc_string: string option;
+ }
+ end = Value_info_proto
+and Node_proto:
+ sig
+ type t = {
+ mutable input: string list;
+ mutable output: string list;
+ mutable name: string option;
+ mutable op_type: string option;
+ mutable domain: string option;
+ mutable attribute: Onnx_piqi.attribute_proto list;
+ mutable doc_string: string option;
+ }
+ end = Node_proto
+and Training_info_proto:
+ sig
+ type t = {
+ mutable initialization: Onnx_piqi.graph_proto option;
+ mutable algorithm: Onnx_piqi.graph_proto option;
+ mutable initialization_binding: Onnx_piqi.string_string_entry_proto list;
+ mutable update_binding: Onnx_piqi.string_string_entry_proto list;
+ }
+ end = Training_info_proto
+and Model_proto:
+ sig
+ type t = {
+ mutable ir_version: Onnx_piqi.protobuf_int64 option;
+ mutable opset_import: Onnx_piqi.operator_set_id_proto list;
+ mutable producer_name: string option;
+ mutable producer_version: string option;
+ mutable domain: string option;
+ mutable model_version: Onnx_piqi.protobuf_int64 option;
+ mutable doc_string: string option;
+ mutable graph: Onnx_piqi.graph_proto option;
+ mutable metadata_props: Onnx_piqi.string_string_entry_proto list;
+ mutable training_info: Onnx_piqi.training_info_proto list;
+ mutable functions: Onnx_piqi.function_proto list;
+ }
+ end = Model_proto
+and String_string_entry_proto:
+ sig
+ type t = {
+ mutable key: string option;
+ mutable value: string option;
+ }
+ end = String_string_entry_proto
+and Tensor_annotation:
+ sig
+ type t = {
+ mutable tensor_name: string option;
+ mutable quant_parameter_tensor_names: Onnx_piqi.string_string_entry_proto list;
+ }
+ end = Tensor_annotation
+and Graph_proto:
+ sig
+ type t = {
+ mutable node: Onnx_piqi.node_proto list;
+ mutable name: string option;
+ mutable initializer_: Onnx_piqi.tensor_proto list;
+ mutable sparse_initializer: Onnx_piqi.sparse_tensor_proto list;
+ mutable doc_string: string option;
+ mutable input: Onnx_piqi.value_info_proto list;
+ mutable output: Onnx_piqi.value_info_proto list;
+ mutable value_info: Onnx_piqi.value_info_proto list;
+ mutable quantization_annotation: Onnx_piqi.tensor_annotation list;
+ }
+ end = Graph_proto
+and Tensor_proto:
+ sig
+ type t = {
+ mutable dims: Onnx_piqi.protobuf_int64 list;
+ mutable data_type: Onnx_piqi.protobuf_int32 option;
+ mutable segment: Onnx_piqi.tensor_proto_segment option;
+ mutable float_data: Onnx_piqi.float32 list;
+ mutable int32_data: Onnx_piqi.protobuf_int32 list;
+ mutable string_data: Onnx_piqi.binary list;
+ mutable int64_data: Onnx_piqi.protobuf_int64 list;
+ mutable name: string option;
+ mutable doc_string: string option;
+ mutable raw_data: Onnx_piqi.binary option;
+ mutable externall_data: Onnx_piqi.string_string_entry_proto list;
+ mutable data_location: Onnx_piqi.tensor_proto_data_location option;
+ mutable double_data: Onnx_piqi.float64 list;
+ mutable uint64_data: Onnx_piqi.uint64 list;
+ }
+ end = Tensor_proto
+and Tensor_proto_segment:
+ sig
+ type t = {
+ mutable begin_: Onnx_piqi.protobuf_int64 option;
+ mutable end_: Onnx_piqi.protobuf_int64 option;
+ }
+ end = Tensor_proto_segment
+and Sparse_tensor_proto:
+ sig
+ type t = {
+ mutable values: Onnx_piqi.tensor_proto option;
+ mutable indices: Onnx_piqi.tensor_proto option;
+ mutable dims: Onnx_piqi.protobuf_int64 list;
+ }
+ end = Sparse_tensor_proto
+and Tensor_shape_proto:
+ sig
+ type t = {
+ mutable dim: Onnx_piqi.tensor_shape_proto_dimension list;
+ }
+ end = Tensor_shape_proto
+and Tensor_shape_proto_dimension:
+ sig
+ type t = {
+ mutable dim_value: Onnx_piqi.protobuf_int64 option;
+ mutable dim_param: string option;
+ mutable denotation: string option;
+ }
+ end = Tensor_shape_proto_dimension
+and Type_proto:
+ sig
+ type t = {
+ mutable tensor_type: Onnx_piqi.type_proto_tensor option;
+ mutable sequence_type: Onnx_piqi.type_proto_sequence option;
+ mutable map_type: Onnx_piqi.type_proto_map option;
+ mutable optional_type: Onnx_piqi.type_proto_optional option;
+ mutable sparse_tensor_type: Onnx_piqi.type_proto_sparse_tensor option;
+ mutable denotation: string option;
+ }
+ end = Type_proto
+and Type_proto_tensor:
+ sig
+ type t = {
+ mutable elem_type: Onnx_piqi.protobuf_int32 option;
+ mutable shape: Onnx_piqi.tensor_shape_proto option;
+ }
+ end = Type_proto_tensor
+and Type_proto_sequence:
+ sig
+ type t = {
+ mutable elem_type: Onnx_piqi.type_proto option;
+ }
+ end = Type_proto_sequence
+and Type_proto_map:
+ sig
+ type t = {
+ mutable key_type: Onnx_piqi.protobuf_int32 option;
+ mutable value_type: Onnx_piqi.type_proto option;
+ }
+ end = Type_proto_map
+and Type_proto_optional:
+ sig
+ type t = {
+ mutable elem_type: Onnx_piqi.type_proto option;
+ }
+ end = Type_proto_optional
+and Type_proto_sparse_tensor:
+ sig
+ type t = {
+ mutable elem_type: Onnx_piqi.protobuf_int32 option;
+ mutable shape: Onnx_piqi.tensor_shape_proto option;
+ }
+ end = Type_proto_sparse_tensor
+and Operator_set_id_proto:
+ sig
+ type t = {
+ mutable domain: string option;
+ mutable version: Onnx_piqi.protobuf_int64 option;
+ }
+ end = Operator_set_id_proto
+and Function_proto:
+ sig
+ type t = {
+ mutable name: string option;
+ mutable input: string list;
+ mutable output: string list;
+ mutable attribute: string list;
+ mutable node: Onnx_piqi.node_proto list;
+ mutable doc_string: string option;
+ mutable opset_import: Onnx_piqi.operator_set_id_proto list;
+ mutable domain: string option;
+ }
+ end = Function_proto
+
+
+let rec parse_int64 x = Piqirun.int64_of_zigzag_varint x
+and packed_parse_int64 x = Piqirun.int64_of_packed_zigzag_varint x
+
+and parse_int32 x = Piqirun.int32_of_zigzag_varint x
+and packed_parse_int32 x = Piqirun.int32_of_packed_zigzag_varint x
+
+and parse_string x = Piqirun.string_of_block x
+
+and parse_float32 x = Piqirun.float_of_fixed32 x
+and packed_parse_float32 x = Piqirun.float_of_packed_fixed32 x
+
+and parse_protobuf_int64 x = Piqirun.int64_of_signed_varint x
+and packed_parse_protobuf_int64 x = Piqirun.int64_of_packed_signed_varint x
+
+and parse_binary x = Piqirun.string_of_block x
+
+and parse_protobuf_int32 x = Piqirun.int32_of_signed_varint x
+and packed_parse_protobuf_int32 x = Piqirun.int32_of_packed_signed_varint x
+
+and parse_float64 x = Piqirun.float_of_fixed64 x
+and packed_parse_float64 x = Piqirun.float_of_packed_fixed64 x
+
+and parse_uint64 x = Piqirun.int64_of_varint x
+and packed_parse_uint64 x = Piqirun.int64_of_packed_varint x
+
+and parse_attribute_proto x =
+ let x = Piqirun.parse_record x in
+ let _name, x = Piqirun.parse_optional_field 1 parse_string x in
+ let _f, x = Piqirun.parse_optional_field 2 parse_float32 x in
+ let _i, x = Piqirun.parse_optional_field 3 parse_protobuf_int64 x in
+ let _s, x = Piqirun.parse_optional_field 4 parse_binary x in
+ let _t, x = Piqirun.parse_optional_field 5 parse_tensor_proto x in
+ let _g, x = Piqirun.parse_optional_field 6 parse_graph_proto x in
+ let _floats, x = Piqirun.parse_repeated_field 7 parse_float32 x in
+ let _ints, x = Piqirun.parse_repeated_field 8 parse_protobuf_int64 x in
+ let _strings, x = Piqirun.parse_repeated_field 9 parse_binary x in
+ let _tensors, x = Piqirun.parse_repeated_field 10 parse_tensor_proto x in
+ let _graphs, x = Piqirun.parse_repeated_field 11 parse_graph_proto x in
+ let _doc_string, x = Piqirun.parse_optional_field 13 parse_string x in
+ let _tp, x = Piqirun.parse_optional_field 14 parse_type_proto x in
+ let _type_protos, x = Piqirun.parse_repeated_field 15 parse_type_proto x in
+ let _type_, x = Piqirun.parse_optional_field 20 parse_attribute_proto_attribute_type x in
+ let _ref_attr_name, x = Piqirun.parse_optional_field 21 parse_string x in
+ let _sparse_tensor, x = Piqirun.parse_optional_field 22 parse_sparse_tensor_proto x in
+ let _sparse_tensors, x = Piqirun.parse_repeated_field 23 parse_sparse_tensor_proto x in
+ Piqirun.check_unparsed_fields x;
+ {
+ Attribute_proto.name = _name;
+ Attribute_proto.f = _f;
+ Attribute_proto.i = _i;
+ Attribute_proto.s = _s;
+ Attribute_proto.t = _t;
+ Attribute_proto.g = _g;
+ Attribute_proto.floats = _floats;
+ Attribute_proto.ints = _ints;
+ Attribute_proto.strings = _strings;
+ Attribute_proto.tensors = _tensors;
+ Attribute_proto.graphs = _graphs;
+ Attribute_proto.doc_string = _doc_string;
+ Attribute_proto.tp = _tp;
+ Attribute_proto.type_protos = _type_protos;
+ Attribute_proto.type_ = _type_;
+ Attribute_proto.ref_attr_name = _ref_attr_name;
+ Attribute_proto.sparse_tensor = _sparse_tensor;
+ Attribute_proto.sparse_tensors = _sparse_tensors;
+ }
+
+and parse_attribute_proto_attribute_type x =
+ match Piqirun.int32_of_signed_varint x with
+ | 0l -> `undefined
+ | 1l -> `float
+ | 2l -> `int
+ | 3l -> `string
+ | 4l -> `tensor
+ | 5l -> `graph
+ | 11l -> `sparse_tensor
+ | 13l -> `type_proto
+ | 6l -> `floats
+ | 7l -> `ints
+ | 8l -> `strings
+ | 9l -> `tensors
+ | 10l -> `graphs
+ | 12l -> `sparse_tensors
+ | 14l -> `type_protos
+ | x -> Piqirun.error_enum_const x
+and packed_parse_attribute_proto_attribute_type x =
+ match Piqirun.int32_of_packed_signed_varint x with
+ | 0l -> `undefined
+ | 1l -> `float
+ | 2l -> `int
+ | 3l -> `string
+ | 4l -> `tensor
+ | 5l -> `graph
+ | 11l -> `sparse_tensor
+ | 13l -> `type_proto
+ | 6l -> `floats
+ | 7l -> `ints
+ | 8l -> `strings
+ | 9l -> `tensors
+ | 10l -> `graphs
+ | 12l -> `sparse_tensors
+ | 14l -> `type_protos
+ | x -> Piqirun.error_enum_const x
+
+and parse_value_info_proto x =
+ let x = Piqirun.parse_record x in
+ let _name, x = Piqirun.parse_optional_field 1 parse_string x in
+ let _type_, x = Piqirun.parse_optional_field 2 parse_type_proto x in
+ let _doc_string, x = Piqirun.parse_optional_field 3 parse_string x in
+ Piqirun.check_unparsed_fields x;
+ {
+ Value_info_proto.name = _name;
+ Value_info_proto.type_ = _type_;
+ Value_info_proto.doc_string = _doc_string;
+ }
+
+and parse_node_proto x =
+ let x = Piqirun.parse_record x in
+ let _input, x = Piqirun.parse_repeated_field 1 parse_string x in
+ let _output, x = Piqirun.parse_repeated_field 2 parse_string x in
+ let _name, x = Piqirun.parse_optional_field 3 parse_string x in
+ let _op_type, x = Piqirun.parse_optional_field 4 parse_string x in
+ let _attribute, x = Piqirun.parse_repeated_field 5 parse_attribute_proto x in
+ let _doc_string, x = Piqirun.parse_optional_field 6 parse_string x in
+ let _domain, x = Piqirun.parse_optional_field 7 parse_string x in
+ Piqirun.check_unparsed_fields x;
+ {
+ Node_proto.input = _input;
+ Node_proto.output = _output;
+ Node_proto.name = _name;
+ Node_proto.op_type = _op_type;
+ Node_proto.attribute = _attribute;
+ Node_proto.doc_string = _doc_string;
+ Node_proto.domain = _domain;
+ }
+
+and parse_training_info_proto x =
+ let x = Piqirun.parse_record x in
+ let _initialization, x = Piqirun.parse_optional_field 1 parse_graph_proto x in
+ let _algorithm, x = Piqirun.parse_optional_field 2 parse_graph_proto x in
+ let _initialization_binding, x = Piqirun.parse_repeated_field 3 parse_string_string_entry_proto x in
+ let _update_binding, x = Piqirun.parse_repeated_field 4 parse_string_string_entry_proto x in
+ Piqirun.check_unparsed_fields x;
+ {
+ Training_info_proto.initialization = _initialization;
+ Training_info_proto.algorithm = _algorithm;
+ Training_info_proto.initialization_binding = _initialization_binding;
+ Training_info_proto.update_binding = _update_binding;
+ }
+
+and parse_model_proto x =
+ let x = Piqirun.parse_record x in
+ let _ir_version, x = Piqirun.parse_optional_field 1 parse_protobuf_int64 x in
+ let _producer_name, x = Piqirun.parse_optional_field 2 parse_string x in
+ let _producer_version, x = Piqirun.parse_optional_field 3 parse_string x in
+ let _domain, x = Piqirun.parse_optional_field 4 parse_string x in
+ let _model_version, x = Piqirun.parse_optional_field 5 parse_protobuf_int64 x in
+ let _doc_string, x = Piqirun.parse_optional_field 6 parse_string x in
+ let _graph, x = Piqirun.parse_optional_field 7 parse_graph_proto x in
+ let _opset_import, x = Piqirun.parse_repeated_field 8 parse_operator_set_id_proto x in
+ let _metadata_props, x = Piqirun.parse_repeated_field 14 parse_string_string_entry_proto x in
+ let _training_info, x = Piqirun.parse_repeated_field 20 parse_training_info_proto x in
+ let _functions, x = Piqirun.parse_repeated_field 25 parse_function_proto x in
+ Piqirun.check_unparsed_fields x;
+ {
+ Model_proto.ir_version = _ir_version;
+ Model_proto.producer_name = _producer_name;
+ Model_proto.producer_version = _producer_version;
+ Model_proto.domain = _domain;
+ Model_proto.model_version = _model_version;
+ Model_proto.doc_string = _doc_string;
+ Model_proto.graph = _graph;
+ Model_proto.opset_import = _opset_import;
+ Model_proto.metadata_props = _metadata_props;
+ Model_proto.training_info = _training_info;
+ Model_proto.functions = _functions;
+ }
+
+and parse_string_string_entry_proto x =
+ let x = Piqirun.parse_record x in
+ let _key, x = Piqirun.parse_optional_field 1 parse_string x in
+ let _value, x = Piqirun.parse_optional_field 2 parse_string x in
+ Piqirun.check_unparsed_fields x;
+ {
+ String_string_entry_proto.key = _key;
+ String_string_entry_proto.value = _value;
+ }
+
+and parse_tensor_annotation x =
+ let x = Piqirun.parse_record x in
+ let _tensor_name, x = Piqirun.parse_optional_field 1 parse_string x in
+ let _quant_parameter_tensor_names, x = Piqirun.parse_repeated_field 2 parse_string_string_entry_proto x in
+ Piqirun.check_unparsed_fields x;
+ {
+ Tensor_annotation.tensor_name = _tensor_name;
+ Tensor_annotation.quant_parameter_tensor_names = _quant_parameter_tensor_names;
+ }
+
+and parse_graph_proto x =
+ let x = Piqirun.parse_record x in
+ let _node, x = Piqirun.parse_repeated_field 1 parse_node_proto x in
+ let _name, x = Piqirun.parse_optional_field 2 parse_string x in
+ let _initializer_, x = Piqirun.parse_repeated_field 5 parse_tensor_proto x in
+ let _doc_string, x = Piqirun.parse_optional_field 10 parse_string x in
+ let _input, x = Piqirun.parse_repeated_field 11 parse_value_info_proto x in
+ let _output, x = Piqirun.parse_repeated_field 12 parse_value_info_proto x in
+ let _value_info, x = Piqirun.parse_repeated_field 13 parse_value_info_proto x in
+ let _quantization_annotation, x = Piqirun.parse_repeated_field 14 parse_tensor_annotation x in
+ let _sparse_initializer, x = Piqirun.parse_repeated_field 15 parse_sparse_tensor_proto x in
+ Piqirun.check_unparsed_fields x;
+ {
+ Graph_proto.node = _node;
+ Graph_proto.name = _name;
+ Graph_proto.initializer_ = _initializer_;
+ Graph_proto.doc_string = _doc_string;
+ Graph_proto.input = _input;
+ Graph_proto.output = _output;
+ Graph_proto.value_info = _value_info;
+ Graph_proto.quantization_annotation = _quantization_annotation;
+ Graph_proto.sparse_initializer = _sparse_initializer;
+ }
+
+and parse_tensor_proto x =
+ let x = Piqirun.parse_record x in
+ let _dims, x = Piqirun.parse_repeated_field 1 parse_protobuf_int64 x in
+ let _data_type, x = Piqirun.parse_optional_field 2 parse_protobuf_int32 x in
+ let _segment, x = Piqirun.parse_optional_field 3 parse_tensor_proto_segment x in
+ let _float_data, x = Piqirun.parse_packed_repeated_field 4 packed_parse_float32 parse_float32 x in
+ let _int32_data, x = Piqirun.parse_packed_repeated_field 5 packed_parse_protobuf_int32 parse_protobuf_int32 x in
+ let _string_data, x = Piqirun.parse_repeated_field 6 parse_binary x in
+ let _int64_data, x = Piqirun.parse_packed_repeated_field 7 packed_parse_protobuf_int64 parse_protobuf_int64 x in
+ let _name, x = Piqirun.parse_optional_field 8 parse_string x in
+ let _raw_data, x = Piqirun.parse_optional_field 9 parse_binary x in
+ let _double_data, x = Piqirun.parse_packed_repeated_field 10 packed_parse_float64 parse_float64 x in
+ let _uint64_data, x = Piqirun.parse_packed_repeated_field 11 packed_parse_uint64 parse_uint64 x in
+ let _doc_string, x = Piqirun.parse_optional_field 12 parse_string x in
+ let _externall_data, x = Piqirun.parse_repeated_field 13 parse_string_string_entry_proto x in
+ let _data_location, x = Piqirun.parse_optional_field 14 parse_tensor_proto_data_location x in
+ Piqirun.check_unparsed_fields x;
+ {
+ Tensor_proto.dims = _dims;
+ Tensor_proto.data_type = _data_type;
+ Tensor_proto.segment = _segment;
+ Tensor_proto.float_data = _float_data;
+ Tensor_proto.int32_data = _int32_data;
+ Tensor_proto.string_data = _string_data;
+ Tensor_proto.int64_data = _int64_data;
+ Tensor_proto.name = _name;
+ Tensor_proto.raw_data = _raw_data;
+ Tensor_proto.double_data = _double_data;
+ Tensor_proto.uint64_data = _uint64_data;
+ Tensor_proto.doc_string = _doc_string;
+ Tensor_proto.externall_data = _externall_data;
+ Tensor_proto.data_location = _data_location;
+ }
+
+and parse_tensor_proto_segment x =
+ let x = Piqirun.parse_record x in
+ let _begin_, x = Piqirun.parse_optional_field 1 parse_protobuf_int64 x in
+ let _end_, x = Piqirun.parse_optional_field 2 parse_protobuf_int64 x in
+ Piqirun.check_unparsed_fields x;
+ {
+ Tensor_proto_segment.begin_ = _begin_;
+ Tensor_proto_segment.end_ = _end_;
+ }
+
+and parse_tensor_proto_data_type x =
+ match Piqirun.int32_of_signed_varint x with
+ | 0l -> `undefined
+ | 1l -> `float
+ | 2l -> `uint8
+ | 3l -> `int8
+ | 4l -> `uint16
+ | 5l -> `int16
+ | 6l -> `int32
+ | 7l -> `int64
+ | 8l -> `string
+ | 9l -> `bool
+ | 10l -> `float16
+ | 11l -> `double
+ | 12l -> `uint32
+ | 13l -> `uint64
+ | 14l -> `complex64
+ | 15l -> `complex128
+ | 16l -> `bfloat16
+ | x -> Piqirun.error_enum_const x
+and packed_parse_tensor_proto_data_type x =
+ match Piqirun.int32_of_packed_signed_varint x with
+ | 0l -> `undefined
+ | 1l -> `float
+ | 2l -> `uint8
+ | 3l -> `int8
+ | 4l -> `uint16
+ | 5l -> `int16
+ | 6l -> `int32
+ | 7l -> `int64
+ | 8l -> `string
+ | 9l -> `bool
+ | 10l -> `float16
+ | 11l -> `double
+ | 12l -> `uint32
+ | 13l -> `uint64
+ | 14l -> `complex64
+ | 15l -> `complex128
+ | 16l -> `bfloat16
+ | x -> Piqirun.error_enum_const x
+
+and parse_tensor_proto_data_location x =
+ match Piqirun.int32_of_signed_varint x with
+ | 0l -> `default
+ | 1l -> `externall
+ | x -> Piqirun.error_enum_const x
+and packed_parse_tensor_proto_data_location x =
+ match Piqirun.int32_of_packed_signed_varint x with
+ | 0l -> `default
+ | 1l -> `externall
+ | x -> Piqirun.error_enum_const x
+
+and parse_sparse_tensor_proto x =
+ let x = Piqirun.parse_record x in
+ let _values, x = Piqirun.parse_optional_field 1 parse_tensor_proto x in
+ let _indices, x = Piqirun.parse_optional_field 2 parse_tensor_proto x in
+ let _dims, x = Piqirun.parse_repeated_field 3 parse_protobuf_int64 x in
+ Piqirun.check_unparsed_fields x;
+ {
+ Sparse_tensor_proto.values = _values;
+ Sparse_tensor_proto.indices = _indices;
+ Sparse_tensor_proto.dims = _dims;
+ }
+
+and parse_tensor_shape_proto x =
+ let x = Piqirun.parse_record x in
+ let _dim, x = Piqirun.parse_repeated_field 1 parse_tensor_shape_proto_dimension x in
+ Piqirun.check_unparsed_fields x;
+ {
+ Tensor_shape_proto.dim = _dim;
+ }
+
+and parse_tensor_shape_proto_dimension x =
+ let x = Piqirun.parse_record x in
+ let _dim_value, x = Piqirun.parse_optional_field 1 parse_protobuf_int64 x in
+ let _dim_param, x = Piqirun.parse_optional_field 2 parse_string x in
+ let _denotation, x = Piqirun.parse_optional_field 3 parse_string x in
+ Piqirun.check_unparsed_fields x;
+ {
+ Tensor_shape_proto_dimension.dim_value = _dim_value;
+ Tensor_shape_proto_dimension.dim_param = _dim_param;
+ Tensor_shape_proto_dimension.denotation = _denotation;
+ }
+
+and parse_type_proto x =
+ let x = Piqirun.parse_record x in
+ let _tensor_type, x = Piqirun.parse_optional_field 1 parse_type_proto_tensor x in
+ let _sequence_type, x = Piqirun.parse_optional_field 4 parse_type_proto_sequence x in
+ let _map_type, x = Piqirun.parse_optional_field 5 parse_type_proto_map x in
+ let _denotation, x = Piqirun.parse_optional_field 6 parse_string x in
+ let _sparse_tensor_type, x = Piqirun.parse_optional_field 8 parse_type_proto_sparse_tensor x in
+ let _optional_type, x = Piqirun.parse_optional_field 9 parse_type_proto_optional x in
+ Piqirun.check_unparsed_fields x;
+ {
+ Type_proto.tensor_type = _tensor_type;
+ Type_proto.sequence_type = _sequence_type;
+ Type_proto.map_type = _map_type;
+ Type_proto.denotation = _denotation;
+ Type_proto.sparse_tensor_type = _sparse_tensor_type;
+ Type_proto.optional_type = _optional_type;
+ }
+
+and parse_type_proto_tensor x =
+ let x = Piqirun.parse_record x in
+ let _elem_type, x = Piqirun.parse_optional_field 1 parse_protobuf_int32 x in
+ let _shape, x = Piqirun.parse_optional_field 2 parse_tensor_shape_proto x in
+ Piqirun.check_unparsed_fields x;
+ {
+ Type_proto_tensor.elem_type = _elem_type;
+ Type_proto_tensor.shape = _shape;
+ }
+
+and parse_type_proto_sequence x =
+ let x = Piqirun.parse_record x in
+ let _elem_type, x = Piqirun.parse_optional_field 1 parse_type_proto x in
+ Piqirun.check_unparsed_fields x;
+ {
+ Type_proto_sequence.elem_type = _elem_type;
+ }
+
+and parse_type_proto_map x =
+ let x = Piqirun.parse_record x in
+ let _key_type, x = Piqirun.parse_optional_field 1 parse_protobuf_int32 x in
+ let _value_type, x = Piqirun.parse_optional_field 2 parse_type_proto x in
+ Piqirun.check_unparsed_fields x;
+ {
+ Type_proto_map.key_type = _key_type;
+ Type_proto_map.value_type = _value_type;
+ }
+
+and parse_type_proto_optional x =
+ let x = Piqirun.parse_record x in
+ let _elem_type, x = Piqirun.parse_optional_field 1 parse_type_proto x in
+ Piqirun.check_unparsed_fields x;
+ {
+ Type_proto_optional.elem_type = _elem_type;
+ }
+
+and parse_type_proto_sparse_tensor x =
+ let x = Piqirun.parse_record x in
+ let _elem_type, x = Piqirun.parse_optional_field 1 parse_protobuf_int32 x in
+ let _shape, x = Piqirun.parse_optional_field 2 parse_tensor_shape_proto x in
+ Piqirun.check_unparsed_fields x;
+ {
+ Type_proto_sparse_tensor.elem_type = _elem_type;
+ Type_proto_sparse_tensor.shape = _shape;
+ }
+
+and parse_operator_set_id_proto x =
+ let x = Piqirun.parse_record x in
+ let _domain, x = Piqirun.parse_optional_field 1 parse_string x in
+ let _version, x = Piqirun.parse_optional_field 2 parse_protobuf_int64 x in
+ Piqirun.check_unparsed_fields x;
+ {
+ Operator_set_id_proto.domain = _domain;
+ Operator_set_id_proto.version = _version;
+ }
+
+and parse_function_proto x =
+ let x = Piqirun.parse_record x in
+ let _name, x = Piqirun.parse_optional_field 1 parse_string x in
+ let _input, x = Piqirun.parse_repeated_field 4 parse_string x in
+ let _output, x = Piqirun.parse_repeated_field 5 parse_string x in
+ let _attribute, x = Piqirun.parse_repeated_field 6 parse_string x in
+ let _node, x = Piqirun.parse_repeated_field 7 parse_node_proto x in
+ let _doc_string, x = Piqirun.parse_optional_field 8 parse_string x in
+ let _opset_import, x = Piqirun.parse_repeated_field 9 parse_operator_set_id_proto x in
+ let _domain, x = Piqirun.parse_optional_field 10 parse_string x in
+ Piqirun.check_unparsed_fields x;
+ {
+ Function_proto.name = _name;
+ Function_proto.input = _input;
+ Function_proto.output = _output;
+ Function_proto.attribute = _attribute;
+ Function_proto.node = _node;
+ Function_proto.doc_string = _doc_string;
+ Function_proto.opset_import = _opset_import;
+ Function_proto.domain = _domain;
+ }
+
+and parse_version x =
+ match Piqirun.int32_of_signed_varint x with
+ | 0l -> `start_version
+ | 1l -> `ir_version_2017_10_10
+ | 2l -> `ir_version_2017_10_30
+ | 3l -> `ir_version_2017_11_3
+ | 4l -> `ir_version_2019_1_22
+ | 5l -> `ir_version_2019_3_18
+ | 6l -> `ir_version_2019_9_19
+ | 7l -> `ir_version_2020_5_8
+ | 8l -> `ir_version
+ | x -> Piqirun.error_enum_const x
+and packed_parse_version x =
+ match Piqirun.int32_of_packed_signed_varint x with
+ | 0l -> `start_version
+ | 1l -> `ir_version_2017_10_10
+ | 2l -> `ir_version_2017_10_30
+ | 3l -> `ir_version_2017_11_3
+ | 4l -> `ir_version_2019_1_22
+ | 5l -> `ir_version_2019_3_18
+ | 6l -> `ir_version_2019_9_19
+ | 7l -> `ir_version_2020_5_8
+ | 8l -> `ir_version
+ | x -> Piqirun.error_enum_const x
+
+and parse_operator_status x =
+ match Piqirun.int32_of_signed_varint x with
+ | 0l -> `experimental
+ | 1l -> `stable
+ | x -> Piqirun.error_enum_const x
+and packed_parse_operator_status x =
+ match Piqirun.int32_of_packed_signed_varint x with
+ | 0l -> `experimental
+ | 1l -> `stable
+ | x -> Piqirun.error_enum_const x
+
+
+let rec gen__int64 code x = Piqirun.int64_to_zigzag_varint code x
+and packed_gen__int64 x = Piqirun.int64_to_packed_zigzag_varint x
+
+and gen__int32 code x = Piqirun.int32_to_zigzag_varint code x
+and packed_gen__int32 x = Piqirun.int32_to_packed_zigzag_varint x
+
+and gen__string code x = Piqirun.string_to_block code x
+
+and gen__float32 code x = Piqirun.float_to_fixed32 code x
+and packed_gen__float32 x = Piqirun.float_to_packed_fixed32 x
+
+and gen__protobuf_int64 code x = Piqirun.int64_to_signed_varint code x
+and packed_gen__protobuf_int64 x = Piqirun.int64_to_packed_signed_varint x
+
+and gen__binary code x = Piqirun.string_to_block code x
+
+and gen__protobuf_int32 code x = Piqirun.int32_to_signed_varint code x
+and packed_gen__protobuf_int32 x = Piqirun.int32_to_packed_signed_varint x
+
+and gen__float64 code x = Piqirun.float_to_fixed64 code x
+and packed_gen__float64 x = Piqirun.float_to_packed_fixed64 x
+
+and gen__uint64 code x = Piqirun.int64_to_varint code x
+and packed_gen__uint64 x = Piqirun.int64_to_packed_varint x
+
+and gen__attribute_proto code x =
+ let _name = Piqirun.gen_optional_field 1 gen__string x.Attribute_proto.name in
+ let _f = Piqirun.gen_optional_field 2 gen__float32 x.Attribute_proto.f in
+ let _i = Piqirun.gen_optional_field 3 gen__protobuf_int64 x.Attribute_proto.i in
+ let _s = Piqirun.gen_optional_field 4 gen__binary x.Attribute_proto.s in
+ let _t = Piqirun.gen_optional_field 5 gen__tensor_proto x.Attribute_proto.t in
+ let _g = Piqirun.gen_optional_field 6 gen__graph_proto x.Attribute_proto.g in
+ let _floats = Piqirun.gen_repeated_field 7 gen__float32 x.Attribute_proto.floats in
+ let _ints = Piqirun.gen_repeated_field 8 gen__protobuf_int64 x.Attribute_proto.ints in
+ let _strings = Piqirun.gen_repeated_field 9 gen__binary x.Attribute_proto.strings in
+ let _tensors = Piqirun.gen_repeated_field 10 gen__tensor_proto x.Attribute_proto.tensors in
+ let _graphs = Piqirun.gen_repeated_field 11 gen__graph_proto x.Attribute_proto.graphs in
+ let _doc_string = Piqirun.gen_optional_field 13 gen__string x.Attribute_proto.doc_string in
+ let _tp = Piqirun.gen_optional_field 14 gen__type_proto x.Attribute_proto.tp in
+ let _type_protos = Piqirun.gen_repeated_field 15 gen__type_proto x.Attribute_proto.type_protos in
+ let _type_ = Piqirun.gen_optional_field 20 gen__attribute_proto_attribute_type x.Attribute_proto.type_ in
+ let _ref_attr_name = Piqirun.gen_optional_field 21 gen__string x.Attribute_proto.ref_attr_name in
+ let _sparse_tensor = Piqirun.gen_optional_field 22 gen__sparse_tensor_proto x.Attribute_proto.sparse_tensor in
+ let _sparse_tensors = Piqirun.gen_repeated_field 23 gen__sparse_tensor_proto x.Attribute_proto.sparse_tensors in
+ Piqirun.gen_record code (_name :: _f :: _i :: _s :: _t :: _g :: _floats :: _ints :: _strings :: _tensors :: _graphs :: _doc_string :: _tp :: _type_protos :: _type_ :: _ref_attr_name :: _sparse_tensor :: _sparse_tensors :: [])
+
+and gen__attribute_proto_attribute_type code x =
+ Piqirun.int32_to_signed_varint code (match x with
+ | `undefined -> 0l
+ | `float -> 1l
+ | `int -> 2l
+ | `string -> 3l
+ | `tensor -> 4l
+ | `graph -> 5l
+ | `sparse_tensor -> 11l
+ | `type_proto -> 13l
+ | `floats -> 6l
+ | `ints -> 7l
+ | `strings -> 8l
+ | `tensors -> 9l
+ | `graphs -> 10l
+ | `sparse_tensors -> 12l
+ | `type_protos -> 14l
+ )
+and packed_gen__attribute_proto_attribute_type x =
+ Piqirun.int32_to_packed_signed_varint (match x with
+ | `undefined -> 0l
+ | `float -> 1l
+ | `int -> 2l
+ | `string -> 3l
+ | `tensor -> 4l
+ | `graph -> 5l
+ | `sparse_tensor -> 11l
+ | `type_proto -> 13l
+ | `floats -> 6l
+ | `ints -> 7l
+ | `strings -> 8l
+ | `tensors -> 9l
+ | `graphs -> 10l
+ | `sparse_tensors -> 12l
+ | `type_protos -> 14l
+ )
+
+and gen__value_info_proto code x =
+ let _name = Piqirun.gen_optional_field 1 gen__string x.Value_info_proto.name in
+ let _type_ = Piqirun.gen_optional_field 2 gen__type_proto x.Value_info_proto.type_ in
+ let _doc_string = Piqirun.gen_optional_field 3 gen__string x.Value_info_proto.doc_string in
+ Piqirun.gen_record code (_name :: _type_ :: _doc_string :: [])
+
+and gen__node_proto code x =
+ let _input = Piqirun.gen_repeated_field 1 gen__string x.Node_proto.input in
+ let _output = Piqirun.gen_repeated_field 2 gen__string x.Node_proto.output in
+ let _name = Piqirun.gen_optional_field 3 gen__string x.Node_proto.name in
+ let _op_type = Piqirun.gen_optional_field 4 gen__string x.Node_proto.op_type in
+ let _attribute = Piqirun.gen_repeated_field 5 gen__attribute_proto x.Node_proto.attribute in
+ let _doc_string = Piqirun.gen_optional_field 6 gen__string x.Node_proto.doc_string in
+ let _domain = Piqirun.gen_optional_field 7 gen__string x.Node_proto.domain in
+ Piqirun.gen_record code (_input :: _output :: _name :: _op_type :: _attribute :: _doc_string :: _domain :: [])
+
+and gen__training_info_proto code x =
+ let _initialization = Piqirun.gen_optional_field 1 gen__graph_proto x.Training_info_proto.initialization in
+ let _algorithm = Piqirun.gen_optional_field 2 gen__graph_proto x.Training_info_proto.algorithm in
+ let _initialization_binding = Piqirun.gen_repeated_field 3 gen__string_string_entry_proto x.Training_info_proto.initialization_binding in
+ let _update_binding = Piqirun.gen_repeated_field 4 gen__string_string_entry_proto x.Training_info_proto.update_binding in
+ Piqirun.gen_record code (_initialization :: _algorithm :: _initialization_binding :: _update_binding :: [])
+
+and gen__model_proto code x =
+ let _ir_version = Piqirun.gen_optional_field 1 gen__protobuf_int64 x.Model_proto.ir_version in
+ let _producer_name = Piqirun.gen_optional_field 2 gen__string x.Model_proto.producer_name in
+ let _producer_version = Piqirun.gen_optional_field 3 gen__string x.Model_proto.producer_version in
+ let _domain = Piqirun.gen_optional_field 4 gen__string x.Model_proto.domain in
+ let _model_version = Piqirun.gen_optional_field 5 gen__protobuf_int64 x.Model_proto.model_version in
+ let _doc_string = Piqirun.gen_optional_field 6 gen__string x.Model_proto.doc_string in
+ let _graph = Piqirun.gen_optional_field 7 gen__graph_proto x.Model_proto.graph in
+ let _opset_import = Piqirun.gen_repeated_field 8 gen__operator_set_id_proto x.Model_proto.opset_import in
+ let _metadata_props = Piqirun.gen_repeated_field 14 gen__string_string_entry_proto x.Model_proto.metadata_props in
+ let _training_info = Piqirun.gen_repeated_field 20 gen__training_info_proto x.Model_proto.training_info in
+ let _functions = Piqirun.gen_repeated_field 25 gen__function_proto x.Model_proto.functions in
+ Piqirun.gen_record code (_ir_version :: _producer_name :: _producer_version :: _domain :: _model_version :: _doc_string :: _graph :: _opset_import :: _metadata_props :: _training_info :: _functions :: [])
+
+and gen__string_string_entry_proto code x =
+ let _key = Piqirun.gen_optional_field 1 gen__string x.String_string_entry_proto.key in
+ let _value = Piqirun.gen_optional_field 2 gen__string x.String_string_entry_proto.value in
+ Piqirun.gen_record code (_key :: _value :: [])
+
+and gen__tensor_annotation code x =
+ let _tensor_name = Piqirun.gen_optional_field 1 gen__string x.Tensor_annotation.tensor_name in
+ let _quant_parameter_tensor_names = Piqirun.gen_repeated_field 2 gen__string_string_entry_proto x.Tensor_annotation.quant_parameter_tensor_names in
+ Piqirun.gen_record code (_tensor_name :: _quant_parameter_tensor_names :: [])
+
+and gen__graph_proto code x =
+ let _node = Piqirun.gen_repeated_field 1 gen__node_proto x.Graph_proto.node in
+ let _name = Piqirun.gen_optional_field 2 gen__string x.Graph_proto.name in
+ let _initializer_ = Piqirun.gen_repeated_field 5 gen__tensor_proto x.Graph_proto.initializer_ in
+ let _doc_string = Piqirun.gen_optional_field 10 gen__string x.Graph_proto.doc_string in
+ let _input = Piqirun.gen_repeated_field 11 gen__value_info_proto x.Graph_proto.input in
+ let _output = Piqirun.gen_repeated_field 12 gen__value_info_proto x.Graph_proto.output in
+ let _value_info = Piqirun.gen_repeated_field 13 gen__value_info_proto x.Graph_proto.value_info in
+ let _quantization_annotation = Piqirun.gen_repeated_field 14 gen__tensor_annotation x.Graph_proto.quantization_annotation in
+ let _sparse_initializer = Piqirun.gen_repeated_field 15 gen__sparse_tensor_proto x.Graph_proto.sparse_initializer in
+ Piqirun.gen_record code (_node :: _name :: _initializer_ :: _doc_string :: _input :: _output :: _value_info :: _quantization_annotation :: _sparse_initializer :: [])
+
+and gen__tensor_proto code x =
+ let _dims = Piqirun.gen_repeated_field 1 gen__protobuf_int64 x.Tensor_proto.dims in
+ let _data_type = Piqirun.gen_optional_field 2 gen__protobuf_int32 x.Tensor_proto.data_type in
+ let _segment = Piqirun.gen_optional_field 3 gen__tensor_proto_segment x.Tensor_proto.segment in
+ let _float_data = Piqirun.gen_packed_repeated_field 4 packed_gen__float32 x.Tensor_proto.float_data in
+ let _int32_data = Piqirun.gen_packed_repeated_field 5 packed_gen__protobuf_int32 x.Tensor_proto.int32_data in
+ let _string_data = Piqirun.gen_repeated_field 6 gen__binary x.Tensor_proto.string_data in
+ let _int64_data = Piqirun.gen_packed_repeated_field 7 packed_gen__protobuf_int64 x.Tensor_proto.int64_data in
+ let _name = Piqirun.gen_optional_field 8 gen__string x.Tensor_proto.name in
+ let _raw_data = Piqirun.gen_optional_field 9 gen__binary x.Tensor_proto.raw_data in
+ let _double_data = Piqirun.gen_packed_repeated_field 10 packed_gen__float64 x.Tensor_proto.double_data in
+ let _uint64_data = Piqirun.gen_packed_repeated_field 11 packed_gen__uint64 x.Tensor_proto.uint64_data in
+ let _doc_string = Piqirun.gen_optional_field 12 gen__string x.Tensor_proto.doc_string in
+ let _externall_data = Piqirun.gen_repeated_field 13 gen__string_string_entry_proto x.Tensor_proto.externall_data in
+ let _data_location = Piqirun.gen_optional_field 14 gen__tensor_proto_data_location x.Tensor_proto.data_location in
+ Piqirun.gen_record code (_dims :: _data_type :: _segment :: _float_data :: _int32_data :: _string_data :: _int64_data :: _name :: _raw_data :: _double_data :: _uint64_data :: _doc_string :: _externall_data :: _data_location :: [])
+
+and gen__tensor_proto_segment code x =
+ let _begin_ = Piqirun.gen_optional_field 1 gen__protobuf_int64 x.Tensor_proto_segment.begin_ in
+ let _end_ = Piqirun.gen_optional_field 2 gen__protobuf_int64 x.Tensor_proto_segment.end_ in
+ Piqirun.gen_record code (_begin_ :: _end_ :: [])
+
+and gen__tensor_proto_data_type code x =
+ Piqirun.int32_to_signed_varint code (match x with
+ | `undefined -> 0l
+ | `float -> 1l
+ | `uint8 -> 2l
+ | `int8 -> 3l
+ | `uint16 -> 4l
+ | `int16 -> 5l
+ | `int32 -> 6l
+ | `int64 -> 7l
+ | `string -> 8l
+ | `bool -> 9l
+ | `float16 -> 10l
+ | `double -> 11l
+ | `uint32 -> 12l
+ | `uint64 -> 13l
+ | `complex64 -> 14l
+ | `complex128 -> 15l
+ | `bfloat16 -> 16l
+ )
+and packed_gen__tensor_proto_data_type x =
+ Piqirun.int32_to_packed_signed_varint (match x with
+ | `undefined -> 0l
+ | `float -> 1l
+ | `uint8 -> 2l
+ | `int8 -> 3l
+ | `uint16 -> 4l
+ | `int16 -> 5l
+ | `int32 -> 6l
+ | `int64 -> 7l
+ | `string -> 8l
+ | `bool -> 9l
+ | `float16 -> 10l
+ | `double -> 11l
+ | `uint32 -> 12l
+ | `uint64 -> 13l
+ | `complex64 -> 14l
+ | `complex128 -> 15l
+ | `bfloat16 -> 16l
+ )
+
+and gen__tensor_proto_data_location code x =
+ Piqirun.int32_to_signed_varint code (match x with
+ | `default -> 0l
+ | `externall -> 1l
+ )
+and packed_gen__tensor_proto_data_location x =
+ Piqirun.int32_to_packed_signed_varint (match x with
+ | `default -> 0l
+ | `externall -> 1l
+ )
+
+and gen__sparse_tensor_proto code x =
+ let _values = Piqirun.gen_optional_field 1 gen__tensor_proto x.Sparse_tensor_proto.values in
+ let _indices = Piqirun.gen_optional_field 2 gen__tensor_proto x.Sparse_tensor_proto.indices in
+ let _dims = Piqirun.gen_repeated_field 3 gen__protobuf_int64 x.Sparse_tensor_proto.dims in
+ Piqirun.gen_record code (_values :: _indices :: _dims :: [])
+
+and gen__tensor_shape_proto code x =
+ let _dim = Piqirun.gen_repeated_field 1 gen__tensor_shape_proto_dimension x.Tensor_shape_proto.dim in
+ Piqirun.gen_record code (_dim :: [])
+
+and gen__tensor_shape_proto_dimension code x =
+ let _dim_value = Piqirun.gen_optional_field 1 gen__protobuf_int64 x.Tensor_shape_proto_dimension.dim_value in
+ let _dim_param = Piqirun.gen_optional_field 2 gen__string x.Tensor_shape_proto_dimension.dim_param in
+ let _denotation = Piqirun.gen_optional_field 3 gen__string x.Tensor_shape_proto_dimension.denotation in
+ Piqirun.gen_record code (_dim_value :: _dim_param :: _denotation :: [])
+
+and gen__type_proto code x =
+ let _tensor_type = Piqirun.gen_optional_field 1 gen__type_proto_tensor x.Type_proto.tensor_type in
+ let _sequence_type = Piqirun.gen_optional_field 4 gen__type_proto_sequence x.Type_proto.sequence_type in
+ let _map_type = Piqirun.gen_optional_field 5 gen__type_proto_map x.Type_proto.map_type in
+ let _denotation = Piqirun.gen_optional_field 6 gen__string x.Type_proto.denotation in
+ let _sparse_tensor_type = Piqirun.gen_optional_field 8 gen__type_proto_sparse_tensor x.Type_proto.sparse_tensor_type in
+ let _optional_type = Piqirun.gen_optional_field 9 gen__type_proto_optional x.Type_proto.optional_type in
+ Piqirun.gen_record code (_tensor_type :: _sequence_type :: _map_type :: _denotation :: _sparse_tensor_type :: _optional_type :: [])
+
+and gen__type_proto_tensor code x =
+ let _elem_type = Piqirun.gen_optional_field 1 gen__protobuf_int32 x.Type_proto_tensor.elem_type in
+ let _shape = Piqirun.gen_optional_field 2 gen__tensor_shape_proto x.Type_proto_tensor.shape in
+ Piqirun.gen_record code (_elem_type :: _shape :: [])
+
+and gen__type_proto_sequence code x =
+ let _elem_type = Piqirun.gen_optional_field 1 gen__type_proto x.Type_proto_sequence.elem_type in
+ Piqirun.gen_record code (_elem_type :: [])
+
+and gen__type_proto_map code x =
+ let _key_type = Piqirun.gen_optional_field 1 gen__protobuf_int32 x.Type_proto_map.key_type in
+ let _value_type = Piqirun.gen_optional_field 2 gen__type_proto x.Type_proto_map.value_type in
+ Piqirun.gen_record code (_key_type :: _value_type :: [])
+
+and gen__type_proto_optional code x =
+ let _elem_type = Piqirun.gen_optional_field 1 gen__type_proto x.Type_proto_optional.elem_type in
+ Piqirun.gen_record code (_elem_type :: [])
+
+and gen__type_proto_sparse_tensor code x =
+ let _elem_type = Piqirun.gen_optional_field 1 gen__protobuf_int32 x.Type_proto_sparse_tensor.elem_type in
+ let _shape = Piqirun.gen_optional_field 2 gen__tensor_shape_proto x.Type_proto_sparse_tensor.shape in
+ Piqirun.gen_record code (_elem_type :: _shape :: [])
+
+and gen__operator_set_id_proto code x =
+ let _domain = Piqirun.gen_optional_field 1 gen__string x.Operator_set_id_proto.domain in
+ let _version = Piqirun.gen_optional_field 2 gen__protobuf_int64 x.Operator_set_id_proto.version in
+ Piqirun.gen_record code (_domain :: _version :: [])
+
+and gen__function_proto code x =
+ let _name = Piqirun.gen_optional_field 1 gen__string x.Function_proto.name in
+ let _input = Piqirun.gen_repeated_field 4 gen__string x.Function_proto.input in
+ let _output = Piqirun.gen_repeated_field 5 gen__string x.Function_proto.output in
+ let _attribute = Piqirun.gen_repeated_field 6 gen__string x.Function_proto.attribute in
+ let _node = Piqirun.gen_repeated_field 7 gen__node_proto x.Function_proto.node in
+ let _doc_string = Piqirun.gen_optional_field 8 gen__string x.Function_proto.doc_string in
+ let _opset_import = Piqirun.gen_repeated_field 9 gen__operator_set_id_proto x.Function_proto.opset_import in
+ let _domain = Piqirun.gen_optional_field 10 gen__string x.Function_proto.domain in
+ Piqirun.gen_record code (_name :: _input :: _output :: _attribute :: _node :: _doc_string :: _opset_import :: _domain :: [])
+
+and gen__version code x =
+ Piqirun.int32_to_signed_varint code (match x with
+ | `start_version -> 0l
+ | `ir_version_2017_10_10 -> 1l
+ | `ir_version_2017_10_30 -> 2l
+ | `ir_version_2017_11_3 -> 3l
+ | `ir_version_2019_1_22 -> 4l
+ | `ir_version_2019_3_18 -> 5l
+ | `ir_version_2019_9_19 -> 6l
+ | `ir_version_2020_5_8 -> 7l
+ | `ir_version -> 8l
+ )
+and packed_gen__version x =
+ Piqirun.int32_to_packed_signed_varint (match x with
+ | `start_version -> 0l
+ | `ir_version_2017_10_10 -> 1l
+ | `ir_version_2017_10_30 -> 2l
+ | `ir_version_2017_11_3 -> 3l
+ | `ir_version_2019_1_22 -> 4l
+ | `ir_version_2019_3_18 -> 5l
+ | `ir_version_2019_9_19 -> 6l
+ | `ir_version_2020_5_8 -> 7l
+ | `ir_version -> 8l
+ )
+
+and gen__operator_status code x =
+ Piqirun.int32_to_signed_varint code (match x with
+ | `experimental -> 0l
+ | `stable -> 1l
+ )
+and packed_gen__operator_status x =
+ Piqirun.int32_to_packed_signed_varint (match x with
+ | `experimental -> 0l
+ | `stable -> 1l
+ )
+
+
+let gen_int64 x = gen__int64 (-1) x
+let gen_int32 x = gen__int32 (-1) x
+let gen_string x = gen__string (-1) x
+let gen_float32 x = gen__float32 (-1) x
+let gen_protobuf_int64 x = gen__protobuf_int64 (-1) x
+let gen_binary x = gen__binary (-1) x
+let gen_protobuf_int32 x = gen__protobuf_int32 (-1) x
+let gen_float64 x = gen__float64 (-1) x
+let gen_uint64 x = gen__uint64 (-1) x
+let gen_attribute_proto x = gen__attribute_proto (-1) x
+let gen_attribute_proto_attribute_type x = gen__attribute_proto_attribute_type (-1) x
+let gen_value_info_proto x = gen__value_info_proto (-1) x
+let gen_node_proto x = gen__node_proto (-1) x
+let gen_training_info_proto x = gen__training_info_proto (-1) x
+let gen_model_proto x = gen__model_proto (-1) x
+let gen_string_string_entry_proto x = gen__string_string_entry_proto (-1) x
+let gen_tensor_annotation x = gen__tensor_annotation (-1) x
+let gen_graph_proto x = gen__graph_proto (-1) x
+let gen_tensor_proto x = gen__tensor_proto (-1) x
+let gen_tensor_proto_segment x = gen__tensor_proto_segment (-1) x
+let gen_tensor_proto_data_type x = gen__tensor_proto_data_type (-1) x
+let gen_tensor_proto_data_location x = gen__tensor_proto_data_location (-1) x
+let gen_sparse_tensor_proto x = gen__sparse_tensor_proto (-1) x
+let gen_tensor_shape_proto x = gen__tensor_shape_proto (-1) x
+let gen_tensor_shape_proto_dimension x = gen__tensor_shape_proto_dimension (-1) x
+let gen_type_proto x = gen__type_proto (-1) x
+let gen_type_proto_tensor x = gen__type_proto_tensor (-1) x
+let gen_type_proto_sequence x = gen__type_proto_sequence (-1) x
+let gen_type_proto_map x = gen__type_proto_map (-1) x
+let gen_type_proto_optional x = gen__type_proto_optional (-1) x
+let gen_type_proto_sparse_tensor x = gen__type_proto_sparse_tensor (-1) x
+let gen_operator_set_id_proto x = gen__operator_set_id_proto (-1) x
+let gen_function_proto x = gen__function_proto (-1) x
+let gen_version x = gen__version (-1) x
+let gen_operator_status x = gen__operator_status (-1) x
+
+
+let rec default_int64 () = 0L
+and default_int32 () = 0l
+and default_string () = ""
+and default_float32 () = 0.0
+and default_protobuf_int64 () = default_int64 ()
+and default_binary () = ""
+and default_protobuf_int32 () = default_int32 ()
+and default_float64 () = 0.0
+and default_uint64 () = 0L
+and default_attribute_proto () =
+ {
+ Attribute_proto.name = None;
+ Attribute_proto.f = None;
+ Attribute_proto.i = None;
+ Attribute_proto.s = None;
+ Attribute_proto.t = None;
+ Attribute_proto.g = None;
+ Attribute_proto.floats = [];
+ Attribute_proto.ints = [];
+ Attribute_proto.strings = [];
+ Attribute_proto.tensors = [];
+ Attribute_proto.graphs = [];
+ Attribute_proto.doc_string = None;
+ Attribute_proto.tp = None;
+ Attribute_proto.type_protos = [];
+ Attribute_proto.type_ = None;
+ Attribute_proto.ref_attr_name = None;
+ Attribute_proto.sparse_tensor = None;
+ Attribute_proto.sparse_tensors = [];
+ }
+and default_attribute_proto_attribute_type () = `undefined
+and default_value_info_proto () =
+ {
+ Value_info_proto.name = None;
+ Value_info_proto.type_ = None;
+ Value_info_proto.doc_string = None;
+ }
+and default_node_proto () =
+ {
+ Node_proto.input = [];
+ Node_proto.output = [];
+ Node_proto.name = None;
+ Node_proto.op_type = None;
+ Node_proto.attribute = [];
+ Node_proto.doc_string = None;
+ Node_proto.domain = None;
+ }
+and default_training_info_proto () =
+ {
+ Training_info_proto.initialization = None;
+ Training_info_proto.algorithm = None;
+ Training_info_proto.initialization_binding = [];
+ Training_info_proto.update_binding = [];
+ }
+and default_model_proto () =
+ {
+ Model_proto.ir_version = None;
+ Model_proto.producer_name = None;
+ Model_proto.producer_version = None;
+ Model_proto.domain = None;
+ Model_proto.model_version = None;
+ Model_proto.doc_string = None;
+ Model_proto.graph = None;
+ Model_proto.opset_import = [];
+ Model_proto.metadata_props = [];
+ Model_proto.training_info = [];
+ Model_proto.functions = [];
+ }
+and default_string_string_entry_proto () =
+ {
+ String_string_entry_proto.key = None;
+ String_string_entry_proto.value = None;
+ }
+and default_tensor_annotation () =
+ {
+ Tensor_annotation.tensor_name = None;
+ Tensor_annotation.quant_parameter_tensor_names = [];
+ }
+and default_graph_proto () =
+ {
+ Graph_proto.node = [];
+ Graph_proto.name = None;
+ Graph_proto.initializer_ = [];
+ Graph_proto.doc_string = None;
+ Graph_proto.input = [];
+ Graph_proto.output = [];
+ Graph_proto.value_info = [];
+ Graph_proto.quantization_annotation = [];
+ Graph_proto.sparse_initializer = [];
+ }
+and default_tensor_proto () =
+ {
+ Tensor_proto.dims = [];
+ Tensor_proto.data_type = None;
+ Tensor_proto.segment = None;
+ Tensor_proto.float_data = [];
+ Tensor_proto.int32_data = [];
+ Tensor_proto.string_data = [];
+ Tensor_proto.int64_data = [];
+ Tensor_proto.name = None;
+ Tensor_proto.raw_data = None;
+ Tensor_proto.double_data = [];
+ Tensor_proto.uint64_data = [];
+ Tensor_proto.doc_string = None;
+ Tensor_proto.externall_data = [];
+ Tensor_proto.data_location = None;
+ }
+and default_tensor_proto_segment () =
+ {
+ Tensor_proto_segment.begin_ = None;
+ Tensor_proto_segment.end_ = None;
+ }
+and default_tensor_proto_data_type () = `undefined
+and default_tensor_proto_data_location () = `default
+and default_sparse_tensor_proto () =
+ {
+ Sparse_tensor_proto.values = None;
+ Sparse_tensor_proto.indices = None;
+ Sparse_tensor_proto.dims = [];
+ }
+and default_tensor_shape_proto () =
+ {
+ Tensor_shape_proto.dim = [];
+ }
+and default_tensor_shape_proto_dimension () =
+ {
+ Tensor_shape_proto_dimension.dim_value = None;
+ Tensor_shape_proto_dimension.dim_param = None;
+ Tensor_shape_proto_dimension.denotation = None;
+ }
+and default_type_proto () =
+ {
+ Type_proto.tensor_type = None;
+ Type_proto.sequence_type = None;
+ Type_proto.map_type = None;
+ Type_proto.denotation = None;
+ Type_proto.sparse_tensor_type = None;
+ Type_proto.optional_type = None;
+ }
+and default_type_proto_tensor () =
+ {
+ Type_proto_tensor.elem_type = None;
+ Type_proto_tensor.shape = None;
+ }
+and default_type_proto_sequence () =
+ {
+ Type_proto_sequence.elem_type = None;
+ }
+and default_type_proto_map () =
+ {
+ Type_proto_map.key_type = None;
+ Type_proto_map.value_type = None;
+ }
+and default_type_proto_optional () =
+ {
+ Type_proto_optional.elem_type = None;
+ }
+and default_type_proto_sparse_tensor () =
+ {
+ Type_proto_sparse_tensor.elem_type = None;
+ Type_proto_sparse_tensor.shape = None;
+ }
+and default_operator_set_id_proto () =
+ {
+ Operator_set_id_proto.domain = None;
+ Operator_set_id_proto.version = None;
+ }
+and default_function_proto () =
+ {
+ Function_proto.name = None;
+ Function_proto.input = [];
+ Function_proto.output = [];
+ Function_proto.attribute = [];
+ Function_proto.node = [];
+ Function_proto.doc_string = None;
+ Function_proto.opset_import = [];
+ Function_proto.domain = None;
+ }
+and default_version () = `start_version
+and default_operator_status () = `experimental
+
+
+include Onnx_piqi
diff --git a/onnx_piqi.mli b/onnx_piqi.mli
new file mode 100644
index 00000000..97ea99f4
--- /dev/null
+++ b/onnx_piqi.mli
@@ -0,0 +1,718 @@
+module rec Onnx_piqi :
+ sig
+ type float32 = float
+ type float64 = float
+ type uint64 = int64
+ type protobuf_int64 = int64
+ type binary = string
+ type protobuf_int32 = int32
+ type attribute_proto_attribute_type =
+ [ `float
+ | `floats
+ | `graph
+ | `graphs
+ | `int
+ | `ints
+ | `sparse_tensor
+ | `sparse_tensors
+ | `string
+ | `strings
+ | `tensor
+ | `tensors
+ | `type_proto
+ | `type_protos
+ | `undefined ]
+ type tensor_proto_data_type =
+ [ `bfloat16
+ | `bool
+ | `complex128
+ | `complex64
+ | `double
+ | `float
+ | `float16
+ | `int16
+ | `int32
+ | `int64
+ | `int8
+ | `string
+ | `uint16
+ | `uint32
+ | `uint64
+ | `uint8
+ | `undefined ]
+ type tensor_proto_data_location = [ `default | `externall ]
+ type version =
+ [ `ir_version
+ | `ir_version_2017_10_10
+ | `ir_version_2017_10_30
+ | `ir_version_2017_11_3
+ | `ir_version_2019_1_22
+ | `ir_version_2019_3_18
+ | `ir_version_2019_9_19
+ | `ir_version_2020_5_8
+ | `start_version ]
+ type operator_status = [ `experimental | `stable ]
+ type attribute_proto = Attribute_proto.t
+ type value_info_proto = Value_info_proto.t
+ type node_proto = Node_proto.t
+ type training_info_proto = Training_info_proto.t
+ type model_proto = Model_proto.t
+ type string_string_entry_proto = String_string_entry_proto.t
+ type tensor_annotation = Tensor_annotation.t
+ type graph_proto = Graph_proto.t
+ type tensor_proto = Tensor_proto.t
+ type tensor_proto_segment = Tensor_proto_segment.t
+ type sparse_tensor_proto = Sparse_tensor_proto.t
+ type tensor_shape_proto = Tensor_shape_proto.t
+ type tensor_shape_proto_dimension = Tensor_shape_proto_dimension.t
+ type type_proto = Type_proto.t
+ type type_proto_tensor = Type_proto_tensor.t
+ type type_proto_sequence = Type_proto_sequence.t
+ type type_proto_map = Type_proto_map.t
+ type type_proto_optional = Type_proto_optional.t
+ type type_proto_sparse_tensor = Type_proto_sparse_tensor.t
+ type operator_set_id_proto = Operator_set_id_proto.t
+ type function_proto = Function_proto.t
+ end
+and Attribute_proto :
+ sig
+ type t = {
+ mutable name : string option;
+ mutable ref_attr_name : string option;
+ mutable doc_string : string option;
+ mutable type_ : Onnx_piqi.attribute_proto_attribute_type option;
+ mutable f : Onnx_piqi.float32 option;
+ mutable i : Onnx_piqi.protobuf_int64 option;
+ mutable s : Onnx_piqi.binary option;
+ mutable t : Onnx_piqi.tensor_proto option;
+ mutable g : Onnx_piqi.graph_proto option;
+ mutable sparse_tensor : Onnx_piqi.sparse_tensor_proto option;
+ mutable tp : Onnx_piqi.type_proto option;
+ mutable floats : Onnx_piqi.float32 list;
+ mutable ints : Onnx_piqi.protobuf_int64 list;
+ mutable strings : Onnx_piqi.binary list;
+ mutable tensors : Onnx_piqi.tensor_proto list;
+ mutable graphs : Onnx_piqi.graph_proto list;
+ mutable sparse_tensors : Onnx_piqi.sparse_tensor_proto list;
+ mutable type_protos : Onnx_piqi.type_proto list;
+ }
+ end
+and Value_info_proto :
+ sig
+ type t = {
+ mutable name : string option;
+ mutable type_ : Onnx_piqi.type_proto option;
+ mutable doc_string : string option;
+ }
+ end
+and Node_proto :
+ sig
+ type t = {
+ mutable input : string list;
+ mutable output : string list;
+ mutable name : string option;
+ mutable op_type : string option;
+ mutable domain : string option;
+ mutable attribute : Onnx_piqi.attribute_proto list;
+ mutable doc_string : string option;
+ }
+ end
+and Training_info_proto :
+ sig
+ type t = {
+ mutable initialization : Onnx_piqi.graph_proto option;
+ mutable algorithm : Onnx_piqi.graph_proto option;
+ mutable initialization_binding :
+ Onnx_piqi.string_string_entry_proto list;
+ mutable update_binding : Onnx_piqi.string_string_entry_proto list;
+ }
+ end
+and Model_proto :
+ sig
+ type t = {
+ mutable ir_version : Onnx_piqi.protobuf_int64 option;
+ mutable opset_import : Onnx_piqi.operator_set_id_proto list;
+ mutable producer_name : string option;
+ mutable producer_version : string option;
+ mutable domain : string option;
+ mutable model_version : Onnx_piqi.protobuf_int64 option;
+ mutable doc_string : string option;
+ mutable graph : Onnx_piqi.graph_proto option;
+ mutable metadata_props : Onnx_piqi.string_string_entry_proto list;
+ mutable training_info : Onnx_piqi.training_info_proto list;
+ mutable functions : Onnx_piqi.function_proto list;
+ }
+ end
+and String_string_entry_proto :
+ sig
+ type t = { mutable key : string option; mutable value : string option; }
+ end
+and Tensor_annotation :
+ sig
+ type t = {
+ mutable tensor_name : string option;
+ mutable quant_parameter_tensor_names :
+ Onnx_piqi.string_string_entry_proto list;
+ }
+ end
+and Graph_proto :
+ sig
+ type t = {
+ mutable node : Onnx_piqi.node_proto list;
+ mutable name : string option;
+ mutable initializer_ : Onnx_piqi.tensor_proto list;
+ mutable sparse_initializer : Onnx_piqi.sparse_tensor_proto list;
+ mutable doc_string : string option;
+ mutable input : Onnx_piqi.value_info_proto list;
+ mutable output : Onnx_piqi.value_info_proto list;
+ mutable value_info : Onnx_piqi.value_info_proto list;
+ mutable quantization_annotation : Onnx_piqi.tensor_annotation list;
+ }
+ end
+and Tensor_proto :
+ sig
+ type t = {
+ mutable dims : Onnx_piqi.protobuf_int64 list;
+ mutable data_type : Onnx_piqi.protobuf_int32 option;
+ mutable segment : Onnx_piqi.tensor_proto_segment option;
+ mutable float_data : Onnx_piqi.float32 list;
+ mutable int32_data : Onnx_piqi.protobuf_int32 list;
+ mutable string_data : Onnx_piqi.binary list;
+ mutable int64_data : Onnx_piqi.protobuf_int64 list;
+ mutable name : string option;
+ mutable doc_string : string option;
+ mutable raw_data : Onnx_piqi.binary option;
+ mutable externall_data : Onnx_piqi.string_string_entry_proto list;
+ mutable data_location : Onnx_piqi.tensor_proto_data_location option;
+ mutable double_data : Onnx_piqi.float64 list;
+ mutable uint64_data : Onnx_piqi.uint64 list;
+ }
+ end
+and Tensor_proto_segment :
+ sig
+ type t = {
+ mutable begin_ : Onnx_piqi.protobuf_int64 option;
+ mutable end_ : Onnx_piqi.protobuf_int64 option;
+ }
+ end
+and Sparse_tensor_proto :
+ sig
+ type t = {
+ mutable values : Onnx_piqi.tensor_proto option;
+ mutable indices : Onnx_piqi.tensor_proto option;
+ mutable dims : Onnx_piqi.protobuf_int64 list;
+ }
+ end
+and Tensor_shape_proto :
+ sig
+ type t = { mutable dim : Onnx_piqi.tensor_shape_proto_dimension list; }
+ end
+and Tensor_shape_proto_dimension :
+ sig
+ type t = {
+ mutable dim_value : Onnx_piqi.protobuf_int64 option;
+ mutable dim_param : string option;
+ mutable denotation : string option;
+ }
+ end
+and Type_proto :
+ sig
+ type t = {
+ mutable tensor_type : Onnx_piqi.type_proto_tensor option;
+ mutable sequence_type : Onnx_piqi.type_proto_sequence option;
+ mutable map_type : Onnx_piqi.type_proto_map option;
+ mutable optional_type : Onnx_piqi.type_proto_optional option;
+ mutable sparse_tensor_type : Onnx_piqi.type_proto_sparse_tensor option;
+ mutable denotation : string option;
+ }
+ end
+and Type_proto_tensor :
+ sig
+ type t = {
+ mutable elem_type : Onnx_piqi.protobuf_int32 option;
+ mutable shape : Onnx_piqi.tensor_shape_proto option;
+ }
+ end
+and Type_proto_sequence :
+ sig type t = { mutable elem_type : Onnx_piqi.type_proto option; } end
+and Type_proto_map :
+ sig
+ type t = {
+ mutable key_type : Onnx_piqi.protobuf_int32 option;
+ mutable value_type : Onnx_piqi.type_proto option;
+ }
+ end
+and Type_proto_optional :
+ sig type t = { mutable elem_type : Onnx_piqi.type_proto option; } end
+and Type_proto_sparse_tensor :
+ sig
+ type t = {
+ mutable elem_type : Onnx_piqi.protobuf_int32 option;
+ mutable shape : Onnx_piqi.tensor_shape_proto option;
+ }
+ end
+and Operator_set_id_proto :
+ sig
+ type t = {
+ mutable domain : string option;
+ mutable version : Onnx_piqi.protobuf_int64 option;
+ }
+ end
+and Function_proto :
+ sig
+ type t = {
+ mutable name : string option;
+ mutable input : string list;
+ mutable output : string list;
+ mutable attribute : string list;
+ mutable node : Onnx_piqi.node_proto list;
+ mutable doc_string : string option;
+ mutable opset_import : Onnx_piqi.operator_set_id_proto list;
+ mutable domain : string option;
+ }
+ end
+val parse_int64 : Piqirun.t -> int64
+val packed_parse_int64 : Piqirun.IBuf.t -> int64
+val parse_int32 : Piqirun.t -> int32
+val packed_parse_int32 : Piqirun.IBuf.t -> int32
+val parse_string : Piqirun.t -> string
+val parse_float32 : Piqirun.t -> Onnx_piqi.float32
+val packed_parse_float32 : Piqirun.IBuf.t -> Onnx_piqi.float32
+val parse_protobuf_int64 : Piqirun.t -> Onnx_piqi.protobuf_int64
+val packed_parse_protobuf_int64 : Piqirun.IBuf.t -> Onnx_piqi.protobuf_int64
+val parse_binary : Piqirun.t -> Onnx_piqi.binary
+val parse_protobuf_int32 : Piqirun.t -> Onnx_piqi.protobuf_int32
+val packed_parse_protobuf_int32 : Piqirun.IBuf.t -> Onnx_piqi.protobuf_int32
+val parse_float64 : Piqirun.t -> Onnx_piqi.float64
+val packed_parse_float64 : Piqirun.IBuf.t -> Onnx_piqi.float64
+val parse_uint64 : Piqirun.t -> Onnx_piqi.uint64
+val packed_parse_uint64 : Piqirun.IBuf.t -> Onnx_piqi.uint64
+val parse_attribute_proto : Piqirun.t -> Onnx_piqi.attribute_proto
+val parse_attribute_proto_attribute_type :
+ Piqirun.t -> Onnx_piqi.attribute_proto_attribute_type
+val packed_parse_attribute_proto_attribute_type :
+ Piqirun.IBuf.t ->
+ [> `float
+ | `floats
+ | `graph
+ | `graphs
+ | `int
+ | `ints
+ | `sparse_tensor
+ | `sparse_tensors
+ | `string
+ | `strings
+ | `tensor
+ | `tensors
+ | `type_proto
+ | `type_protos
+ | `undefined ]
+val parse_value_info_proto : Piqirun.t -> Onnx_piqi.value_info_proto
+val parse_node_proto : Piqirun.t -> Onnx_piqi.node_proto
+val parse_training_info_proto : Piqirun.t -> Onnx_piqi.training_info_proto
+val parse_model_proto : Piqirun.t -> Model_proto.t
+val parse_string_string_entry_proto :
+ Piqirun.t -> Onnx_piqi.string_string_entry_proto
+val parse_tensor_annotation : Piqirun.t -> Onnx_piqi.tensor_annotation
+val parse_graph_proto : Piqirun.t -> Onnx_piqi.graph_proto
+val parse_tensor_proto : Piqirun.t -> Onnx_piqi.tensor_proto
+val parse_tensor_proto_segment : Piqirun.t -> Onnx_piqi.tensor_proto_segment
+val parse_tensor_proto_data_type :
+ Piqirun.t ->
+ [> `bfloat16
+ | `bool
+ | `complex128
+ | `complex64
+ | `double
+ | `float
+ | `float16
+ | `int16
+ | `int32
+ | `int64
+ | `int8
+ | `string
+ | `uint16
+ | `uint32
+ | `uint64
+ | `uint8
+ | `undefined ]
+val packed_parse_tensor_proto_data_type :
+ Piqirun.IBuf.t ->
+ [> `bfloat16
+ | `bool
+ | `complex128
+ | `complex64
+ | `double
+ | `float
+ | `float16
+ | `int16
+ | `int32
+ | `int64
+ | `int8
+ | `string
+ | `uint16
+ | `uint32
+ | `uint64
+ | `uint8
+ | `undefined ]
+val parse_tensor_proto_data_location :
+ Piqirun.t -> Onnx_piqi.tensor_proto_data_location
+val packed_parse_tensor_proto_data_location :
+ Piqirun.IBuf.t -> [> `default | `externall ]
+val parse_sparse_tensor_proto : Piqirun.t -> Onnx_piqi.sparse_tensor_proto
+val parse_tensor_shape_proto : Piqirun.t -> Onnx_piqi.tensor_shape_proto
+val parse_tensor_shape_proto_dimension :
+ Piqirun.t -> Onnx_piqi.tensor_shape_proto_dimension
+val parse_type_proto : Piqirun.t -> Onnx_piqi.type_proto
+val parse_type_proto_tensor : Piqirun.t -> Onnx_piqi.type_proto_tensor
+val parse_type_proto_sequence : Piqirun.t -> Onnx_piqi.type_proto_sequence
+val parse_type_proto_map : Piqirun.t -> Onnx_piqi.type_proto_map
+val parse_type_proto_optional : Piqirun.t -> Onnx_piqi.type_proto_optional
+val parse_type_proto_sparse_tensor :
+ Piqirun.t -> Onnx_piqi.type_proto_sparse_tensor
+val parse_operator_set_id_proto :
+ Piqirun.t -> Onnx_piqi.operator_set_id_proto
+val parse_function_proto : Piqirun.t -> Onnx_piqi.function_proto
+val parse_version :
+ Piqirun.t ->
+ [> `ir_version
+ | `ir_version_2017_10_10
+ | `ir_version_2017_10_30
+ | `ir_version_2017_11_3
+ | `ir_version_2019_1_22
+ | `ir_version_2019_3_18
+ | `ir_version_2019_9_19
+ | `ir_version_2020_5_8
+ | `start_version ]
+val packed_parse_version :
+ Piqirun.IBuf.t ->
+ [> `ir_version
+ | `ir_version_2017_10_10
+ | `ir_version_2017_10_30
+ | `ir_version_2017_11_3
+ | `ir_version_2019_1_22
+ | `ir_version_2019_3_18
+ | `ir_version_2019_9_19
+ | `ir_version_2020_5_8
+ | `start_version ]
+val parse_operator_status : Piqirun.t -> [> `experimental | `stable ]
+val packed_parse_operator_status :
+ Piqirun.IBuf.t -> [> `experimental | `stable ]
+val gen__int64 : int -> int64 -> Piqirun.OBuf.t
+val packed_gen__int64 : int64 -> Piqirun.OBuf.t
+val gen__int32 : int -> int32 -> Piqirun.OBuf.t
+val packed_gen__int32 : int32 -> Piqirun.OBuf.t
+val gen__string : int -> string -> Piqirun.OBuf.t
+val gen__float32 : int -> Onnx_piqi.float32 -> Piqirun.OBuf.t
+val packed_gen__float32 : Onnx_piqi.float32 -> Piqirun.OBuf.t
+val gen__protobuf_int64 : int -> Onnx_piqi.protobuf_int64 -> Piqirun.OBuf.t
+val packed_gen__protobuf_int64 : Onnx_piqi.protobuf_int64 -> Piqirun.OBuf.t
+val gen__binary : int -> Onnx_piqi.binary -> Piqirun.OBuf.t
+val gen__protobuf_int32 : int -> Int32.t -> Piqirun.OBuf.t
+val packed_gen__protobuf_int32 : Int32.t -> Piqirun.OBuf.t
+val gen__float64 : int -> float -> Piqirun.OBuf.t
+val packed_gen__float64 : Onnx_piqi.float64 -> Piqirun.OBuf.t
+val gen__uint64 : int -> int64 -> Piqirun.OBuf.t
+val packed_gen__uint64 : Onnx_piqi.uint64 -> Piqirun.OBuf.t
+val gen__attribute_proto : int -> Onnx_piqi.attribute_proto -> Piqirun.OBuf.t
+val gen__attribute_proto_attribute_type :
+ int -> Onnx_piqi.attribute_proto_attribute_type -> Piqirun.OBuf.t
+val packed_gen__attribute_proto_attribute_type :
+ [< `float
+ | `floats
+ | `graph
+ | `graphs
+ | `int
+ | `ints
+ | `sparse_tensor
+ | `sparse_tensors
+ | `string
+ | `strings
+ | `tensor
+ | `tensors
+ | `type_proto
+ | `type_protos
+ | `undefined ] ->
+ Piqirun.OBuf.t
+val gen__value_info_proto :
+ int -> Onnx_piqi.value_info_proto -> Piqirun.OBuf.t
+val gen__node_proto : int -> Onnx_piqi.node_proto -> Piqirun.OBuf.t
+val gen__training_info_proto :
+ int -> Onnx_piqi.training_info_proto -> Piqirun.OBuf.t
+val gen__model_proto : int -> Model_proto.t -> Piqirun.OBuf.t
+val gen__string_string_entry_proto :
+ int -> Onnx_piqi.string_string_entry_proto -> Piqirun.OBuf.t
+val gen__tensor_annotation :
+ int -> Onnx_piqi.tensor_annotation -> Piqirun.OBuf.t
+val gen__graph_proto : int -> Onnx_piqi.graph_proto -> Piqirun.OBuf.t
+val gen__tensor_proto : int -> Onnx_piqi.tensor_proto -> Piqirun.OBuf.t
+val gen__tensor_proto_segment :
+ int -> Onnx_piqi.tensor_proto_segment -> Piqirun.OBuf.t
+val gen__tensor_proto_data_type :
+ int ->
+ [< `bfloat16
+ | `bool
+ | `complex128
+ | `complex64
+ | `double
+ | `float
+ | `float16
+ | `int16
+ | `int32
+ | `int64
+ | `int8
+ | `string
+ | `uint16
+ | `uint32
+ | `uint64
+ | `uint8
+ | `undefined ] ->
+ Piqirun.OBuf.t
+val packed_gen__tensor_proto_data_type :
+ [< `bfloat16
+ | `bool
+ | `complex128
+ | `complex64
+ | `double
+ | `float
+ | `float16
+ | `int16
+ | `int32
+ | `int64
+ | `int8
+ | `string
+ | `uint16
+ | `uint32
+ | `uint64
+ | `uint8
+ | `undefined ] ->
+ Piqirun.OBuf.t
+val gen__tensor_proto_data_location :
+ int -> Onnx_piqi.tensor_proto_data_location -> Piqirun.OBuf.t
+val packed_gen__tensor_proto_data_location :
+ [< `default | `externall ] -> Piqirun.OBuf.t
+val gen__sparse_tensor_proto :
+ int -> Onnx_piqi.sparse_tensor_proto -> Piqirun.OBuf.t
+val gen__tensor_shape_proto :
+ int -> Onnx_piqi.tensor_shape_proto -> Piqirun.OBuf.t
+val gen__tensor_shape_proto_dimension :
+ int -> Onnx_piqi.tensor_shape_proto_dimension -> Piqirun.OBuf.t
+val gen__type_proto : int -> Onnx_piqi.type_proto -> Piqirun.OBuf.t
+val gen__type_proto_tensor :
+ int -> Onnx_piqi.type_proto_tensor -> Piqirun.OBuf.t
+val gen__type_proto_sequence :
+ int -> Onnx_piqi.type_proto_sequence -> Piqirun.OBuf.t
+val gen__type_proto_map : int -> Onnx_piqi.type_proto_map -> Piqirun.OBuf.t
+val gen__type_proto_optional :
+ int -> Onnx_piqi.type_proto_optional -> Piqirun.OBuf.t
+val gen__type_proto_sparse_tensor :
+ int -> Onnx_piqi.type_proto_sparse_tensor -> Piqirun.OBuf.t
+val gen__operator_set_id_proto :
+ int -> Onnx_piqi.operator_set_id_proto -> Piqirun.OBuf.t
+val gen__function_proto : int -> Onnx_piqi.function_proto -> Piqirun.OBuf.t
+val gen__version :
+ int ->
+ [< `ir_version
+ | `ir_version_2017_10_10
+ | `ir_version_2017_10_30
+ | `ir_version_2017_11_3
+ | `ir_version_2019_1_22
+ | `ir_version_2019_3_18
+ | `ir_version_2019_9_19
+ | `ir_version_2020_5_8
+ | `start_version ] ->
+ Piqirun.OBuf.t
+val packed_gen__version :
+ [< `ir_version
+ | `ir_version_2017_10_10
+ | `ir_version_2017_10_30
+ | `ir_version_2017_11_3
+ | `ir_version_2019_1_22
+ | `ir_version_2019_3_18
+ | `ir_version_2019_9_19
+ | `ir_version_2020_5_8
+ | `start_version ] ->
+ Piqirun.OBuf.t
+val gen__operator_status :
+ int -> [< `experimental | `stable ] -> Piqirun.OBuf.t
+val packed_gen__operator_status :
+ [< `experimental | `stable ] -> Piqirun.OBuf.t
+val gen_int64 : int64 -> Piqirun.OBuf.t
+val gen_int32 : int32 -> Piqirun.OBuf.t
+val gen_string : string -> Piqirun.OBuf.t
+val gen_float32 : Onnx_piqi.float32 -> Piqirun.OBuf.t
+val gen_protobuf_int64 : Onnx_piqi.protobuf_int64 -> Piqirun.OBuf.t
+val gen_binary : Onnx_piqi.binary -> Piqirun.OBuf.t
+val gen_protobuf_int32 : Int32.t -> Piqirun.OBuf.t
+val gen_float64 : float -> Piqirun.OBuf.t
+val gen_uint64 : int64 -> Piqirun.OBuf.t
+val gen_attribute_proto : Onnx_piqi.attribute_proto -> Piqirun.OBuf.t
+val gen_attribute_proto_attribute_type :
+ Onnx_piqi.attribute_proto_attribute_type -> Piqirun.OBuf.t
+val gen_value_info_proto : Onnx_piqi.value_info_proto -> Piqirun.OBuf.t
+val gen_node_proto : Onnx_piqi.node_proto -> Piqirun.OBuf.t
+val gen_training_info_proto : Onnx_piqi.training_info_proto -> Piqirun.OBuf.t
+val gen_model_proto : Model_proto.t -> Piqirun.OBuf.t
+val gen_string_string_entry_proto :
+ Onnx_piqi.string_string_entry_proto -> Piqirun.OBuf.t
+val gen_tensor_annotation : Onnx_piqi.tensor_annotation -> Piqirun.OBuf.t
+val gen_graph_proto : Onnx_piqi.graph_proto -> Piqirun.OBuf.t
+val gen_tensor_proto : Onnx_piqi.tensor_proto -> Piqirun.OBuf.t
+val gen_tensor_proto_segment :
+ Onnx_piqi.tensor_proto_segment -> Piqirun.OBuf.t
+val gen_tensor_proto_data_type :
+ [< `bfloat16
+ | `bool
+ | `complex128
+ | `complex64
+ | `double
+ | `float
+ | `float16
+ | `int16
+ | `int32
+ | `int64
+ | `int8
+ | `string
+ | `uint16
+ | `uint32
+ | `uint64
+ | `uint8
+ | `undefined ] ->
+ Piqirun.OBuf.t
+val gen_tensor_proto_data_location :
+ Onnx_piqi.tensor_proto_data_location -> Piqirun.OBuf.t
+val gen_sparse_tensor_proto : Onnx_piqi.sparse_tensor_proto -> Piqirun.OBuf.t
+val gen_tensor_shape_proto : Onnx_piqi.tensor_shape_proto -> Piqirun.OBuf.t
+val gen_tensor_shape_proto_dimension :
+ Onnx_piqi.tensor_shape_proto_dimension -> Piqirun.OBuf.t
+val gen_type_proto : Onnx_piqi.type_proto -> Piqirun.OBuf.t
+val gen_type_proto_tensor : Onnx_piqi.type_proto_tensor -> Piqirun.OBuf.t
+val gen_type_proto_sequence : Onnx_piqi.type_proto_sequence -> Piqirun.OBuf.t
+val gen_type_proto_map : Onnx_piqi.type_proto_map -> Piqirun.OBuf.t
+val gen_type_proto_optional : Onnx_piqi.type_proto_optional -> Piqirun.OBuf.t
+val gen_type_proto_sparse_tensor :
+ Onnx_piqi.type_proto_sparse_tensor -> Piqirun.OBuf.t
+val gen_operator_set_id_proto :
+ Onnx_piqi.operator_set_id_proto -> Piqirun.OBuf.t
+val gen_function_proto : Onnx_piqi.function_proto -> Piqirun.OBuf.t
+val gen_version :
+ [< `ir_version
+ | `ir_version_2017_10_10
+ | `ir_version_2017_10_30
+ | `ir_version_2017_11_3
+ | `ir_version_2019_1_22
+ | `ir_version_2019_3_18
+ | `ir_version_2019_9_19
+ | `ir_version_2020_5_8
+ | `start_version ] ->
+ Piqirun.OBuf.t
+val gen_operator_status : [< `experimental | `stable ] -> Piqirun.OBuf.t
+val default_int64 : unit -> int64
+val default_int32 : unit -> int32
+val default_string : unit -> string
+val default_float32 : unit -> float
+val default_protobuf_int64 : unit -> int64
+val default_binary : unit -> string
+val default_protobuf_int32 : unit -> int32
+val default_float64 : unit -> float
+val default_uint64 : unit -> int64
+val default_attribute_proto : unit -> Attribute_proto.t
+val default_attribute_proto_attribute_type : unit -> [> `undefined ]
+val default_value_info_proto : unit -> Value_info_proto.t
+val default_node_proto : unit -> Node_proto.t
+val default_training_info_proto : unit -> Training_info_proto.t
+val default_model_proto : unit -> Model_proto.t
+val default_string_string_entry_proto : unit -> String_string_entry_proto.t
+val default_tensor_annotation : unit -> Tensor_annotation.t
+val default_graph_proto : unit -> Graph_proto.t
+val default_tensor_proto : unit -> Tensor_proto.t
+val default_tensor_proto_segment : unit -> Tensor_proto_segment.t
+val default_tensor_proto_data_type : unit -> [> `undefined ]
+val default_tensor_proto_data_location : unit -> [> `default ]
+val default_sparse_tensor_proto : unit -> Sparse_tensor_proto.t
+val default_tensor_shape_proto : unit -> Tensor_shape_proto.t
+val default_tensor_shape_proto_dimension :
+ unit -> Tensor_shape_proto_dimension.t
+val default_type_proto : unit -> Type_proto.t
+val default_type_proto_tensor : unit -> Type_proto_tensor.t
+val default_type_proto_sequence : unit -> Type_proto_sequence.t
+val default_type_proto_map : unit -> Type_proto_map.t
+val default_type_proto_optional : unit -> Type_proto_optional.t
+val default_type_proto_sparse_tensor : unit -> Type_proto_sparse_tensor.t
+val default_operator_set_id_proto : unit -> Operator_set_id_proto.t
+val default_function_proto : unit -> Function_proto.t
+val default_version : unit -> [> `start_version ]
+val default_operator_status : unit -> [> `experimental ]
+type float32 = float
+type float64 = float
+type uint64 = int64
+type protobuf_int64 = int64
+type binary = string
+type protobuf_int32 = int32
+type attribute_proto_attribute_type =
+ [ `float
+ | `floats
+ | `graph
+ | `graphs
+ | `int
+ | `ints
+ | `sparse_tensor
+ | `sparse_tensors
+ | `string
+ | `strings
+ | `tensor
+ | `tensors
+ | `type_proto
+ | `type_protos
+ | `undefined ]
+type tensor_proto_data_type =
+ [ `bfloat16
+ | `bool
+ | `complex128
+ | `complex64
+ | `double
+ | `float
+ | `float16
+ | `int16
+ | `int32
+ | `int64
+ | `int8
+ | `string
+ | `uint16
+ | `uint32
+ | `uint64
+ | `uint8
+ | `undefined ]
+type tensor_proto_data_location = [ `default | `externall ]
+type version =
+ [ `ir_version
+ | `ir_version_2017_10_10
+ | `ir_version_2017_10_30
+ | `ir_version_2017_11_3
+ | `ir_version_2019_1_22
+ | `ir_version_2019_3_18
+ | `ir_version_2019_9_19
+ | `ir_version_2020_5_8
+ | `start_version ]
+type operator_status = [ `experimental | `stable ]
+type attribute_proto = Attribute_proto.t
+type value_info_proto = Value_info_proto.t
+type node_proto = Node_proto.t
+type training_info_proto = Training_info_proto.t
+type model_proto = Model_proto.t
+type string_string_entry_proto = String_string_entry_proto.t
+type tensor_annotation = Tensor_annotation.t
+type graph_proto = Graph_proto.t
+type tensor_proto = Tensor_proto.t
+type tensor_proto_segment = Tensor_proto_segment.t
+type sparse_tensor_proto = Sparse_tensor_proto.t
+type tensor_shape_proto = Tensor_shape_proto.t
+type tensor_shape_proto_dimension = Tensor_shape_proto_dimension.t
+type type_proto = Type_proto.t
+type type_proto_tensor = Type_proto_tensor.t
+type type_proto_sequence = Type_proto_sequence.t
+type type_proto_map = Type_proto_map.t
+type type_proto_optional = Type_proto_optional.t
+type type_proto_sparse_tensor = Type_proto_sparse_tensor.t
+type operator_set_id_proto = Operator_set_id_proto.t
+type function_proto = Function_proto.t
diff --git a/standalone/dune b/standalone/dune
index 7eea4820..60266cff 100644
--- a/standalone/dune
+++ b/standalone/dune
@@ -12,4 +12,10 @@
(libraries menhirLib yojson cmdliner logs logs.cli logs.fmt fmt.tty base unix str ppx_deriving_yojson.runtime)
(preprocess (pps ppx_deriving_yojson ppx_deriving.show ppx_deriving.ord ppx_deriving.eq))
(package caisar)
+ (libraries menhirLib yojson cmdliner logs logs.cli
+ ocplib-endian logs.fmt fmt.tty base unix str
+ ppx_deriving_yojson.runtime piqirun.pb zarith
+ stdio)
+ (preprocess (pps ppx_deriving_yojson ppx_deriving.show ppx_deriving.ord ppx_deriving.eq))
+ (package caisar)
)
--
GitLab