提交 85c369e1 authored 作者: Kelvin Xu's avatar Kelvin Xu 提交者: Kelvin Xu

perform negative replacement only for sum op

上级 758be180
......@@ -3890,7 +3890,7 @@ def local_sum_prod_mul_by_scalar(node):
return [T.mul(scalars[0], node.op(non_scalars[0]))]
else:
return [scalars[0]]
if node_inps.owner and node_inps.owner.op == T.neg:
if isinstance(node.op, T.Sum) and node_inps.owner and node_inps.owner.op == T.neg:
return [T.neg(node.op(node_inps.owner.inputs[0]))]
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论