Home
last modified time | relevance | path

Searched refs:child_iterator_ (Results 1 – 21 of 21) sorted by relevance

/third_party/mindspore/mindspore/ccsrc/minddata/dataset/engine/datasetops/
Dshuffle_op.cc119 child_iterator_ = std::make_unique<ChildIterator>(this, worker_id, child_idx); in operator ()()
127 if (child_iterator_->EofHandled()) { in operator ()()
166 RETURN_IF_NOT_OK(child_iterator_->FetchNextTensorRow(&new_row)); in operator ()()
215 RETURN_IF_NOT_OK(child_iterator_->FetchNextTensorRow(&new_row)); in InitShuffleBuffer()
217 if (child_iterator_->EofHandled()) { in InitShuffleBuffer()
234 RETURN_IF_NOT_OK(child_iterator_->FetchNextTensorRow(&new_row)); in InitShuffleBuffer()
Dbucket_batch_by_length_op.cc61 child_iterator_ = std::make_unique<ChildIterator>(this, 0, 0); in operator ()()
62 RETURN_IF_NOT_OK(child_iterator_->FetchNextTensorRow(&current_row)); in operator ()()
63 while (!child_iterator_->EofHandled()) { in operator ()()
79 RETURN_IF_NOT_OK(child_iterator_->FetchNextTensorRow(&current_row)); in operator ()()
94 RETURN_IF_NOT_OK(child_iterator_->FetchNextTensorRow(&current_row)); in operator ()()
Dbuild_sentence_piece_vocab_op.cc50 child_iterator_ = std::make_unique<ChildIterator>(this, 0, 0); in operator ()()
52 RETURN_IF_NOT_OK(child_iterator_->FetchNextTensorRow(&new_row)); in operator ()()
55 while (child_iterator_->EofHandled() == false) { in operator ()()
58 RETURN_IF_NOT_OK(child_iterator_->FetchNextTensorRow(&new_row)); in operator ()()
60 RETURN_IF_NOT_OK(child_iterator_->FetchNextTensorRow(&new_row)); in operator ()()
Dfilter_op.cc61 child_iterator_ = std::make_unique<ChildIterator>(this, 0, 0); in operator ()()
63 RETURN_IF_NOT_OK(child_iterator_->FetchNextTensorRow(&new_row)); in operator ()()
65 while (child_iterator_->EofHandled() == false) { in operator ()()
69 RETURN_IF_NOT_OK(child_iterator_->FetchNextTensorRow(&new_row)); in operator ()()
73 RETURN_IF_NOT_OK(child_iterator_->FetchNextTensorRow(&new_row)); in operator ()()
Dbarrier_op.cc44 child_iterator_ = std::make_unique<ChildIterator>(this, worker_id, child_idx); in operator ()()
116 RETURN_IF_NOT_OK((child_iterator_)->FetchNextTensorRow(new_row)); in getNextTensorRow()
124 if ((child_iterator_)->EofHandled()) { in getNextTensorRow()
Dbuild_vocab_op.cc92 child_iterator_ = std::make_unique<ChildIterator>(this, 0, 0); in operator ()()
94 RETURN_IF_NOT_OK(child_iterator_->FetchNextTensorRow(&new_row)); in operator ()()
110 while (child_iterator_->EofHandled() == false) { in operator ()()
113 RETURN_IF_NOT_OK(child_iterator_->FetchNextTensorRow(&new_row)); in operator ()()
115 RETURN_IF_NOT_OK(child_iterator_->FetchNextTensorRow(&new_row)); in operator ()()
Ddevice_queue_op.cc134 child_iterator_ = std::make_unique<ChildIterator>(this, 0, 0); in operator ()()
211 RETURN_IF_NOT_OK(child_iterator_->FetchNextTensorRow(&curr_row)); in SendDataToAscend()
255 RETURN_IF_NOT_OK(child_iterator_->FetchNextTensorRow(&curr_row)); in SendDataToAscend()
285 RETURN_IF_NOT_OK(child_iterator_->FetchNextTensorRow(&curr_row)); in SendDataToAscend()
575 RETURN_IF_NOT_OK(child_iterator_->FetchNextTensorRow(&current_row)); in SendDataToGPU()
593 RETURN_IF_NOT_OK(child_iterator_->FetchNextTensorRow(&current_row)); in SendDataToGPU()
600 RETURN_IF_NOT_OK(child_iterator_->FetchNextTensorRow(&current_row)); in SendDataToGPU()
644 while (!(child_iterator_->EofHandled())) { in SendDataToCPU()
646 RETURN_IF_NOT_OK(child_iterator_->FetchNextTensorRow(&curr_row)); in SendDataToCPU()
Dbatch_op.cc104 child_iterator_ = std::make_unique<ChildIterator>(this, 0, 0); in operator ()()
105 RETURN_IF_NOT_OK(child_iterator_->FetchNextTensorRow(&new_row)); in operator ()()
108 while (child_iterator_->EofHandled() == false) { in operator ()()
119 RETURN_IF_NOT_OK(child_iterator_->FetchNextTensorRow(&new_row)); in operator ()()
134 RETURN_IF_NOT_OK(child_iterator_->FetchNextTensorRow(&new_row)); in operator ()()
581 child_iterator_ = std::make_unique<ChildIterator>(this, 0, 0); in GetNextRowPullMode()
Dskip_op.h59 std::unique_ptr<ChildIterator> child_iterator_; // An iterator for fetching. variable
Dtake_op.h69 std::unique_ptr<ChildIterator> child_iterator_; // An iterator for fetching. variable
Dbarrier_op.h90 std::unique_ptr<ChildIterator> child_iterator_; variable
Drename_op.h81 std::unique_ptr<ChildIterator> child_iterator_; // An iterator for fetching. variable
Dshuffle_op.h123 std::unique_ptr<ChildIterator> child_iterator_; // An iterator for fetching. variable
Dbucket_batch_by_length_op.h83 std::unique_ptr<ChildIterator> child_iterator_; variable
Dfilter_op.h86 std::unique_ptr<ChildIterator> child_iterator_; variable
Dbuild_vocab_op.h91 …std::unique_ptr<ChildIterator> child_iterator_; // child iterator for fetching TensorRows 1… variable
Dbuild_sentence_piece_vocab_op.h100 …std::unique_ptr<ChildIterator> child_iterator_; // child iterator for fetching TensorRows 1 by… variable
Ddevice_queue_op.h176 std::unique_ptr<ChildIterator> child_iterator_; variable
Dbatch_op.h276 …std::unique_ptr<ChildIterator> child_iterator_; // child iterator for fetching TensorRows 1 … variable
/third_party/mindspore/mindspore/ccsrc/minddata/dataset/engine/datasetops/map_op/
Dmap_op.cc141 child_iterator_ = std::make_unique<ChildIterator>(this, 0, 0); in operator ()()
143 RETURN_IF_NOT_OK(child_iterator_->FetchNextTensorRow(&new_row)); in operator ()()
166 RETURN_IF_NOT_OK(child_iterator_->FetchNextTensorRow(&new_row)); in operator ()()
179 RETURN_IF_NOT_OK(child_iterator_->FetchNextTensorRow(&new_row)); in operator ()()
Dmap_op.h150 std::unique_ptr<ChildIterator> child_iterator_; // An iterator for fetching. variable