Searched refs:max_right_halo_size (Results 1 – 1 of 1) sorted by relevance
580 int64 max_right_halo_size = in PadFromPartialReplicateShape() local584 padded_src_shape.dimensions(dim) - max_right_halo_size)); in PadFromPartialReplicateShape()810 int64 max_right_halo_size = in ExchangeHalo() local812 if (max_left_halo_size + max_right_halo_size + input_shard_size >= in ExchangeHalo()815 max_right_halo_size > input_shard_size)) { in ExchangeHalo()853 for (int64 i = 0; i < CeilOfRatio(max_right_halo_size, input_shard_size); in ExchangeHalo()866 std::min(max_right_halo_size - input_shard_size * i, input_shard_size); in ExchangeHalo()