Custom grad fn wrapper.
(unused_op, *result_grads)
| 291 | |
| 292 | @ops.RegisterGradient(name) |
| 293 | def internal_grad_fn(unused_op, *result_grads): # pylint: disable=unused-variable |
| 294 | """Custom grad fn wrapper.""" |
| 295 | return tape_grad_fn(*result_grads) |
| 296 | |
| 297 | original_tensors = all_tensors |
| 298 | with ops.get_default_graph().gradient_override_map({"IdentityN": name}): |
nothing calls this directly
no test coverage detected