|
17 | 17 | make_opsetid, |
18 | 18 | make_tensor_value_info, |
19 | 19 | ) |
| 20 | +fromonnx.reference.op_runimportto_array_extended |
20 | 21 | fromonnx.numpy_helperimportfrom_array,to_array |
21 | 22 | fromonnx.backend.baseimportDevice,DeviceType |
22 | 23 | fromonnx_array_api.referenceimportExtendedReferenceEvaluator |
| 24 | +fromonnx_array_api.light_api.make_helperimportmake_node_extended |
23 | 25 | fromonnx_array_api.light_apiimporttranslate |
24 | 26 | fromonnx_array_api.plotting.text_plotimportonnx_simple_text_plot |
25 | 27 |
|
@@ -85,13 +87,15 @@ def run( |
85 | 87 | locs= { |
86 | 88 | "np":numpy, |
87 | 89 | "to_array":to_array, |
| 90 | +"to_array_extended":to_array_extended, |
88 | 91 | "from_array":from_array, |
89 | 92 | "TensorProto":TensorProto, |
90 | 93 | "make_function":make_function, |
91 | 94 | "make_opsetid":make_opsetid, |
92 | 95 | "make_model":make_model, |
93 | 96 | "make_graph":make_graph, |
94 | 97 | "make_node":make_node, |
| 98 | +"make_node_extended":make_node_extended, |
95 | 99 | "make_tensor_value_info":make_tensor_value_info, |
96 | 100 | } |
97 | 101 | globs=locs.copy() |
|