diff --git a/Makefile b/Makefile
index b9f6d170932ea621b1b18f44a6cf2747023130fd..88393974dd9593a87f098410a54d45a4879f8bcf 100644
--- a/Makefile
+++ b/Makefile
@@ -1,5 +1,5 @@
all:
- dune build --root=. @install caisar.opam nnet.opam
+ dune build --root=. @install caisar.opam nnet.opam onnx.opam
test:
dune runtest --root=.
diff --git a/caisar.opam b/caisar.opam
index 828f386599d8a7747a58e0c23ba407f12557780f..b2a9fd775a6f73d2d6edca3561298f9f1f4f7236 100644
--- a/caisar.opam
+++ b/caisar.opam
@@ -4,13 +4,12 @@ version: "0.1"
synopsis: "Framework for neural network verification"
depends: [
"ocaml" {>= "4.10"}
- "dune" {>= "2.9" & >= "2.9.1"}
"dune-site" {>= "2.9.0"}
"piqi" {>= "0.7.6"}
"piqilib" {>= "0.6.14"}
"zarith" {>= "1.7"}
"ocplib-endian" {>= "1.0"}
- "dune" {>= "2.7.1"}
+ "dune" {>= "2.9" & >= "2.7.1"}
"base" {>= "v0.14.0"}
"stdio" {>= "v0.14.0"}
"cmdliner" {>= "1.0.4"}
@@ -23,6 +22,7 @@ depends: [
"csv" {>= "2.4"}
"why3" {>= "1.4"}
"re"
+ "onnx"
"odoc" {with-doc}
]
build: [
diff --git a/dune-project b/dune-project
index 949ae81a3bcd226648af8e38c1f01c4a4fc97495..c0e7bd02c1cfce445db33902de2f142a99a6afde 100644
--- a/dune-project
+++ b/dune-project
@@ -14,7 +14,6 @@
(synopsis "Framework for neural network verification")
(depends
(ocaml (>= 4.10))
- (dune (>= 2.9.1))
(dune-site (>= 2.9.0))
(piqi (>= 0.7.6))
(piqilib (>= 0.6.14))
@@ -33,6 +32,7 @@
(csv (>= 2.4))
(why3 (>= 1.4))
re
+ onnx
)
(sites
(share stdlib)
@@ -57,5 +57,6 @@
(ocaml (>= 4.10))
(dune (>= 2.9.1))
(base (>= v0.14.0))
+ (ocaml-protoc-plugin (= 4.2.0))
)
)
diff --git a/lib/onnx/dune b/lib/onnx/dune
index b6defdaae529d3da1f7f80a934d3e4eac796ae90..67bb0943c99bbe97d57ac1ccbf9261f0f9860edb 100644
--- a/lib/onnx/dune
+++ b/lib/onnx/dune
@@ -1,15 +1,15 @@
(library
(name onnx)
(public_name onnx)
- (libraries base csv piqirun.pb)
+ (libraries base stdio piqirun.pb ocaml-protoc-plugin)
(synopsis "ONNX parser"))
(rule
- (targets onnx_piqi.ml onnx_piqi.mli)
+ (targets onnx_protoc.ml onnx_protoc.mli)
(action
(run ./generate_onnx_interface.sh)
)
(deps
- onnx.proto
+ onnx_protoc.proto
generate_onnx_interface.sh)
(mode
promote))
diff --git a/lib/onnx/generate_onnx_interface.sh b/lib/onnx/generate_onnx_interface.sh
index 3925d5189410e11aa2f305b3f788884de5dd3e19..8b38cd016e29e570e9e2fd6fda2249fa7fa4452a 100755
--- a/lib/onnx/generate_onnx_interface.sh
+++ b/lib/onnx/generate_onnx_interface.sh
@@ -1,7 +1,3 @@
#!/bin/sh
-
-piqi of-proto onnx.proto
-sed -ie 's/-START-VERSION/START-VERSION/' onnx.proto.piqi
-piqic-ocaml onnx.proto.piqi
-sed -ie 's/external/externall/g' onnx_piqi.ml
-ocamlfind ocamlc -package piqirun.pb -i onnx_piqi.ml > onnx_piqi.mli
+protoc --ocaml_out=. onnx_protoc.proto
+ocamlfind ocamlc -package ocaml-protoc-plugin -i onnx_protoc.ml > onnx_protoc.mli
diff --git a/lib/onnx/onnx.ml b/lib/onnx/onnx.ml
index bd0947f4d2ad66ebe37b56855583804841ea4299..4b169b93f19fa002d2eb0dfe1b5778768f47be08 100644
--- a/lib/onnx/onnx.ml
+++ b/lib/onnx/onnx.ml
@@ -7,26 +7,62 @@
open Base
module Format = Caml.Format
module Fun = Caml.Fun
-module Opiqi = Onnx_piqi
+module Oproto = Onnx_protoc (* Autogenerated during compilation *)
-type t = Opiqi.Model_proto.t
+module Oprotom = Oproto.Onnx.ModelProto
+
+type t = Oproto.Onnx.ModelProto.t [@@deriving show]
(* ONNX format handling. *)
let parse_in_channel in_channel =
let open Result in
try
+ let buf = Stdio.In_channel.input_all in_channel in
+ let reader = Ocaml_protoc_plugin.Reader.create buf in
let res =
- let buf = Piqirun.init_from_channel in_channel in
- Opiqi.parse_model_proto buf
+ match Oprotom.from_proto reader with
+ | Ok r -> Ok r
+ | _ -> Error "Error parsing protobuf"
in
- Ok res
+ res
with
| Sys_error s -> Error s
| Failure msg -> Error (Format.sprintf "Unexpected error: %s." msg)
+let get_input_output_shape (model : t) =
+ let ins, outs =
+ match model.graph with
+ | Some g -> (Some g.input, Some g.output)
+ | None -> (None, None)
+ in
+ let get_nested_dims (s : Oproto.Onnx.ValueInfoProto.t list) =
+ match List.nth s 0 with
+ | Some { type' = Some { value = `Tensor_type { shape = Some v; _ }; _ }; _ }
+ ->
+ v
+ | _ -> []
+ in
+ let input_shape, output_shape =
+ match (ins, outs) with
+ | Some i, Some o -> (get_nested_dims i, get_nested_dims o)
+ | _ -> ([], [])
+ in
+ (* TODO: here we only get the flattened dimension of inputs and outputs, but
+ more interesting parsing could be done later on. *)
+ let flattened_dim (dim : Oproto.Onnx.TensorShapeProto.Dimension.t list) =
+ List.fold ~init:1 dim ~f:(fun acc x ->
+ match x.value with
+ | `Dim_value v -> acc * v
+ | `Dim_param _ -> acc
+ | `not_set -> acc)
+ in
+ let input_flat_dim = flattened_dim input_shape in
+ let output_flat_dim = flattened_dim output_shape in
+ (input_flat_dim, output_flat_dim)
+
let parse filename =
- let in_channel = Stdlib.open_in_bin filename in
+ let in_channel = Stdlib.open_in filename in
Fun.protect
~finally:(fun () -> Stdlib.close_in in_channel)
(fun () -> parse_in_channel in_channel)
diff --git a/lib/onnx/onnx.mli b/lib/onnx/onnx.mli
index ab8a61c3fe8f388deb1ade0d942a260725722477..b93ffd3b5699db96057b907c9ccb07e5051b1227 100644
--- a/lib/onnx/onnx.mli
+++ b/lib/onnx/onnx.mli
@@ -4,9 +4,14 @@
(* *)
(**************************************************************************)
-module Opiqi = Onnx_piqi
+module Oproto = Onnx_protoc
-type t = Onnx_piqi.Model_proto.t
+type t = Onnx_protoc.Onnx.ModelProto.t
+[@@deriving show]
val parse : string -> (t, string) Result.t
(** Parse an ONNX file. *)
+
+val get_input_output_shape : t -> int*int
+(** Get the flattened input and output shape of a neural
+ network in an ONNX model. *)
diff --git a/lib/onnx/onnx.proto.piqi b/lib/onnx/onnx.proto.piqi
deleted file mode 100644
index 7a90f70424e4171ad65f4e4cbb6ec7874a86d9e1..0000000000000000000000000000000000000000
--- a/lib/onnx/onnx.proto.piqi
+++ /dev/null
@@ -1,942 +0,0 @@
-.protobuf-package "onnx"
-
-.record [
- .name AttributeProto
- .field [
- .name name
- .type string
- .optional
- .code 1
- ]
- .field [
- .name ref-attr-name
- .type string
- .optional
- .code 21
- ]
- .field [
- .name doc-string
- .type string
- .optional
- .code 13
- ]
- .field [
- .name type
- .type AttributeProto-AttributeType
- .optional
- .code 20
- ]
- .field [
- .name f
- .type float32
- .optional
- .code 2
- ]
- .field [
- .name i
- .type protobuf-int64
- .optional
- .code 3
- ]
- .field [
- .name s
- .type binary
- .optional
- .code 4
- ]
- .field [
- .name t
- .type TensorProto
- .optional
- .code 5
- ]
- .field [
- .name g
- .type GraphProto
- .optional
- .code 6
- ]
- .field [
- .name sparse-tensor
- .type SparseTensorProto
- .optional
- .code 22
- ]
- .field [
- .name tp
- .type TypeProto
- .optional
- .code 14
- ]
- .field [
- .name floats
- .type float32
- .repeated
- .code 7
- ]
- .field [
- .name ints
- .type protobuf-int64
- .repeated
- .code 8
- ]
- .field [
- .name strings
- .type binary
- .repeated
- .code 9
- ]
- .field [
- .name tensors
- .type TensorProto
- .repeated
- .code 10
- ]
- .field [
- .name graphs
- .type GraphProto
- .repeated
- .code 11
- ]
- .field [
- .name sparse-tensors
- .type SparseTensorProto
- .repeated
- .code 23
- ]
- .field [
- .name type-protos
- .type TypeProto
- .repeated
- .code 15
- ]
-]
-
-.enum [
- .name AttributeProto-AttributeType
- .option [
- .name UNDEFINED
- .code 0
- .protobuf-name "AttributeProto_UNDEFINED"
- ]
- .option [
- .name FLOAT
- .code 1
- .protobuf-name "AttributeProto_FLOAT"
- ]
- .option [
- .name INT
- .code 2
- .protobuf-name "AttributeProto_INT"
- ]
- .option [
- .name STRING
- .code 3
- .protobuf-name "AttributeProto_STRING"
- ]
- .option [
- .name TENSOR
- .code 4
- .protobuf-name "AttributeProto_TENSOR"
- ]
- .option [
- .name GRAPH
- .code 5
- .protobuf-name "AttributeProto_GRAPH"
- ]
- .option [
- .name SPARSE-TENSOR
- .code 11
- .protobuf-name "AttributeProto_SPARSE_TENSOR"
- ]
- .option [
- .name TYPE-PROTO
- .code 13
- .protobuf-name "AttributeProto_TYPE_PROTO"
- ]
- .option [
- .name FLOATS
- .code 6
- .protobuf-name "AttributeProto_FLOATS"
- ]
- .option [
- .name INTS
- .code 7
- .protobuf-name "AttributeProto_INTS"
- ]
- .option [
- .name STRINGS
- .code 8
- .protobuf-name "AttributeProto_STRINGS"
- ]
- .option [
- .name TENSORS
- .code 9
- .protobuf-name "AttributeProto_TENSORS"
- ]
- .option [
- .name GRAPHS
- .code 10
- .protobuf-name "AttributeProto_GRAPHS"
- ]
- .option [
- .name SPARSE-TENSORS
- .code 12
- .protobuf-name "AttributeProto_SPARSE_TENSORS"
- ]
- .option [
- .name TYPE-PROTOS
- .code 14
- .protobuf-name "AttributeProto_TYPE_PROTOS"
- ]
-]
-
-.record [
- .name ValueInfoProto
- .field [
- .name name
- .type string
- .optional
- .code 1
- ]
- .field [
- .name type
- .type TypeProto
- .optional
- .code 2
- ]
- .field [
- .name doc-string
- .type string
- .optional
- .code 3
- ]
-]
-
-.record [
- .name NodeProto
- .field [
- .name input
- .type string
- .repeated
- .code 1
- ]
- .field [
- .name output
- .type string
- .repeated
- .code 2
- ]
- .field [
- .name name
- .type string
- .optional
- .code 3
- ]
- .field [
- .name op-type
- .type string
- .optional
- .code 4
- ]
- .field [
- .name domain
- .type string
- .optional
- .code 7
- ]
- .field [
- .name attribute
- .type AttributeProto
- .repeated
- .code 5
- ]
- .field [
- .name doc-string
- .type string
- .optional
- .code 6
- ]
-]
-
-.record [
- .name TrainingInfoProto
- .field [
- .name initialization
- .type GraphProto
- .optional
- .code 1
- ]
- .field [
- .name algorithm
- .type GraphProto
- .optional
- .code 2
- ]
- .field [
- .name initialization-binding
- .type StringStringEntryProto
- .repeated
- .code 3
- ]
- .field [
- .name update-binding
- .type StringStringEntryProto
- .repeated
- .code 4
- ]
-]
-
-.record [
- .name ModelProto
- .field [
- .name ir-version
- .type protobuf-int64
- .optional
- .code 1
- ]
- .field [
- .name opset-import
- .type OperatorSetIdProto
- .repeated
- .code 8
- ]
- .field [
- .name producer-name
- .type string
- .optional
- .code 2
- ]
- .field [
- .name producer-version
- .type string
- .optional
- .code 3
- ]
- .field [
- .name domain
- .type string
- .optional
- .code 4
- ]
- .field [
- .name model-version
- .type protobuf-int64
- .optional
- .code 5
- ]
- .field [
- .name doc-string
- .type string
- .optional
- .code 6
- ]
- .field [
- .name graph
- .type GraphProto
- .optional
- .code 7
- ]
- .field [
- .name metadata-props
- .type StringStringEntryProto
- .repeated
- .code 14
- ]
- .field [
- .name training-info
- .type TrainingInfoProto
- .repeated
- .code 20
- ]
- .field [
- .name functions
- .type FunctionProto
- .repeated
- .code 25
- ]
-]
-
-.record [
- .name StringStringEntryProto
- .field [
- .name key
- .type string
- .optional
- .code 1
- ]
- .field [
- .name value
- .type string
- .optional
- .code 2
- ]
-]
-
-.record [
- .name TensorAnnotation
- .field [
- .name tensor-name
- .type string
- .optional
- .code 1
- ]
- .field [
- .name quant-parameter-tensor-names
- .type StringStringEntryProto
- .repeated
- .code 2
- ]
-]
-
-.record [
- .name GraphProto
- .field [
- .name node
- .type NodeProto
- .repeated
- .code 1
- ]
- .field [
- .name name
- .type string
- .optional
- .code 2
- ]
- .field [
- .name initializer
- .type TensorProto
- .repeated
- .code 5
- ]
- .field [
- .name sparse-initializer
- .type SparseTensorProto
- .repeated
- .code 15
- ]
- .field [
- .name doc-string
- .type string
- .optional
- .code 10
- ]
- .field [
- .name input
- .type ValueInfoProto
- .repeated
- .code 11
- ]
- .field [
- .name output
- .type ValueInfoProto
- .repeated
- .code 12
- ]
- .field [
- .name value-info
- .type ValueInfoProto
- .repeated
- .code 13
- ]
- .field [
- .name quantization-annotation
- .type TensorAnnotation
- .repeated
- .code 14
- ]
-]
-
-.record [
- .name TensorProto
- .field [
- .name dims
- .type protobuf-int64
- .repeated
- .code 1
- ]
- .field [
- .name data-type
- .type protobuf-int32
- .optional
- .code 2
- ]
- .field [
- .name segment
- .type TensorProto-Segment
- .optional
- .code 3
- ]
- .field [
- .name float-data
- .type float32
- .repeated
- .code 4
- .protobuf-packed
- ]
- .field [
- .name int32-data
- .type protobuf-int32
- .repeated
- .code 5
- .protobuf-packed
- ]
- .field [
- .name string-data
- .type binary
- .repeated
- .code 6
- ]
- .field [
- .name int64-data
- .type protobuf-int64
- .repeated
- .code 7
- .protobuf-packed
- ]
- .field [
- .name name
- .type string
- .optional
- .code 8
- ]
- .field [
- .name doc-string
- .type string
- .optional
- .code 12
- ]
- .field [
- .name raw-data
- .type binary
- .optional
- .code 9
- ]
- .field [
- .name external-data
- .type StringStringEntryProto
- .repeated
- .code 13
- ]
- .field [
- .name data-location
- .type TensorProto-DataLocation
- .optional
- .code 14
- ]
- .field [
- .name double-data
- .type float64
- .repeated
- .code 10
- .protobuf-packed
- ]
- .field [
- .name uint64-data
- .type uint64
- .repeated
- .code 11
- .protobuf-packed
- ]
-]
-
-.record [
- .name TensorProto-Segment
- .field [
- .name begin
- .type protobuf-int64
- .optional
- .code 1
- ]
- .field [
- .name end
- .type protobuf-int64
- .optional
- .code 2
- ]
-]
-
-.enum [
- .name TensorProto-DataType
- .option [
- .name UNDEFINED
- .code 0
- .protobuf-name "TensorProto_UNDEFINED"
- ]
- .option [
- .name FLOAT
- .code 1
- .protobuf-name "TensorProto_FLOAT"
- ]
- .option [
- .name UINT8
- .code 2
- .protobuf-name "TensorProto_UINT8"
- ]
- .option [
- .name INT8
- .code 3
- .protobuf-name "TensorProto_INT8"
- ]
- .option [
- .name UINT16
- .code 4
- .protobuf-name "TensorProto_UINT16"
- ]
- .option [
- .name INT16
- .code 5
- .protobuf-name "TensorProto_INT16"
- ]
- .option [
- .name INT32
- .code 6
- .protobuf-name "TensorProto_INT32"
- ]
- .option [
- .name INT64
- .code 7
- .protobuf-name "TensorProto_INT64"
- ]
- .option [
- .name STRING
- .code 8
- .protobuf-name "TensorProto_STRING"
- ]
- .option [
- .name BOOL
- .code 9
- .protobuf-name "TensorProto_BOOL"
- ]
- .option [
- .name FLOAT16
- .code 10
- .protobuf-name "TensorProto_FLOAT16"
- ]
- .option [
- .name DOUBLE
- .code 11
- .protobuf-name "TensorProto_DOUBLE"
- ]
- .option [
- .name UINT32
- .code 12
- .protobuf-name "TensorProto_UINT32"
- ]
- .option [
- .name UINT64
- .code 13
- .protobuf-name "TensorProto_UINT64"
- ]
- .option [
- .name COMPLEX64
- .code 14
- .protobuf-name "TensorProto_COMPLEX64"
- ]
- .option [
- .name COMPLEX128
- .code 15
- .protobuf-name "TensorProto_COMPLEX128"
- ]
- .option [
- .name BFLOAT16
- .code 16
- .protobuf-name "TensorProto_BFLOAT16"
- ]
-]
-
-.enum [
- .name TensorProto-DataLocation
- .option [
- .name DEFAULT
- .code 0
- .protobuf-name "TensorProto_DEFAULT"
- ]
- .option [
- .name EXTERNAL
- .code 1
- .protobuf-name "TensorProto_EXTERNAL"
- ]
-]
-
-.record [
- .name SparseTensorProto
- .field [
- .name values
- .type TensorProto
- .optional
- .code 1
- ]
- .field [
- .name indices
- .type TensorProto
- .optional
- .code 2
- ]
- .field [
- .name dims
- .type protobuf-int64
- .repeated
- .code 3
- ]
-]
-
-.record [
- .name TensorShapeProto
- .field [
- .name dim
- .type TensorShapeProto-Dimension
- .repeated
- .code 1
- ]
-]
-
-.record [
- .name TensorShapeProto-Dimension
- .field [
- .name dim-value
- .type protobuf-int64
- .optional
- .code 1
- ]
- .field [
- .name dim-param
- .type string
- .optional
- .code 2
- ]
- .field [
- .name denotation
- .type string
- .optional
- .code 3
- ]
-]
-
-.record [
- .name TypeProto
- .field [
- .name tensor-type
- .type TypeProto-Tensor
- .optional
- .code 1
- ]
- .field [
- .name sequence-type
- .type TypeProto-Sequence
- .optional
- .code 4
- ]
- .field [
- .name map-type
- .type TypeProto-Map
- .optional
- .code 5
- ]
- .field [
- .name optional-type
- .type TypeProto-Optional
- .optional
- .code 9
- ]
- .field [
- .name sparse-tensor-type
- .type TypeProto-SparseTensor
- .optional
- .code 8
- ]
- .field [
- .name denotation
- .type string
- .optional
- .code 6
- ]
-]
-
-.record [
- .name TypeProto-Tensor
- .field [
- .name elem-type
- .type protobuf-int32
- .optional
- .code 1
- ]
- .field [
- .name shape
- .type TensorShapeProto
- .optional
- .code 2
- ]
-]
-
-.record [
- .name TypeProto-Sequence
- .field [
- .name elem-type
- .type TypeProto
- .optional
- .code 1
- ]
-]
-
-.record [
- .name TypeProto-Map
- .field [
- .name key-type
- .type protobuf-int32
- .optional
- .code 1
- ]
- .field [
- .name value-type
- .type TypeProto
- .optional
- .code 2
- ]
-]
-
-.record [
- .name TypeProto-Optional
- .field [
- .name elem-type
- .type TypeProto
- .optional
- .code 1
- ]
-]
-
-.record [
- .name TypeProto-SparseTensor
- .field [
- .name elem-type
- .type protobuf-int32
- .optional
- .code 1
- ]
- .field [
- .name shape
- .type TensorShapeProto
- .optional
- .code 2
- ]
-]
-
-.record [
- .name OperatorSetIdProto
- .field [
- .name domain
- .type string
- .optional
- .code 1
- ]
- .field [
- .name version
- .type protobuf-int64
- .optional
- .code 2
- ]
-]
-
-.record [
- .name FunctionProto
- .field [
- .name name
- .type string
- .optional
- .code 1
- ]
- .field [
- .name input
- .type string
- .repeated
- .code 4
- ]
- .field [
- .name output
- .type string
- .repeated
- .code 5
- ]
- .field [
- .name attribute
- .type string
- .repeated
- .code 6
- ]
- .field [
- .name node
- .type NodeProto
- .repeated
- .code 7
- ]
- .field [
- .name doc-string
- .type string
- .optional
- .code 8
- ]
- .field [
- .name opset-import
- .type OperatorSetIdProto
- .repeated
- .code 9
- ]
- .field [
- .name domain
- .type string
- .optional
- .code 10
- ]
-]
-
-.enum [
- .name Version
- .option [
- .name START-VERSION
- .code 0
- ]
- .option [
- .name IR-VERSION-2017-10-10
- .code 1
- ]
- .option [
- .name IR-VERSION-2017-10-30
- .code 2
- ]
- .option [
- .name IR-VERSION-2017-11-3
- .code 3
- ]
- .option [
- .name IR-VERSION-2019-1-22
- .code 4
- ]
- .option [
- .name IR-VERSION-2019-3-18
- .code 5
- ]
- .option [
- .name IR-VERSION-2019-9-19
- .code 6
- ]
- .option [
- .name IR-VERSION-2020-5-8
- .code 7
- ]
- .option [
- .name IR-VERSION
- .code 8
- ]
-]
-
-.enum [
- .name OperatorStatus
- .option [
- .name EXPERIMENTAL
- .code 0
- ]
- .option [
- .name STABLE
- .code 1
- ]
-]
diff --git a/lib/onnx/onnx_piqi.ml b/lib/onnx/onnx_piqi.ml
deleted file mode 100644
index 2a0636f17e3fc914af696767c6305dc8719689d6..0000000000000000000000000000000000000000
--- a/lib/onnx/onnx_piqi.ml
+++ /dev/null
@@ -1,1320 +0,0 @@
-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/lib/onnx/onnx_piqi.mli b/lib/onnx/onnx_piqi.mli
deleted file mode 100644
index 97ea99f42f0c288d2508620f5b35b109cd9a5637..0000000000000000000000000000000000000000
--- a/lib/onnx/onnx_piqi.mli
+++ /dev/null
@@ -1,718 +0,0 @@
-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/lib/onnx/onnx.proto b/lib/onnx/onnx_protoc.proto
similarity index 100%
rename from lib/onnx/onnx.proto
rename to lib/onnx/onnx_protoc.proto
diff --git a/onnx.opam b/onnx.opam
index dd5558135befdf02b5def8c6aae20d3f34470fb0..1440f206e7b77aecba8249d0490dd381388aa5ce 100644
--- a/onnx.opam
+++ b/onnx.opam
@@ -6,6 +6,7 @@ depends: [
"ocaml" {>= "4.10"}
"dune" {>= "2.9" & >= "2.9.1"}
"base" {>= "v0.14.0"}
+ "ocaml-protoc-plugin" {= "4.2.0"}
"odoc" {with-doc}
]
build: [
diff --git a/src/language.ml b/src/language.ml
index 1b8a2773f450bc0feba571807646e41e6d64dd40..a9e6d176a41aed8ac06baabac66b055e39769ab0 100644
--- a/src/language.ml
+++ b/src/language.ml
@@ -19,8 +19,15 @@ let loaded_nets = Why3.Term.Hls.create 10
let lookup_loaded_nets = Why3.Term.Hls.find_opt loaded_nets
-let register_ioshape nb_inputs nb_outputs ioshape_input_type filename th_uc =
+let register_astuple nb_inputs nb_outputs filename env =
let open Why3 in
+ let net = Pmodule.read_module env [ "caisar" ] "IOShape" in
+ let ioshape_input_type =
+ Ty.ty_app Theory.(ns_find_ts net.mod_theory.th_export [ "input_type" ]) []
+ in
+ let id_as_tuple = Ident.id_fresh "AsTuple" in
+ let th_uc = Pmodule.create_module env id_as_tuple in
+ let th_uc = Pmodule.use_export th_uc net in
let ls_net_apply =
let f _ = ioshape_input_type in
Term.create_fsymbol
@@ -41,18 +48,7 @@ let nnet_parser env _ filename _ =
let header = Nnet.parse filename in
match header with
| Error s -> Loc.errorm "%s" s
- | Ok header ->
- let nnet = Pmodule.read_module env [ "caisar" ] "IOShape" in
- let nnet_input_type =
- Ty.ty_app
- Theory.(ns_find_ts nnet.mod_theory.th_export [ "input_type" ])
- []
- in
- let id_as_tuple = Ident.id_fresh "AsTuple" in
- let th_uc = Pmodule.create_module env id_as_tuple in
- let th_uc = Pmodule.use_export th_uc nnet in
- register_ioshape header.n_inputs header.n_outputs nnet_input_type filename
- th_uc
+ | Ok header -> register_astuple header.n_inputs header.n_outputs filename env
let onnx_parser env _ filename _ =
let open Why3 in
@@ -60,44 +56,8 @@ let onnx_parser env _ filename _ =
match header with
| Error s -> Loc.errorm "%s" s
| Ok model ->
- let onnx = Pmodule.read_module env [ "caisar" ] "IOShape" in
- let onnx_input_type =
- Ty.ty_app
- Theory.(ns_find_ts onnx.mod_theory.th_export [ "input_type" ])
- []
- in
- let id_as_tuple = Ident.id_fresh "AsTuple" in
- let th_uc = Pmodule.create_module env id_as_tuple in
- let th_uc = Pmodule.use_export th_uc onnx in
- let ins, outs =
- match model.graph with
- | Some g -> (Some g.input, Some g.output)
- | None -> (None, None)
- in
- let get_nested_dims (s : Onnx.Opiqi.value_info_proto list) =
- match List.nth s 0 with
- | Some { type_ = Some { tensor_type = Some { shape = Some v; _ }; _ }; _ }
- ->
- v.dim
- | _ -> []
- in
- let input_shape, output_shape =
- match (ins, outs) with
- | Some i, Some o -> (get_nested_dims i, get_nested_dims o)
- | _ -> ([], [])
- in
- (* TODO: here we only get the flattened dimension of inputs and outputs, but
- more interesting parsing could be done later on. *)
- let flattened_dim (dim : Onnx.Opiqi.tensor_shape_proto_dimension list) =
- List.fold ~init:1 dim ~f:(fun acc x ->
- match x.dim_value with
- | Some v -> acc * Int64.to_int_exn v
- | None -> acc)
- in
- let input_flat_dim = flattened_dim input_shape in
- let output_flat_dim = flattened_dim output_shape in
- register_ioshape input_flat_dim output_flat_dim onnx_input_type filename
- th_uc
+ let input_flat_dim, output_flat_dim = Onnx.get_input_output_shape model in
+ register_astuple input_flat_dim output_flat_dim filename env
let register_nnet_support () =
Why3.(
diff --git a/src/transformations/native_nn_prover.ml b/src/transformations/native_nn_prover.ml
index 95b89b07e1cf969070cf545f9c6d7a26af0ac56b..3e84a490bb83c26db4d97f8c8c3574ce804b713f 100644
--- a/src/transformations/native_nn_prover.ml
+++ b/src/transformations/native_nn_prover.ml
@@ -1,4 +1,5 @@
(**************************************************************************)
+
(* *)
(* This file is part of CAISAR. *)
(* *)
@@ -30,7 +31,7 @@ let get_input_variables =
let rec aux acc (term : Term.term) =
match term.t_node with
| Term.Tapp (ls, args) -> (
- match Language.lookup_loaded_nnets ls with
+ match Language.lookup_loaded_nets ls with
| None -> acc
| Some _ ->
let add i acc = function
@@ -52,7 +53,7 @@ let simplify_goal env input_variables =
let rec aux meta hls (term : Term.term) =
match term.t_node with
| Term.Tapp (ls, _) -> (
- match Language.lookup_loaded_nnets ls with
+ match Language.lookup_loaded_nets ls with
| None -> Term.t_map (aux meta hls) term
| Some nnet ->
meta := nnet.filename :: !meta;
diff --git a/tests/dune b/tests/dune
index eac4b0ca195dff598a99f13008ccbf1bacf24e70..d62110c4fc1329e713e41fe0b50289b81bd914a8 100644
--- a/tests/dune
+++ b/tests/dune
@@ -2,6 +2,7 @@
(deps
(package caisar)
TestNetwork.nnet
+ TestNetworkONNX.onnx
bin/pyrat.py
bin/Marabou
))
diff --git a/tests/marabout.t b/tests/marabout.t
index db5c02bccd1ad88e859caaf31682762d3b9ae946..b1735d3404aaf80cc8fbc875de3e478b20e3fecb 100644
--- a/tests/marabout.t
+++ b/tests/marabout.t
@@ -21,11 +21,11 @@ Test verify
> theory T
> use TestNetwork.AsTuple
> use ieee_float.Float64
- > use caisar.NNet
+ > use caisar.IOShape
>
> goal G: forall x1 x2 x3 x4 x5.
> (0.0:t) .< x1 .< (0.5:t) ->
- > let (y1,_,_,_,_) = nnet_apply x1 x2 x3 x4 x5 in
+ > let (y1,_,_,_,_) = net_apply x1 x2 x3 x4 x5 in
> (0.0:t) .< y1 .< (0.5:t)
> end
> EOF
diff --git a/tests/simple.t b/tests/simple.t
index 7b41c27d4cfea8e64d257eb3418f3db299257035..8a517809b2b7a31f4e2f6e9f54f6ea28f82b93ad 100644
--- a/tests/simple.t
+++ b/tests/simple.t
@@ -21,16 +21,16 @@ Test verify
> theory T
> use TestNetwork.AsTuple
> use ieee_float.Float64
- > use caisar.NNet
+ > use caisar.IOShape
>
> goal G: forall x1 x2 x3 x4 x5.
> (0.0:t) .< x1 .< (0.5:t) ->
- > let (y1,_,_,_,_) = nnet_apply x1 x2 x3 x4 x5 in
+ > let (y1,_,_,_,_) = net_apply x1 x2 x3 x4 x5 in
> (0.0:t) .< y1 .< (0.5:t)
>
> goal H: forall x1 x2 x3 x4 x5.
> (0.0:t) .< x1 .< (0.5:t) /\ (0.5:t) .< x2 .< (1.0:t) ->
- > let (y1,y2,_,_,_) = nnet_apply x1 x2 x3 x4 x5 in
+ > let (y1,y2,_,_,_) = net_apply x1 x2 x3 x4 x5 in
> ((0.0:t) .< y1 \/ (0.5:t) .< y1) /\ (0.0:t) .< y2 .< (0.5:t)
> end
> EOF
diff --git a/tests/simple_onnx.t b/tests/simple_onnx.t
index fb59e4e358028bc322e652d9db7cd76a0c74afc9..6a35e67c6761bd7281ea2d0c7c635b47f6417fbd 100644
--- a/tests/simple_onnx.t
+++ b/tests/simple_onnx.t
@@ -1,30 +1,18 @@
Test verify
- $ mkdir bin
-
$ cat - > bin/alt-ergo << EOF
> #!/bin/sh
> echo "2.4.0"
> EOF
- $ cat - > bin/pyrat.py << EOF
- > #!/bin/sh
- > echo "PyRAT 1.0"
- > EOF
-
- $ cat - > bin/Marabou << EOF
- > #!/bin/sh
- > echo "1.0.+"
- > EOF
-
$ chmod u+x bin/alt-ergo bin/pyrat.py bin/Marabou
$ bin/alt-ergo
2.4.0
- $ bin/pyrat.py
- PyRAT 1.0
+ $ bin/pyrat.py --version
+ PyRAT 1.1
- $ bin/Marabou
+ $ bin/Marabou --version
1.0.+
$ PATH=$(pwd)/bin:$PATH
@@ -33,11 +21,11 @@ Test verify
> theory T
> use TestNetworkONNX.AsTuple
> use ieee_float.Float64
- > use caisar.NNet
+ > use caisar.IOShape
>
> goal G: forall x1 x2 x3.
> (0.0:t) .< x1 .< (0.5:t) ->
- > let (y1,_) = onnx_apply x1 x2 x3 in
+ > let (y1,_) = net_apply x1 x2 x3 in
> (0.0:t) .< y1 .< (0.5:t)
> end
> EOF
@@ -45,17 +33,19 @@ Test verify
<autodetect>Generating strategies:
<autodetect>Run: (Marabou --version) > $TMPFILE 2>&1
<autodetect>Run: (alt-ergo --version) > $TMPFILE 2>&1
- <autodetect>Run: (alt-ergo-2.4.0 --version) > $TMPFILE 2>&1
- <autodetect>command 'alt-ergo-2.4.0 --version' failed. Output:
- sh: 1: alt-ergo-2.4.0: not found
-
<autodetect>Run: (pyrat.py --version) > $TMPFILE 2>&1
<autodetect>Found prover Alt-Ergo version 2.4.0, OK.
<autodetect>Found prover Marabou version 1.0.+, OK.
- <autodetect>Found prover PyRAT version 1.0, OK.
+ <autodetect>Found prover PyRAT version 1.1, OK.
<autodetect>3 prover(s) added
- (* This is the prelude for PyRAT *)
+ Goal G: Unknown
+ ()
+ Output:
+ NN: ./TestNetworkONNX.onnx
+ Goal:
0.0 < x0
x0 < 0.5
0.0 < y0
y0 < 0.5
+ Result = Unknown
+