提交 c95da640 authored 作者: James Bergstra's avatar James Bergstra

Added specialization local_neg_neg

上级 6f34947f
...@@ -919,6 +919,16 @@ def local_mul_to_neg(node): ...@@ -919,6 +919,16 @@ def local_mul_to_neg(node):
return False return False
register_specialize(local_mul_to_neg) register_specialize(local_mul_to_neg)
@register_specialize
@gof.local_optimizer([T.neg])
def local_neg_neg(node):
# other specializations shouldn't put this in,
# but sometimes they do
if node.op == T.neg:
if node.inputs[0].owner and node.inputs[0].owner.op == T.neg:
return [node.inputs[0].owner.inputs[0]]
@gof.local_optimizer([T.mul]) @gof.local_optimizer([T.mul])
def local_mul_zero(node): def local_mul_zero(node):
"""As part of canonicalization, we replace multiplication by zero with zero. """As part of canonicalization, we replace multiplication by zero with zero.
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论