fix the bug

91d8c0d4 · SunAhong1993 · 8698ca2d · 91d8c0d4
隐藏空白更改
内联并排

Showing with 2 addition and 2 deletion

x2paddle/optimizer/elimination/dygraph/transpose_elimination.py ...le/optimizer/elimination/dygraph/transpose_elimination.py +2 -2

未找到文件。
--- a/x2paddle/optimizer/elimination/dygraph/transpose_elimination.py
+++ b/x2paddle/optimizer/elimination/dygraph/transpose_elimination.py
@@ -93,7 +93,7 @@ class Dygraph_TransposeElimination(FuseBase):
                        if _graph.layers[out].outputs[0] in _graph.outputs:
                            can_be_optimized = False
                            break
-                        if not _graph.layers[out].attrs.get('keepdim', False):
+                        if _graph.layers[out].attrs.get('keepdim', False):
                            can_be_optimized = False
                            break
                        propagate_layers.append(out)
@@ -206,7 +206,7 @@ class Dygraph_TransposeElimination(FuseBase):
                            if _graph.layers[ipt].outputs[0] in _graph.outputs:
                                can_be_optimized = False
                                break
-                            if not _graph.layers[ipt].attrs.get('keepdim',
+                            if _graph.layers[ipt].attrs.get('keepdim',
                                                                False):
                                can_be_optimized = False
                                break