From b8f898a4cca5c2d1464a0c349e02b1023a12d946 Mon Sep 17 00:00:00 2001
From: Julien Girard <julien.girard2@cea.fr>
Date: Fri, 24 May 2024 11:08:52 +0200
Subject: [PATCH] Clarified opset import and supported onnx versions
---
lib/nir/node.mli | 2 +-
lib/onnx/writer.ml | 2 +-
2 files changed, 2 insertions(+), 2 deletions(-)
diff --git a/lib/nir/node.mli b/lib/nir/node.mli
index d47a8d1..0eb1df4 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 bbc5f2b..306009c 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
--
GitLab