提交 29f197bc authored 作者: James Bergstra's avatar James Bergstra

corrected typo in setsubtensor inplace optimizer

上级 552b1caa
......@@ -281,7 +281,7 @@ def local_inplace_setsubtensor(node):
if isinstance(node.op, T.SetSubtensor) and not node.op.inplace:
new_op = T.SetSubtensor(node.op.idx_list, inplace=True)
new_node = new_op(*node.inputs)
return new_node.outputs
return [new_node]
return False
compile.optdb.register('inplace_setsubtensor', TopoOptimizer(local_inplace_setsubtensor), 60, 'fast_run', 'inplace') #DEBUG
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论