Searched refs:AllGatherShardsInternal (Results 1 – 2 of 2) sorted by relevance
232 HloInstruction* AllGatherShardsInternal(
3503 return AllGatherShardsInternal(b, operand, sharding, next_channel_id, in AllGatherShards()3508 HloInstruction* SpmdPartitioner::AllGatherShardsInternal( in AllGatherShardsInternal() function in xla::spmd::SpmdPartitioner