diff --git a/x2paddle/optimizer/fusion/dygraph/prelu_fuser.py b/x2paddle/optimizer/fusion/dygraph/prelu_fuser.py index ddab5605b9cf6deffa3e247df5430a95fc0474bc..f8a45bd73ed10e8af2d31f944d03548c9f95afc7 100644 --- a/x2paddle/optimizer/fusion/dygraph/prelu_fuser.py +++ b/x2paddle/optimizer/fusion/dygraph/prelu_fuser.py @@ -32,7 +32,7 @@ class DygraphPReLUFuser(FuseBase): conv2_mul_1_y = paddle.full(dtype='float32', shape=[1], fill_value=0.5) conv2_Relu = self.relu1(conv2_Conv2D) conv2_Abs = paddle.abs(x=conv2_Conv2D) - conv2_sub = fluid.layers.elementwise_sub(x=conv2_Conv2D, y=conv2_Abs) + conv2_sub = paddle.subtract(x=conv2_Conv2D, y=conv2_Abs) conv2_mul = paddle.multiply(x=conv2_alphas, y=conv2_sub, axis=1) conv2_mul_1 = paddle.multiply(x=conv2_mul, y=conv2_mul_1_y, axis=1) conv2_add = paddle.add(x=conv2_Relu, y=conv2_mul_1) @@ -60,7 +60,7 @@ class DygraphPReLUFuser(FuseBase): inputs={"x": "prelu-input-0"}, outputs=[gen_name(3)]) self.pattern.add_layer( - "fluid.layers.elementwise_sub", + "paddle.subtract", inputs={"x": "prelu-input-0", "y": gen_name(3)}, outputs=[gen_name(4)]) diff --git a/x2paddle/optimizer/fusion/static/prelu_fuser.py b/x2paddle/optimizer/fusion/static/prelu_fuser.py index c87abdf9a90a5d798d93b78eee896c96ddf3ab70..758d9335f2afb913e4143ba3b0a57989b6d79247 100644 --- a/x2paddle/optimizer/fusion/static/prelu_fuser.py +++ b/x2paddle/optimizer/fusion/static/prelu_fuser.py @@ -31,7 +31,7 @@ class StaticPReLUFuser(FuseBase): conv4_mul_1_y = paddle.full(dtype='float32', shape=[1], fill_value=0.5) conv4_Relu = paddle.nn.functional.relu(x=conv4_BiasAdd) conv4_Abs = paddle.abs(x=conv4_BiasAdd) - conv4_sub = fluid.layers.elementwise_sub(x=conv4_BiasAdd, y=conv4_Abs) + conv4_sub = paddle.subtract(x=conv4_BiasAdd, y=conv4_Abs) conv4_mul = paddle.multiply(x=conv4_alphas, y=conv4_sub) conv4_mul_1 = paddle.multiply(x=conv4_mul, y=conv4_mul_1_y) conv4_add = paddle.add(x=conv4_Relu, y=conv4_mul_1) @@ -59,7 +59,7 @@ class StaticPReLUFuser(FuseBase): inputs={"x": "prelu-input-0"}, outputs=[gen_name(3)]) self.pattern.add_layer( - "fluid.layers.elementwise_sub", + "paddle.subtract", inputs={"x": "prelu-input-0", "y": gen_name(3)}, outputs=[gen_name(4)])