Home
last modified time | relevance | path

Searched refs:max_left_halo_size (Results 1 – 1 of 1) sorted by relevance

/external/tensorflow/tensorflow/compiler/xla/service/spmd/
Dspmd_partitioner_util.cc809 int64 max_left_halo_size = left_halo_size_function.MaxInRange(1, shard_count); in ExchangeHalo() local
812 if (max_left_halo_size + max_right_halo_size + input_shard_size >= in ExchangeHalo()
814 (max_left_halo_size > input_shard_size || in ExchangeHalo()
819 for (int64 i = CeilOfRatio(max_left_halo_size, input_shard_size) - 1; i >= 0; in ExchangeHalo()
832 std::min(max_left_halo_size - input_shard_size * i, input_shard_size); in ExchangeHalo()
940 int64 max_left_halo_size = left_halo_size_function.MaxInRange(1, shard_count); in ExchangeHaloAndGetValidData() local
946 std::max(std::max(int64{0}, max_left_halo_size), in ExchangeHaloAndGetValidData()
959 std::max(int64{0}, max_left_halo_size)); in ExchangeHaloAndGetValidData()