提交 f176bf07 authored 作者: Ziye Fan's avatar Ziye Fan

when recursion, no need to substract the node from its client's input list

上级 faeef95f
...@@ -3600,9 +3600,9 @@ def local_fill_sink(node): ...@@ -3600,9 +3600,9 @@ def local_fill_sink(node):
if (hasattr(client, 'op') and if (hasattr(client, 'op') and
isinstance(client.op, T.Elemwise) and isinstance(client.op, T.Elemwise) and
not client.op == T.fill): not client.op == T.fill):
node_sub_c = client.op([v for v in client.inputs if v is not c]) # node_sub_c = client.op([v for v in client.inputs if v is not c])
# import ipdb; ipdb.set_trace() # import ipdb; ipdb.set_trace()
r = local_fill_sink.transform(node_sub_c) r = local_fill_sink.transform(client)
if r: if r:
return {client: r} return {client: r}
return [c] return [c]
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论