Home
last modified time | relevance | path

Searched refs:shuffle_buffer_state_ (Results 1 – 2 of 2) sorted by relevance

/third_party/mindspore/mindspore/ccsrc/minddata/dataset/engine/datasetops/
Dshuffle_op.cc49 shuffle_buffer_state_(kShuffleStateInit) {} in ShuffleOp()
65 shuffle_buffer_state_ = kShuffleStateInit; in SelfReset()
80 … out << "\nShuffle size: " << shuffle_size_ << "\nShuffle buffer state: " << shuffle_buffer_state_ in Print()
164 if (shuffle_buffer_state_ == kShuffleStateActive) { in operator ()()
171 shuffle_buffer_state_ = kShuffleStateDrain; in operator ()()
177 if (shuffle_buffer_state_ == kShuffleStateDrain) { in operator ()()
207 if (shuffle_buffer_state_ != kShuffleStateInit) { in InitShuffleBuffer()
240 shuffle_buffer_state_ = kShuffleStateActive; // Transition to the active state in InitShuffleBuffer()
244 shuffle_buffer_state_ = kShuffleStateDrain; in InitShuffleBuffer()
Dshuffle_op.h121 int32_t shuffle_buffer_state_; // State tracking for the shuffle buffer phases of work variable