Searched defs:grad_output_slices (Results 1 – 1 of 1) sorted by relevance
94 def _rewrite_output_as_tensor(body_grad_graph, grad_output_slices): argument111 def _rewrite_input_as_indexed_slices(body_grad_graph, grad_output_slices, argument169 def _create_grad_indexed_slices_init(grad_output_slices, forward_input): argument