Searched defs:grad_output_slices (Results 1 – 1 of 1) sorted by relevance
91 def _rewrite_output_as_tensor(body_grad_graph, grad_output_slices): argument108 def _rewrite_input_as_indexed_slices(body_grad_graph, grad_output_slices, argument166 def _create_grad_indexed_slices_init(grad_output_slices, forward_input): argument