From b541e4a548b1e8012cc572b087cd82dc8d5d04da Mon Sep 17 00:00:00 2001 From: SunAhong1993 Date: Wed, 20 Jan 2021 16:11:56 +0800 Subject: [PATCH] fix --- x2paddle/optimizer/fusion/dygraph/prelu_fuser.py | 4 ++-- x2paddle/optimizer/fusion/static/prelu_fuser.py | 4 ++-- 2 files changed, 4 insertions(+), 4 deletions(-) diff --git a/x2paddle/optimizer/fusion/dygraph/prelu_fuser.py b/x2paddle/optimizer/fusion/dygraph/prelu_fuser.py index ddab560..f8a45bd 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 c87abdf..758d933 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)]) -- GitLab