Searched refs:grad_kwargs (Results 1 – 1 of 1) sorted by relevance
518 def grad_wrapper(*wrapper_args, **grad_kwargs): argument525 variables = grad_kwargs.get("variables")