in tt_embeddings_benchmark.py [0:0]
def get_table_batched_offsets_from_dense( merged_indices: torch.Tensor,