Searched refs:tape_grad_fn (Results 1 – 1 of 1) sorted by relevance
215 def tape_grad_fn(*result_grads): function236 return tape_grad_fn(*result_grads)249 f.__name__, all_tensors, original_tensors, tape_grad_fn)