提交 f78fc162 authored 作者: Frederic Bastien's avatar Frederic Bastien

update comment/docstring

上级 7af94c4f
...@@ -2665,9 +2665,9 @@ def local_join_1(node): ...@@ -2665,9 +2665,9 @@ def local_join_1(node):
@register_canonicalize @register_canonicalize
@gof.local_optimizer([T.Join]) @gof.local_optimizer([T.Join])
def local_join_empty(node): def local_join_empty(node):
"""Join(i, x, y, empty) => Join(i, x, y, empty) """Join(i, x, y, empty) => Join(i, x, y)
remove empty inputs to joins. remove empty inputs to joins. The empty inputs can be anywhere.
""" """
if not isinstance(node.op, T.Join): if not isinstance(node.op, T.Join):
return return
...@@ -2688,6 +2688,7 @@ def local_join_empty(node): ...@@ -2688,6 +2688,7 @@ def local_join_empty(node):
if len(new_inputs) < len(node.inputs) - 1: if len(new_inputs) < len(node.inputs) - 1:
if len(new_inputs) == 0: if len(new_inputs) == 0:
# T.join do not work in that case. # T.join do not work in that case.
# constant folding will take care of this case.
return return
ret = T.join(node.inputs[0], *new_inputs) ret = T.join(node.inputs[0], *new_inputs)
o = node.outputs[0] o = node.outputs[0]
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论