Searched defs:dst_sharding (Results 1 – 1 of 1) sorted by relevance
410 const HloSharding& src_sharding, const HloSharding& dst_sharding, in TileToPartialReplicateHaloExchange()504 const HloSharding& src_sharding, const HloSharding& dst_sharding, in PadFromPartialReplicateShape()