File tree Expand file tree Collapse file tree 2 files changed +9
-8
lines changed Expand file tree Collapse file tree 2 files changed +9
-8
lines changed Original file line number Diff line number Diff line change @@ -206,13 +206,14 @@ def plot_dot(
206
206
from onnx_array_api.plotting.graphviz_helper import plot_dot
207
207
208
208
model = onnx.parser.parse_model(
209
- '''
210
- <ir_version: 8, opset_import: [ "": 18]>
211
- agraph (float[N] x) => (float[N] z) {
212
- two = Constant <value_float=2.0> ()
213
- four = Add(two, two)
214
- z = Mul(four, four)
215
- }''')
209
+ '''
210
+ <ir_version: 8, opset_import: [ "": 18]>
211
+ agraph (float[N] x) => (float[N] z) {
212
+ two = Constant <value_float=2.0> ()
213
+ four = Add(two, two)
214
+ z = Mul(four, four)
215
+ }
216
+ ''')
216
217
217
218
ax = plot_dot(model)
218
219
ax.set_title("Dummy graph")
Original file line number Diff line number Diff line change @@ -237,7 +237,7 @@ def enumerate_summarized(
237
237
Executes the onnx model and enumerate intermediate results without their names.
238
238
239
239
:param output_names: requested outputs by names, None for all
240
- :param feed_inputs: dictionary `{ input name: input value }`
240
+ :param feed_inputs: dictionary `` { input name: input value }` `
241
241
:param raise_exc: raises an exception if the execution fails or stop where it is
242
242
:param keep_tensor: keep the tensor in order to compute precise distances
243
243
:return: iterator on ResultExecution
You can’t perform that action at this time.
0 commit comments