Returns the FuncGraph representation of _grad_fn.
(func_graph, grads, name)
| 388 | |
| 389 | |
| 390 | def _create_grad_func(func_graph, grads, name): |
| 391 | """Returns the FuncGraph representation of _grad_fn.""" |
| 392 | return func_graph_module.func_graph_from_py_func( |
| 393 | name, |
| 394 | lambda: _grad_fn(func_graph, grads), [], {}, |
| 395 | func_graph=_CondGradFuncGraph(name, func_graph)) |
| 396 | |
| 397 | |
| 398 | def _resolve_grad_inputs(cond_graph, grad_graph): |
no test coverage detected