提交 b541e4a5 编写于 作者: S SunAhong1993

fix

上级 b7baa928
...@@ -32,7 +32,7 @@ class DygraphPReLUFuser(FuseBase): ...@@ -32,7 +32,7 @@ class DygraphPReLUFuser(FuseBase):
conv2_mul_1_y = paddle.full(dtype='float32', shape=[1], fill_value=0.5) conv2_mul_1_y = paddle.full(dtype='float32', shape=[1], fill_value=0.5)
conv2_Relu = self.relu1(conv2_Conv2D) conv2_Relu = self.relu1(conv2_Conv2D)
conv2_Abs = paddle.abs(x=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 = 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_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) conv2_add = paddle.add(x=conv2_Relu, y=conv2_mul_1)
...@@ -60,7 +60,7 @@ class DygraphPReLUFuser(FuseBase): ...@@ -60,7 +60,7 @@ class DygraphPReLUFuser(FuseBase):
inputs={"x": "prelu-input-0"}, inputs={"x": "prelu-input-0"},
outputs=[gen_name(3)]) outputs=[gen_name(3)])
self.pattern.add_layer( self.pattern.add_layer(
"fluid.layers.elementwise_sub", "paddle.subtract",
inputs={"x": "prelu-input-0", inputs={"x": "prelu-input-0",
"y": gen_name(3)}, "y": gen_name(3)},
outputs=[gen_name(4)]) outputs=[gen_name(4)])
......
...@@ -31,7 +31,7 @@ class StaticPReLUFuser(FuseBase): ...@@ -31,7 +31,7 @@ class StaticPReLUFuser(FuseBase):
conv4_mul_1_y = paddle.full(dtype='float32', shape=[1], fill_value=0.5) conv4_mul_1_y = paddle.full(dtype='float32', shape=[1], fill_value=0.5)
conv4_Relu = paddle.nn.functional.relu(x=conv4_BiasAdd) conv4_Relu = paddle.nn.functional.relu(x=conv4_BiasAdd)
conv4_Abs = paddle.abs(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 = paddle.multiply(x=conv4_alphas, y=conv4_sub)
conv4_mul_1 = paddle.multiply(x=conv4_mul, y=conv4_mul_1_y) conv4_mul_1 = paddle.multiply(x=conv4_mul, y=conv4_mul_1_y)
conv4_add = paddle.add(x=conv4_Relu, y=conv4_mul_1) conv4_add = paddle.add(x=conv4_Relu, y=conv4_mul_1)
...@@ -59,7 +59,7 @@ class StaticPReLUFuser(FuseBase): ...@@ -59,7 +59,7 @@ class StaticPReLUFuser(FuseBase):
inputs={"x": "prelu-input-0"}, inputs={"x": "prelu-input-0"},
outputs=[gen_name(3)]) outputs=[gen_name(3)])
self.pattern.add_layer( self.pattern.add_layer(
"fluid.layers.elementwise_sub", "paddle.subtract",
inputs={"x": "prelu-input-0", inputs={"x": "prelu-input-0",
"y": gen_name(3)}, "y": gen_name(3)},
outputs=[gen_name(4)]) outputs=[gen_name(4)])
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册