Searched refs:result_grads (Results 1 – 2 of 2) sorted by relevance
487 def tape_grad_fn(*result_grads): argument489 result_grads = result_grads[:flat_result_len]491 input_grads, variable_grads = grad_fn(*result_grads, variables=variables)496 input_grads = grad_fn(*result_grads)506 def internal_grad_fn(unused_op, *result_grads): # pylint: disable=unused-variable argument508 return tape_grad_fn(*result_grads)558 def actual_grad_fn(*result_grads): argument561 input_grads, variable_grads = grad_fn(*result_grads, variables=variables)566 input_grads = grad_fn(*result_grads)
461 def gradient_func(unused_op, *result_grads): argument466 result_grads = [468 for (x, t) in zip(result_grads, func.graph.inputs)471 return func(*result_grads)