Searched refs:shuffle_last_row_idx_ (Results 1 – 2 of 2) sorted by relevance
48 shuffle_last_row_idx_(0), in ShuffleOp()64 shuffle_last_row_idx_ = 0; in SelfReset()93 if (shuffle_last_row_idx_ < (shuffle_size_ - 1)) { in AddRowToShuffleBuffer()95 shuffle_last_row_idx_ = (shuffle_buffer_->size()) - 1; in AddRowToShuffleBuffer()97 if (!(*shuffle_buffer_)[shuffle_last_row_idx_].empty()) { in AddRowToShuffleBuffer()101 (*shuffle_buffer_)[shuffle_last_row_idx_] = std::move(new_shuffle_row); in AddRowToShuffleBuffer()135 while (shuffle_last_row_idx_ >= 0) { in operator ()()146 int64_t random_slot = rng_() % (shuffle_last_row_idx_ + 1); in operator ()()155 if (random_slot != shuffle_last_row_idx_) { in operator ()()156 (*shuffle_buffer_)[random_slot] = std::move((*shuffle_buffer_)[shuffle_last_row_idx_]); in operator ()()[all …]
120 int32_t shuffle_last_row_idx_; // Internal tracking of the last slot of our shuffle buffer variable