diff --git a/notebooks/0_how_to_work_with_onnx.ipynb b/notebooks/0_how_to_work_with_onnx.ipynb index 60340f14..a1c7a5ce 100644 --- a/notebooks/0_how_to_work_with_onnx.ipynb +++ b/notebooks/0_how_to_work_with_onnx.ipynb @@ -124,24 +124,24 @@ "metadata": {}, "outputs": [], "source": [ - " graph = onnx.helper.make_graph(\n", - " nodes=[\n", - " Add1_node,\n", - " Add2_node,\n", - " Abs_node,\n", - " Add3_node,\n", - " Round_node,\n", - " ],\n", - " name=\"simple_graph\",\n", - " inputs=[in1, in2, in3],\n", - " outputs=[out1],\n", - " value_info=[\n", - " onnx.helper.make_tensor_value_info(\"sum1\", onnx.TensorProto.FLOAT, [4, 4]),\n", - " onnx.helper.make_tensor_value_info(\"sum2\", onnx.TensorProto.FLOAT, [4, 4]),\n", - " onnx.helper.make_tensor_value_info(\"abs1\", onnx.TensorProto.FLOAT, [4, 4]),\n", - " onnx.helper.make_tensor_value_info(\"sum3\", onnx.TensorProto.FLOAT, [4, 4]),\n", - " ],\n", - " )\n" + "graph = onnx.helper.make_graph(\n", + " nodes=[\n", + " Add1_node,\n", + " Add2_node,\n", + " Abs_node,\n", + " Add3_node,\n", + " Round_node,\n", + " ],\n", + " name=\"simple_graph\",\n", + " inputs=[in1, in2, in3],\n", + " outputs=[out1],\n", + " value_info=[\n", + " onnx.helper.make_tensor_value_info(\"sum1\", onnx.TensorProto.FLOAT, [4, 4]),\n", + " onnx.helper.make_tensor_value_info(\"sum2\", onnx.TensorProto.FLOAT, [4, 4]),\n", + " onnx.helper.make_tensor_value_info(\"abs1\", onnx.TensorProto.FLOAT, [4, 4]),\n", + " onnx.helper.make_tensor_value_info(\"sum3\", onnx.TensorProto.FLOAT, [4, 4]),\n", + " ],\n", + ")\n" ] }, { @@ -471,6 +471,7 @@ " node_pair.append(node)\n", " node_pair.append(successor)\n", " adder_pairs.append((node_pair))\n", + " print(adder_pairs)\n", " node_pair = []\n", " return adder_pairs\n", " " @@ -608,7 +609,7 @@ "metadata": {}, "outputs": [], "source": [ - "onnx_model1 = qonnx.util.basic.qonnx_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')" ] }, @@ -670,7 +671,7 @@ "name": "python", "nbconvert_exporter": "python", "pygments_lexer": "ipython3", - "version": "3.8.18" + "version": "3.8.19" }, "vscode": { "interpreter": {