Add gelu pass

d8b7d67c · wjj19950828 · 2c1d8e82 · d8b7d67c
显示空白变更内容
内联并排

Showing with 2 addition and 4 deletion

x2paddle/optimizer/fusion/onnx_gelu_fuser.py x2paddle/optimizer/fusion/onnx_gelu_fuser.py +2 -4

未找到文件。
--- a/x2paddle/optimizer/fusion/onnx_gelu_fuser.py
+++ b/x2paddle/optimizer/fusion/onnx_gelu_fuser.py
@@ -64,9 +64,7 @@ class GeluFuser(FuseBase):
                    "y": gen_name(0)},
            outputs=[gen_name(3)])
        self.pattern.add_layer(
-            "paddle.erf",
+            "paddle.erf", inputs={"x": gen_name(3)}, outputs=[gen_name(4)])
-            inputs={"x": gen_name(3)},
-            outputs=[gen_name(4)])
        self.pattern.add_layer(
            "paddle.add",
            inputs={"x": gen_name(4),