Searched refs:old_output_slices (Results 1 – 1 of 1) sorted by relevance
213 def _rewrite_grad_indexed_slices_output(old_output_slices, new_input_slices): argument242 values = rewrite(old_output_slices.values.op, new_input_slices.values)243 indices = rewrite(old_output_slices.indices.op, new_input_slices.indices)249 output_slices, old_output_slices): argument272 old_output_slices)278 func_graph.flatten(old_output_slices)[0])