diff --git a/lib/nir/node.mli b/lib/nir/node.mli index d47a8d124cb6de93120269fb0b36d4dfea7c492b..0eb1df44d5d9ff199d3091983fc0800caa91e28f 100644 --- a/lib/nir/node.mli +++ b/lib/nir/node.mli @@ -28,7 +28,7 @@ - a node description of type [descr] [descr] describes several operations. Most of them are direct transcription - of the ONNX v13 standard, which a description is available here: + of the ONNX IR v8 standard, which a description is available here: https://onnx.ai/onnx/operators/index.html. Nodes only require their inputs: it is assumed that a node only returns one diff --git a/lib/onnx/writer.ml b/lib/onnx/writer.ml index bbc5f2b359df47da10f87aeaee6944d1ff4ab9f6..306009c4a13d889a938155554d0dfd8b07294300 100644 --- a/lib/onnx/writer.ml +++ b/lib/onnx/writer.ml @@ -58,7 +58,7 @@ let convert_into_tensor ?name (t : Nir.Gentensor.t) = let default_opset_import = let open Oproto.Onnx in let onnx_domain = "" in - OperatorSetIdProto.make ~domain:onnx_domain ~version:8L () + OperatorSetIdProto.make ~domain:onnx_domain ~version:22L () let nier_simple_to_onnx_protoc (nier_simple : Nir.Ngraph.t) = let open Oproto.Onnx in