提交 800b252d authored 作者: James Bergstra's avatar James Bergstra

added DimShuffle canonicalization to not be inplace

上级 0f685103
......@@ -194,6 +194,11 @@ def local_dimshuffle_lift(node):
register_canonicalize(local_dimshuffle_lift)
register_specialize(local_dimshuffle_lift)
@register_canonicalize
@local_optimizer([])
def local_dimshuffle_no_inplace_at_canonicalize(node):
if isinstance(node.op, T.DimShuffle) and node.op.inplace:
return [T.DimShuffle(node.op.input_broadcastable, node.op.new_order, inplace=False)(node.inputs[0])]
#####################################
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论