From 59c93909950bb013218e86b1277e7c21d95af714 Mon Sep 17 00:00:00 2001
From: markuswt <53354649+markuswt@users.noreply.github.com>
Date: Thu, 26 Jan 2023 05:01:40 +0000
Subject: [PATCH] Set ONNX opset version to 10

---
 python/taso/__init__.py | 3 ++-
 1 file changed, 2 insertions(+), 1 deletion(-)

diff --git a/python/taso/__init__.py b/python/taso/__init__.py
index 5c451cec..6bf68c4e 100644
--- a/python/taso/__init__.py
+++ b/python/taso/__init__.py
@@ -933,7 +933,8 @@ def export_onnx(graph):
         graph_outputs.append(helper.make_tensor_value_info(_output_tensor_name(graph, op, idx),
                              TensorProto.FLOAT, graph.get_output_dims(op, idx)))
     onnx_graph = helper.make_graph(graph_nodes, 'main', graph_inputs, graph_outputs, graph_initializers)
-    onnx_model = helper.make_model(onnx_graph, producer_name='TASO Optimized Model')
+    onnx_opset_id = helper.make_opsetid('', 10)
+    onnx_model = helper.make_model(onnx_graph, opset_imports=[onnx_opset_id], producer_name='TASO Optimized Model')
     return onnx_model
 
 def optimize(graph, alpha = 1.0, budget = 1000, print_subst = False):