Searched refs:shuffle_buffer_ (Results 1 – 2 of 2) sorted by relevance
47 shuffle_buffer_(std::make_unique<TensorTable>()), in ShuffleOp()63 shuffle_buffer_ = std::make_unique<TensorTable>(); in SelfReset()94 shuffle_buffer_->push_back(std::move(new_shuffle_row)); 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()147 TensorRow random_row = std::move((*shuffle_buffer_)[random_slot]); in operator ()()156 (*shuffle_buffer_)[random_slot] = std::move((*shuffle_buffer_)[shuffle_last_row_idx_]); in operator ()()229 while (!new_row.empty() && shuffle_buffer_->size() < static_cast<size_t>(shuffle_size_ - 1)) { in InitShuffleBuffer()
119 std::unique_ptr<TensorTable> shuffle_buffer_; variable