Searched refs:source_left_halo_sizes (Results 1 – 1 of 1) sorted by relevance
3218 std::vector<OffsetCalculation> source_left_halo_sizes(hlo->shape().rank()); in HandleSelectAndScatter() local3255 source_left_halo_sizes[i] = in HandleSelectAndScatter()3286 source_shard_hlo, source.base_shape(), source_left_halo_sizes[i], in HandleSelectAndScatter()