Searched refs:old_output_slices (Results 1 – 1 of 1) sorted by relevance
204 def _rewrite_grad_indexed_slices_output(old_output_slices, new_input_slices): argument233 values = rewrite(old_output_slices.values.op, new_input_slices.values)234 indices = rewrite(old_output_slices.indices.op, new_input_slices.indices)240 output_slices, old_output_slices): argument262 structured_idx = graph.structured_outputs.index(old_output_slices)266 flat_idx = graph.outputs.index(func_graph.flatten(old_output_slices)[0])