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

Code simplication

上级 bf775744
...@@ -5783,7 +5783,7 @@ def local_grad_log_erfc_neg(node): ...@@ -5783,7 +5783,7 @@ def local_grad_log_erfc_neg(node):
# The mul is optional. # The mul is optional.
if node.inputs[0].owner.op != T.mul: if node.inputs[0].owner.op != T.mul:
mul = None mul = None
y = 1 y = []
if not node.inputs[0].owner or node.inputs[0].owner.op != T.exp: if not node.inputs[0].owner or node.inputs[0].owner.op != T.exp:
return False return False
exp = node.inputs[0] exp = node.inputs[0]
...@@ -5902,7 +5902,7 @@ def local_grad_log_erfc_neg(node): ...@@ -5902,7 +5902,7 @@ def local_grad_log_erfc_neg(node):
elif x.dtype == 'float64': elif x.dtype == 'float64':
threshold = 26.641747557 threshold = 26.641747557
ret = T.switch(x < threshold, true_div_no_mul, stab_value) ret = T.switch(x < threshold, true_div_no_mul, stab_value)
if y != 1: if y:
ret = T.mul(ret, *y) ret = T.mul(ret, *y)
ret.values_eq_approx = values_eq_approx_remove_inf_nan ret.values_eq_approx = values_eq_approx_remove_inf_nan
return [ret] return [ret]
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论