Searched refs:replicated_sort (Results 1 – 1 of 1) sorted by relevance
1736 PartitionedHlo replicated_sort(final_sort, final_sort->shape(), in HandleSort() local1738 SetPartitionedHlo(hlo, replicated_sort.Reshard(hlo->sharding())); in HandleSort()1917 PartitionedHlo replicated_sort(sort, replicated_shape, in HandleCustomCall() local1923 replicated_sort.hlo()->shape().tuple_shapes(0), replicated_sort.hlo(), in HandleCustomCall()1927 replicated_sort.hlo()->shape().tuple_shapes(1), replicated_sort.hlo(), in HandleCustomCall()