Searched refs:shuffle_size_ (Results 1 – 4 of 4) sorted by relevance
31 : shuffle_size_(shuffle_size), shuffle_seed_(GetSeed()), reset_every_epoch_(reset_every_epoch) { in ShuffleNode()36 auto node = std::make_shared<ShuffleNode>(nullptr, shuffle_size_, reset_every_epoch_); in Copy()41 out << (Name() + "(shuffle_size:" + std::to_string(shuffle_size_) + in Print()47 …auto op = std::make_shared<ShuffleOp>(shuffle_size_, shuffle_seed_, connector_que_size_, reset_eve… in Build()57 if (shuffle_size_ <= 1) { in ValidateParams()58 …std::string err_msg = "ShuffleNode: Invalid input, shuffle_size: " + std::to_string(shuffle_size_); in ValidateParams()68 args["buffer_size"] = shuffle_size_; in to_json()
57 int32_t ShuffleSize() const { return shuffle_size_; } in ShuffleSize()75 int32_t shuffle_size_;
43 shuffle_size_(shuffle_size), in ShuffleOp()75 out << " [shuffle size: " << shuffle_size_ << "]\n"; in Print()80 … out << "\nShuffle size: " << shuffle_size_ << "\nShuffle buffer state: " << shuffle_buffer_state_ in Print()93 if (shuffle_last_row_idx_ < (shuffle_size_ - 1)) { in AddRowToShuffleBuffer()229 while (!new_row.empty() && shuffle_buffer_->size() < static_cast<size_t>(shuffle_size_ - 1)) { in InitShuffleBuffer()
109 int32_t shuffle_size_; // User config for the size of the shuffle buffer (number of rows) variable