提交 a52b4e2e authored 作者: Frederic's avatar Frederic

fix scan merge and speed up merge optimizer

上级 91aa2db6
...@@ -609,7 +609,8 @@ class MergeOptimizer(Optimizer): ...@@ -609,7 +609,8 @@ class MergeOptimizer(Optimizer):
inputs_match = all(node_in is cand_in inputs_match = all(node_in is cand_in
for node_in, cand_in in zip( for node_in, cand_in in zip(
node.inputs, candidate.inputs)) node.inputs, candidate.inputs))
if not inputs_match or node.op != candidate.op: # No need to compare the op again, as it don't change.
if not inputs_match:
continue continue
try: try:
fgraph.replace_all_validate(pairs, 'MergeOptimizer') fgraph.replace_all_validate(pairs, 'MergeOptimizer')
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论