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