Searched defs:grad_wrapper (Results 1 – 2 of 2) sorted by relevance
518 def grad_wrapper(*wrapper_args, **grad_kwargs): function
3129 def grad_wrapper(*xs): function3294 def grad_wrapper(*xs): function