Searched refs:slice_batch_indices (Results 1 – 1 of 1) sorted by relevance
322 def slice_batch_indices(indices): function354 indices_dataset = indices_dataset.flat_map(slice_batch_indices)