Searched refs:grad_wrapper (Results 1 – 2 of 2) sorted by relevance
518 def grad_wrapper(*wrapper_args, **grad_kwargs): function556 return result, grad_wrapper
3129 def grad_wrapper(*xs): function3133 return nest.pack_sequence_as(value, grad_wrapper(*nest.flatten(value)))3294 def grad_wrapper(*xs): function3314 return nest.pack_sequence_as(value, grad_wrapper(*nest.flatten(value)))