提交 91d8c0d4 编写于 作者: S SunAhong1993

fix the bug

上级 8698ca2d
......@@ -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
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册