Searched refs:halo_exchange_result (Results 1 – 2 of 2) sorted by relevance
79 auto halo_exchange_result = in PadEachPartitionWithHaloExchange() local84 if (halo_exchange_result.has_value()) { in PadEachPartitionWithHaloExchange()85 concat = halo_exchange_result.value(); in PadEachPartitionWithHaloExchange()
466 auto halo_exchange_result = ExchangeHalo( in TileToPartialReplicateHaloExchange() local470 if (halo_exchange_result.has_value()) { in TileToPartialReplicateHaloExchange()471 concat = halo_exchange_result.value(); in TileToPartialReplicateHaloExchange()566 auto halo_exchange_result = ExchangeHalo( in PadFromPartialReplicateShape() local570 if (halo_exchange_result.has_value()) { in PadFromPartialReplicateShape()571 concat = halo_exchange_result.value(); in PadFromPartialReplicateShape()932 auto halo_exchange_result = in ExchangeHaloAndGetValidData() local935 if (!halo_exchange_result) { in ExchangeHaloAndGetValidData()938 auto concat = *halo_exchange_result; in ExchangeHaloAndGetValidData()