提交 a5641ca4 编写于 作者: S SunAhong1993

fix

上级 5282bad4
...@@ -60,7 +60,7 @@ class GraphOptimizer(object): ...@@ -60,7 +60,7 @@ class GraphOptimizer(object):
def optimize(self, graph): def optimize(self, graph):
for pass_name in self.passes: for pass_name in self.passes:
pass_ = PassManager.lookup(pass_name)() pass_ = PassManager.lookup(pass_name)()
if pass_name.endswith("_eliminate_pass"): if pass_name.endswith("_eliminate_pass") or pass_name.endswith("_conv2d_add_fuse_pass"):
pass_.apply(graph) pass_.apply(graph)
else: else:
while True: while True:
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册