Feb 19, 2025 · Feb 18, 2025 · Feb 19, 2025 · Feb 19, 2025 · Feb 19, 2025 · Feb 19, 2025
diff --git a/_unittests/ut_translate_api/test_translate_builder.py b/_unittests/ut_translate_api/test_translate_builder.py
            graph,
            opset_imports=[oh.make_opsetid("", 14), oh.make_opsetid(new_domain, 1)],
            functions=[linear_regression],
            ir_version=10,
        )
        tr = Translater(onnx_model, emitter=BuilderEmitter("mm"))
        code = tr.export(as_str=True)


            def mm() -> "ModelProto":
                g = GraphBuilder({'': 14, 'custom': 1}, ir_version=11)
                g = GraphBuilder({'': 14, 'custom': 1}, ir_version=10)
                g.make_tensor_input("X", TensorProto.FLOAT, ('', ''))
                g.make_tensor_input("A", TensorProto.FLOAT, ('', ''))
                g.make_tensor_input("B", TensorProto.FLOAT, ('', ''))
Original file line number	Diff line number	Diff line change
Expand Up		@@ -227,6 +227,7 @@ def test_local_function(self):
		graph,
		opset_imports=[oh.make_opsetid("", 14), oh.make_opsetid(new_domain, 1)],
		functions=[linear_regression],
		ir_version=10,
		)
		tr = Translater(onnx_model, emitter=BuilderEmitter("mm"))
		code = tr.export(as_str=True)
Expand DownExpand Up		@@ -260,7 +261,7 @@ def make_custom_LinearRegression(g: "GraphBuilder"):


		def mm() -> "ModelProto":
		g = GraphBuilder({'': 14, 'custom': 1}, ir_version=11)
		g = GraphBuilder({'': 14, 'custom': 1}, ir_version=10)
		g.make_tensor_input("X", TensorProto.FLOAT, ('', ''))
		g.make_tensor_input("A", TensorProto.FLOAT, ('', ''))
		g.make_tensor_input("B", TensorProto.FLOAT, ('', ''))
Expand Down