|
68 | 68 | " name='Add2',\n", |
69 | 69 | ")\n", |
70 | 70 | "\n", |
71 | | - "Add3_node = onnx.helper.make_node(\n", |
72 | | - " 'Add',\n", |
73 | | - " inputs=['abs1', 'abs1'],\n", |
74 | | - " outputs=['sum3'],\n", |
75 | | - " name='Add3',\n", |
76 | | - ")\n", |
77 | | - "\n", |
78 | 71 | "Abs_node = onnx.helper.make_node(\n", |
79 | 72 | " 'Abs',\n", |
80 | 73 | " inputs=['sum2'],\n", |
81 | 74 | " outputs=['abs1'],\n", |
82 | 75 | " name='Abs'\n", |
83 | 76 | ")\n", |
84 | 77 | "\n", |
| 78 | + "Add3_node = onnx.helper.make_node(\n", |
| 79 | + " 'Add',\n", |
| 80 | + " inputs=['abs1', 'abs1'],\n", |
| 81 | + " outputs=['sum3'],\n", |
| 82 | + " name='Add3',\n", |
| 83 | + ")\n", |
| 84 | + "\n", |
85 | 85 | "Round_node = onnx.helper.make_node(\n", |
86 | 86 | " 'Round',\n", |
87 | 87 | " inputs=['sum3'],\n", |
88 | 88 | " outputs=['out1'],\n", |
89 | 89 | " name='Round',\n", |
90 | | - ")\n" |
| 90 | + ")" |
91 | 91 | ] |
92 | 92 | }, |
93 | 93 | { |
|
253 | 253 | "metadata": {}, |
254 | 254 | "outputs": [], |
255 | 255 | "source": [ |
256 | | - "in1_values =np.asarray(np.random.uniform(low=-5, high=5, size=(4,4)), dtype=np.float32)\n", |
| 256 | + "in1_values = np.asarray(np.random.uniform(low=-5, high=5, size=(4,4)), dtype=np.float32)\n", |
257 | 257 | "in2_values = np.asarray(np.random.uniform(low=-5, high=5, size=(4,4)), dtype=np.float32)\n", |
258 | 258 | "in3_values = np.asarray(np.random.uniform(low=-5, high=5, size=(4,4)), dtype=np.float32)" |
259 | 259 | ] |
|
350 | 350 | "metadata": {}, |
351 | 351 | "outputs": [], |
352 | 352 | "source": [ |
| 353 | + "import qonnx\n", |
353 | 354 | "from qonnx.core.modelwrapper import ModelWrapper\n", |
354 | 355 | "finn_model = ModelWrapper(onnx_model)" |
355 | 356 | ] |
|
0 commit comments