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

fix python3 run_time_error in layers.ops caused by locals()
feature/fluid_trt_int8
Lfc1993 6 years ago committed by Tao Luo
parent e4a52e082f
commit aa5307ce5b

@ -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')
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')
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

Loading…
Cancel
Save