提交 03aa83de authored 作者: Frederic Bastien's avatar Frederic Bastien

tag an optimizer as being an inplace optimizer.

上级 d382ccc7
...@@ -283,7 +283,6 @@ def local_dimshuffle_lift(node): ...@@ -283,7 +283,6 @@ def local_dimshuffle_lift(node):
else: else:
return DimShuffle(iinput.type.broadcastable, new_order, inplace).make_node(iinput).outputs return DimShuffle(iinput.type.broadcastable, new_order, inplace).make_node(iinput).outputs
@register_specialize
@gof.local_optimizer([]) @gof.local_optimizer([])
def dimshuffle_as_view(node): def dimshuffle_as_view(node):
op = node.op op = node.op
...@@ -293,6 +292,7 @@ def dimshuffle_as_view(node): ...@@ -293,6 +292,7 @@ def dimshuffle_as_view(node):
return [new_op(*node.inputs)] return [new_op(*node.inputs)]
register_specialize(dimshuffle_as_view, 'inplace')
register_canonicalize(local_dimshuffle_lift) register_canonicalize(local_dimshuffle_lift)
register_specialize(local_dimshuffle_lift) register_specialize(local_dimshuffle_lift)
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论