Searched refs:halo_exchange (Results 1 – 1 of 1) sorted by relevance
946 auto halo_exchange = TileToPartialReplicateHaloExchange( in ReshardToPartialReplicateWithAllGather() local951 if (!halo_exchange.has_value()) { in ReshardToPartialReplicateWithAllGather()954 auto halo_exchange_hlo = halo_exchange.value(); in ReshardToPartialReplicateWithAllGather()