提交 aa5307ce 编写于 作者: L Lfc1993 提交者: Tao Luo

fix python3 run_time_error in ops. test=develop (#17170)

fix python3 run_time_error in layers.ops caused by locals()
上级 e4a52e08
...@@ -102,10 +102,9 @@ _hard_shrink_ = generate_layer_fn('hard_shrink') ...@@ -102,10 +102,9 @@ _hard_shrink_ = generate_layer_fn('hard_shrink')
def hard_shrink(x, threshold=None): def hard_shrink(x, threshold=None):
locals_var = locals().keys() locals_var = locals()
kwargs = dict() kwargs = dict()
for name in locals_var: for name, val in locals_var.items():
val = locals()[name]
if val is not None: if val is not None:
kwargs[name] = val kwargs[name] = val
return _hard_shrink_(**kwargs) return _hard_shrink_(**kwargs)
...@@ -124,10 +123,9 @@ _cum_sum_ = generate_layer_fn('cumsum') ...@@ -124,10 +123,9 @@ _cum_sum_ = generate_layer_fn('cumsum')
def cumsum(x, axis=None, exclusive=None, reverse=None): def cumsum(x, axis=None, exclusive=None, reverse=None):
locals_var = locals().keys() locals_var = locals()
kwargs = dict() kwargs = dict()
for name in locals_var: for name, val in locals_var.items():
val = locals()[name]
if val is not None: if val is not None:
kwargs[name] = val kwargs[name] = val
return _cum_sum_(**kwargs) return _cum_sum_(**kwargs)
...@@ -146,10 +144,9 @@ _thresholded_relu_ = generate_layer_fn('thresholded_relu') ...@@ -146,10 +144,9 @@ _thresholded_relu_ = generate_layer_fn('thresholded_relu')
def thresholded_relu(x, threshold=None): def thresholded_relu(x, threshold=None):
locals_var = locals().keys() locals_var = locals()
kwargs = dict() kwargs = dict()
for name in locals_var: for name, val in locals_var.items():
val = locals()[name]
if val is not None: if val is not None:
kwargs[name] = val kwargs[name] = val
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册