提交 0ecaf3d6 authored 作者: Arnaud Bergeron's avatar Arnaud Bergeron

Add comment as per review.

上级 8268d2b6
...@@ -1878,6 +1878,10 @@ def local_useless_inc_subtensor(node): ...@@ -1878,6 +1878,10 @@ def local_useless_inc_subtensor(node):
@register_canonicalize @register_canonicalize
@gof.local_optimizer([AdvancedIncSubtensor1]) @gof.local_optimizer([AdvancedIncSubtensor1])
def local_set_to_inc_subtensor(node): def local_set_to_inc_subtensor(node):
"""
AdvancedIncSubtensor1(x, x[ilist]+other, ilist, set_instead_of_inc=True) ->
AdvancedIncSubtensor1(x, other, ilist, set_instead_of_inc=False)
"""
if (isinstance(node.op, AdvancedIncSubtensor1) and if (isinstance(node.op, AdvancedIncSubtensor1) and
node.op.set_instead_of_inc == True and node.op.set_instead_of_inc == True and
node.inputs[1].owner and node.inputs[1].owner and
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论