提交 785518b0 authored 作者: Frederic Bastien's avatar Frederic Bastien

make the optimization happen in more case.

上级 4bbddc1c
......@@ -575,7 +575,7 @@ local_mul_canonizer = Canonizer(T.mul, T.div, T.inv, mul_calculate, False)
@gof.local_optimizer([T.neg])
def local_neg_to_mul(node):
if node.op == T.neg:
return [-1 * node.inputs[0]]
return [T.mul(-1, node.inputs[0])]
else:
return False
register_canonicalize(local_neg_to_mul)
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论