We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see ourdocumentation.
There was an error while loading.Please reload this page.
1 parent67eb1c2 commit00d0e7bCopy full SHA for 00d0e7b
_unittests/ut_translate_api/test_translate_builder.py
@@ -227,6 +227,7 @@ def test_local_function(self):
227
graph,
228
opset_imports=[oh.make_opsetid("",14),oh.make_opsetid(new_domain,1)],
229
functions=[linear_regression],
230
+ir_version=10,
231
)
232
tr=Translater(onnx_model,emitter=BuilderEmitter("mm"))
233
code=tr.export(as_str=True)
@@ -260,7 +261,7 @@ def make_custom_LinearRegression(g: "GraphBuilder"):
260
261
262
263
def mm() -> "ModelProto":
- g = GraphBuilder({'': 14, 'custom': 1}, ir_version=11)
264
+ g = GraphBuilder({'': 14, 'custom': 1}, ir_version=10)
265
g.make_tensor_input("X", TensorProto.FLOAT, ('', ''))
266
g.make_tensor_input("A", TensorProto.FLOAT, ('', ''))
267
g.make_tensor_input("B", TensorProto.FLOAT, ('', ''))