From 5826a9e2ca3df3ec4b4ba6aca73241fac1e0438a Mon Sep 17 00:00:00 2001
From: mmrahorovic <mmrahorovic@hotmail.com>
Date: Tue, 31 Jan 2023 18:13:18 +0000
Subject: [PATCH] [notebooks]: use new utility function to create onnx model
---
notebooks/advanced/2_custom_op.ipynb | 3 ++-
notebooks/basics/0_how_to_work_with_onnx.ipynb | 5 +++--
2 files changed, 5 insertions(+), 3 deletions(-)
diff --git a/notebooks/advanced/2_custom_op.ipynb b/notebooks/advanced/2_custom_op.ipynb
index c27f8bdca..051a40670 100644
--- a/notebooks/advanced/2_custom_op.ipynb
+++ b/notebooks/advanced/2_custom_op.ipynb
@@ -178,6 +178,7 @@
"source": [
"from qonnx.core.modelwrapper import ModelWrapper\n",
"from onnx import TensorProto\n",
+ "from qonnx.util.basic import qonnx_make_model\n",
"\n",
"def make_graph(ishape, exp, op_type = \"MyPythonPowerOp\"):\n",
" inp = helper.make_tensor_value_info(\n",
@@ -204,7 +205,7 @@
" graph = helper.make_graph(\n",
" nodes=[custom_node], name=\"custom_graph\", inputs=[inp], outputs=[outp]\n",
" )\n",
- " model = helper.make_model(graph, producer_name=\"custom-model\")\n",
+ " model = qonnx_make_model(graph, producer_name=\"custom-model\")\n",
" return ModelWrapper(model)"
]
},
diff --git a/notebooks/basics/0_how_to_work_with_onnx.ipynb b/notebooks/basics/0_how_to_work_with_onnx.ipynb
index 514efd169..b6a5a0481 100644
--- a/notebooks/basics/0_how_to_work_with_onnx.ipynb
+++ b/notebooks/basics/0_how_to_work_with_onnx.ipynb
@@ -36,6 +36,7 @@
"outputs": [],
"source": [
"import onnx\n",
+ "from qonnx.util.basic import qonnx_make_model\n",
"\n",
"Add1_node = onnx.helper.make_node(\n",
" 'Add',\n",
@@ -158,7 +159,7 @@
"metadata": {},
"outputs": [],
"source": [
- "onnx_model = onnx.helper.make_model(graph, producer_name=\"simple-model\")\n",
+ "onnx_model = qonnx_make_model(graph, producer_name=\"simple-model\")\n",
"onnx.save(onnx_model, '/tmp/simple_model.onnx')"
]
},
@@ -550,7 +551,7 @@
"metadata": {},
"outputs": [],
"source": [
- "onnx_model1 = onnx.helper.make_model(graph, producer_name=\"simple-model1\")\n",
+ "onnx_model1 = qonnx_make_model(graph, producer_name=\"simple-model1\")\n",
"onnx.save(onnx_model1, '/tmp/simple_model1.onnx')"
]
},
--
GitLab