Searched refs:result_grads (Results 1 – 1 of 1) sorted by relevance
394 def tape_grad_fn(*result_grads): argument396 result_grads = result_grads[:flat_result_len]398 input_grads, variable_grads = grad_fn(*result_grads, variables=variables)403 input_grads = grad_fn(*result_grads)413 def internal_grad_fn(unused_op, *result_grads): # pylint: disable=unused-variable argument415 return tape_grad_fn(*result_grads)465 def actual_grad_fn(*result_grads): argument468 input_grads, variable_grads = grad_fn(*result_grads, variables=variables)473 input_grads = grad_fn(*result_grads)