Searched refs:data_left_halo_sizes (Results 1 – 1 of 1) sorted by relevance
3216 std::vector<OffsetCalculation> data_left_halo_sizes(hlo->shape().rank()); in HandleSelectAndScatter() local3263 data_left_halo_sizes[i] = in HandleSelectAndScatter()3304 data_shard_hlo, operand.base_shape(), data_left_halo_sizes[i], in HandleSelectAndScatter()3346 data_left_halo_sizes[i].Calculate(partition_ordinals[i], &b_); in HandleSelectAndScatter()3347 if (data_left_halo_sizes[i].Calculate(0) == pad_low) { in HandleSelectAndScatter()