revert line

This commit is contained in:
fsx950223 2020-10-29 01:52:14 +00:00
parent 295d783184
commit 5ba75f4060

View File

@ -514,6 +514,7 @@ def recompute_grad(f):
current_var_scope = variable_scope.get_variable_scope()
with tape_lib.stop_recording():
result = f(*args, **kwargs)
def grad_wrapper(*wrapper_args, **grad_kwargs):
"""Wrapper function to accomodate lack of kwargs in graph mode decorator."""