Searched refs:_grad_fn (Results 1 – 2 of 2) sorted by relevance
| /external/tensorflow/tensorflow/python/ops/ |
| D | cond_v2.py | 353 def _grad_fn(func_graph, grads): function
395 lambda: _grad_fn(func_graph, grads), [], {},
|
| D | while_v2.py | 649 lambda *args: _grad_fn(ys, xs, args, body_graph),
678 def _grad_fn(ys, xs, args, func_graph): function
|