Searched refs:tape_grad_fn (Results 1 – 1 of 1) sorted by relevance
394 def tape_grad_fn(*result_grads): function415 return tape_grad_fn(*result_grads)428 f.__name__, all_tensors, original_tensors, tape_grad_fn)