Searched refs:result_grads (Results 1 – 1 of 1) sorted by relevance
215 def tape_grad_fn(*result_grads): argument217 result_grads = result_grads[:len(flat_result)]219 input_grads, variable_grads = grad_fn(*result_grads, variables=variables)224 input_grads = grad_fn(*result_grads)234 def internal_grad_fn(unused_op, *result_grads): # pylint: disable=unused-variable argument236 return tape_grad_fn(*result_grads)277 def actual_grad_fn(*result_grads): argument280 input_grads, variable_grads = grad_fn(*result_grads, variables=variables)285 input_grads = grad_fn(*result_grads)