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

fix the bug

上级 8698ca2d
...@@ -93,7 +93,7 @@ class Dygraph_TransposeElimination(FuseBase): ...@@ -93,7 +93,7 @@ class Dygraph_TransposeElimination(FuseBase):
if _graph.layers[out].outputs[0] in _graph.outputs: if _graph.layers[out].outputs[0] in _graph.outputs:
can_be_optimized = False can_be_optimized = False
break break
if not _graph.layers[out].attrs.get('keepdim', False): if _graph.layers[out].attrs.get('keepdim', False):
can_be_optimized = False can_be_optimized = False
break break
propagate_layers.append(out) propagate_layers.append(out)
...@@ -206,7 +206,7 @@ class Dygraph_TransposeElimination(FuseBase): ...@@ -206,7 +206,7 @@ class Dygraph_TransposeElimination(FuseBase):
if _graph.layers[ipt].outputs[0] in _graph.outputs: if _graph.layers[ipt].outputs[0] in _graph.outputs:
can_be_optimized = False can_be_optimized = False
break break
if not _graph.layers[ipt].attrs.get('keepdim', if _graph.layers[ipt].attrs.get('keepdim',
False): False):
can_be_optimized = False can_be_optimized = False
break break
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册