Searched defs:body_grad_graph (Results 1 – 2 of 2) sorted by relevance
| /external/tensorflow/tensorflow/python/ops/ |
| D | while_v2_indexed_slices_rewriter.py | 31 def rewrite_grad_indexed_slices(grads, body_grad_graph, loop_vars, argument
92 def _rewrite_output_as_tensor(body_grad_graph, grad_output_slices): argument
109 def _rewrite_input_as_indexed_slices(body_grad_graph, grad_output_slices, argument
|
| D | while_v2.py | 717 def _resolve_grad_captures(body_graph, body_grad_graph, while_op): argument
758 def _get_structured_grad_output(outputs, grads, body_grad_graph): argument
|