Searched refs:_rewrite_grad_indexed_slices_output (Results 1 – 1 of 1) sorted by relevance
148 new_output_slices = _rewrite_grad_indexed_slices_output(grad_output_slices,204 def _rewrite_grad_indexed_slices_output(old_output_slices, new_input_slices): function