提交 d8b2a977 authored 作者: Frederic's avatar Frederic

Don't introduce useless mul

上级 7b7b6618
...@@ -4120,6 +4120,9 @@ def local_sum_prod_mul_by_scalar(node): ...@@ -4120,6 +4120,9 @@ def local_sum_prod_mul_by_scalar(node):
if new_op_input_nb_elements != 1: if new_op_input_nb_elements != 1:
mul_inputs.append(new_op_output) mul_inputs.append(new_op_output)
if len(mul_inputs) == 1:
return mul_inputs
else:
return [T.mul(*mul_inputs)] return [T.mul(*mul_inputs)]
if isinstance(node.op, T.Sum) and 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:
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论