Searched refs:topk_sharding (Results 1 – 1 of 1) sorted by relevance
1689 auto topk_sharding = in HandleSort() local1691 auto shard_shape = MakePartitionedShape(replicated_shape, topk_sharding); in HandleSort()