提交 2ec5abc5 authored 作者: Amjad Almahairi's avatar Amjad Almahairi

name change, remove unnecessary check

上级 3b0e4efe
...@@ -4397,16 +4397,15 @@ def local_mul_to_sqr(node): ...@@ -4397,16 +4397,15 @@ def local_mul_to_sqr(node):
if node.inputs[0] is node.inputs[1]: if node.inputs[0] is node.inputs[1]:
return [T.sqr(node.inputs[0])] return [T.sqr(node.inputs[0])]
@register_specialize @register_canonicalize
@gof.local_optimizer([T.int_div, T.floor_div]) @gof.local_optimizer([T.int_div, T.floor_div])
def local_div_by_one(node): def local_intdiv_by_one(node):
"""x // 1 -> x """x // 1 -> x
""" """
if node.op in [T.int_div]: if node.op in [T.int_div]:
if len(node.inputs) == 2: if isinstance(node.inputs[1], T.TensorConstant) and \
if isinstance(node.inputs[1], T.TensorConstant) and \ node.inputs[1].value == 1:
node.inputs[1].value == 1: return [node.inputs[0]]
return [node.inputs[0]]
@gof.local_optimizer([T.pow]) @gof.local_optimizer([T.pow])
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论