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