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

register opt dimshuffle_as_view at the inplace opt stage.

上级 bc365ed0
......@@ -426,8 +426,10 @@ def dimshuffle_as_view(node):
new_op = DimShuffle(op.input_broadcastable, op.new_order, inplace=True)
return [new_op(*node.inputs)]
register_specialize(dimshuffle_as_view, 'inplace')
compile.optdb.register('dimshuffle_as_view',
TopoOptimizer(dimshuffle_as_view,
failure_callback=TopoOptimizer.warn_inplace), 60,
'fast_run', 'inplace')
register_canonicalize(local_dimshuffle_lift)
register_specialize(local_dimshuffle_lift)
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论