refactor(python_ffi): 前端控制算子类型是否添加 onnx:: 前缀以便自定义算子

YdrMaster · YdrMaster · commit 1375be35a2ec · 2024-01-23T16:52:54.000+08:00
Signed-off-by: YdrMaster &lt;ydrml@hotmail.com&gt;
diff --git a/src/09python_ffi/src/import.cpp b/src/09python_ffi/src/import.cpp
@@ -65,7 +65,7 @@ namespace refactor::python_ffi {
             ctx_.insert({std::move(name), {std::move(value)}});
         }
         return std::make_shared<OpBox>(Operator::build(
-            ctx_, fmt::format("onnx::{}", opType), std::move(attrs_)));
+            ctx_, std::move(opType), std::move(attrs_)));
     }
 
     Arc<Compiler>
diff --git a/src/09python_ffi/src/refactor_graph/onnx.py b/src/09python_ffi/src/refactor_graph/onnx.py
@@ -63,7 +63,7 @@ def make_compiler(model: ModelProto, external_data_path: str = "") -> Compiler:
         {
             node.name: _make_operator(
                 context,
-                node.op_type,
+                "onnx::" + node.op_type,
                 _parse_attribute(node, external_data_path),
             )
             for node in model.graph.node

Original file line number	Diff line number	Diff line change
`@@ -65,7 +65,7 @@ namespace refactor::python_ffi {`
`65`	`65`	`ctx_.insert({std::move(name), {std::move(value)}});`
`66`	`66`	`}`
`67`	`67`	`return std::make_shared<OpBox>(Operator::build(`
`68`		`- ctx_, fmt::format("onnx::{}", opType), std::move(attrs_)));`
	`68`	`+ ctx_, std::move(opType), std::move(attrs_)));`
`69`	`69`	`}`
`70`	`70`
`71`	`71`	`Arc<Compiler>`
Original file line number	Diff line number	Diff line change
`@@ -63,7 +63,7 @@ def make_compiler(model: ModelProto, external_data_path: str = "") -> Compiler:`
`63`	`63`	`{`
`64`	`64`	`node.name: _make_operator(`
`65`	`65`	`context,`
`66`		`- node.op_type,`
	`66`	`+ "onnx::" + node.op_type,`
`67`	`67`	`_parse_attribute(node, external_data_path),`
`68`	`68`	`)`
`69`	`69`	`for node in model.graph.node`