Searched refs:custom_call_sharding (Results 1 – 1 of 1) sorted by relevance
1843 auto custom_call_sharding = in HandleCustomCall() local1846 MakePartitionedShape(replicated_shape, custom_call_sharding); in HandleCustomCall()1849 topk->set_sharding(custom_call_sharding); in HandleCustomCall()