From a749631731419d10f5ea6aabc490c42544bb8ff2 Mon Sep 17 00:00:00 2001 From: SunAhong1993 Date: Wed, 13 Jan 2021 12:48:57 +0800 Subject: [PATCH] fix the tf --- x2paddle/optimizer/fusion/dygraph/tf_batchnorm_fuser.py | 6 +++--- x2paddle/optimizer/fusion/static/tf_batchnorm_fuser.py | 6 +++--- 2 files changed, 6 insertions(+), 6 deletions(-) diff --git a/x2paddle/optimizer/fusion/dygraph/tf_batchnorm_fuser.py b/x2paddle/optimizer/fusion/dygraph/tf_batchnorm_fuser.py index f3af6b0..6a53b1d 100644 --- a/x2paddle/optimizer/fusion/dygraph/tf_batchnorm_fuser.py +++ b/x2paddle/optimizer/fusion/dygraph/tf_batchnorm_fuser.py @@ -74,7 +74,7 @@ class DygraphTFBatchNormFuser(FuseBase): inputs={}, outputs=[gen_name(8)]) pattern.add_layer( - "fluid.layers.elementwise_sub", + "paddle.subtract", inputs={"x": gen_name(8), "y": gen_name(7)}, outputs=[gen_name(9)]) pattern.add_layer( @@ -131,7 +131,7 @@ class DygraphTFBatchNormFuser(FuseBase): inputs={}, outputs=[gen_name(8)]) pattern.add_layer( - "fluid.layers.elementwise_sub", + "paddle.subtract", inputs={"x": gen_name(8), "y": gen_name(7)}, outputs=[gen_name(9)]) pattern.add_layer( @@ -180,7 +180,7 @@ class DygraphTFBatchNormFuser(FuseBase): if matches[out_layer_id].kernel == "paddle.multiply": gamma_layer_id = graph.edges_in[out_layer_id][1] gamma_layer = matches[gamma_layer_id] - if layer.kernel == "fluid.layers.elementwise_sub": + if layer.kernel == "paddle.subtract": in_layer_id = graph.edges_in[layer_id][0] beta_layer = matches[in_layer_id] in_layer_id = graph.edges_in[layer_id][1] diff --git a/x2paddle/optimizer/fusion/static/tf_batchnorm_fuser.py b/x2paddle/optimizer/fusion/static/tf_batchnorm_fuser.py index 796556b..1299b34 100644 --- a/x2paddle/optimizer/fusion/static/tf_batchnorm_fuser.py +++ b/x2paddle/optimizer/fusion/static/tf_batchnorm_fuser.py @@ -73,7 +73,7 @@ class StaticTFBatchNormFuser(FuseBase): inputs={}, outputs=[gen_name(8)]) pattern.add_layer( - "fluid.layers.elementwise_sub", + "paddle.subtract", inputs={"x": gen_name(8), "y": gen_name(7)}, outputs=[gen_name(9)]) pattern.add_layer( @@ -130,7 +130,7 @@ class StaticTFBatchNormFuser(FuseBase): inputs={}, outputs=[gen_name(8)]) pattern.add_layer( - "fluid.layers.elementwise_sub", + "paddle.subtract", inputs={"x": gen_name(8), "y": gen_name(7)}, outputs=[gen_name(9)]) pattern.add_layer( @@ -179,7 +179,7 @@ class StaticTFBatchNormFuser(FuseBase): if matches[out_layer_id].kernel == "paddle.multiply": gamma_layer_id = graph.edges_in[out_layer_id][1] gamma_layer = matches[gamma_layer_id] - if layer.kernel == "fluid.layers.elementwise_sub": + if layer.kernel == "paddle.subtract": in_layer_id = graph.edges_in[layer_id][0] beta_layer = matches[in_layer_id] in_layer_id = graph.edges_in[layer_id][1] -- GitLab