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

Small opt speed up, code clean up

上级 e4db03ae
...@@ -4903,9 +4903,10 @@ register_canonicalize(local_inv_canon) ...@@ -4903,9 +4903,10 @@ register_canonicalize(local_inv_canon)
@gof.local_optimizer([T.pow]) @gof.local_optimizer([T.pow])
def local_pow_canonicalize(node): def local_pow_canonicalize(node):
if node.op == T.pow: if node.op == T.pow:
if local_mul_canonizer.get_constant(node.inputs[1]) == 0: cst = local_mul_canonizer.get_constant(node.inputs[1])
if cst == 0:
return [broadcast_like(1, node.outputs[0], node.fgraph)] return [broadcast_like(1, node.outputs[0], node.fgraph)]
if local_mul_canonizer.get_constant(node.inputs[1]) == 1: if cst == 1:
return [broadcast_like(node.inputs[0], node.outputs[0], node.fgraph)] return [broadcast_like(node.inputs[0], node.outputs[0], node.fgraph)]
else: else:
return False return False
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论