Home
last modified time | relevance | path

Searched refs:out_connector_ (Results 1 – 17 of 17) sorted by relevance

/third_party/mindspore/mindspore/ccsrc/minddata/dataset/engine/datasetops/
Ddataset_op.cc50 out_connector_(nullptr), in DatasetOp()
216 out_connector_ = std::make_unique<DbConnector>(num_producers, // The number of producers in CreateConnector()
222 out_connector_ = nullptr; in CreateConnector()
264 RETURN_IF_NOT_OK(out_connector_->PopWithRetry(static_cast<int>(worker_id), row, retry_if_eoe)); in GetNextRow()
302 Status DatasetOp::EoeReceived(int32_t worker_id) { return out_connector_->SendEOE(worker_id); } in EoeReceived()
307 Status DatasetOp::EofReceived(int32_t worker_id) { return out_connector_->SendEOF(worker_id); } in EofReceived()
319 if (out_connector_) { in PrepareOperator()
320 RETURN_IF_NOT_OK(out_connector_->Register(tree_->AllTasks())); in PrepareOperator()
Dshuffle_op.cc128 RETURN_IF_NOT_OK(out_connector_->SendEOF()); in operator ()()
149 RETURN_IF_NOT_OK(out_connector_->Add(std::move(random_row))); in operator ()()
185 RETURN_IF_NOT_OK(out_connector_->SendEOE()); in operator ()()
219 RETURN_IF_NOT_OK(out_connector_->SendEOF()); in InitShuffleBuffer()
Ddataset_op.h260 return out_connector_->size(); in ConnectorSize()
268 …return out_connector_ == nullptr ? int64_t(-1) : static_cast<int64_t>(out_connector_->out_rows_cou… in ConnectorOutRowsCount()
275 return out_connector_->capacity(); in ConnectorCapacity()
384 std::unique_ptr<DbConnector> out_connector_; // Output Connector variable
Dbarrier_op.cc64 RETURN_IF_NOT_OK(out_connector_->Add(std::move(new_row))); in operator ()()
71 RETURN_IF_NOT_OK(out_connector_->SendEOE()); in operator ()()
77 RETURN_IF_NOT_OK(out_connector_->SendEOF()); in operator ()()
Dbucket_batch_by_length_op.cc92 RETURN_IF_NOT_OK(out_connector_->SendEOE()); in operator ()()
96 RETURN_IF_NOT_OK(out_connector_->SendEOF()); in operator ()()
161 RETURN_IF_NOT_OK(out_connector_->Add(std::move(batched_bucket), 0)); in PadAndBatchBucket()
Dfilter_op.cc186 RETURN_IF_NOT_OK(out_connector_->SendEOE(static_cast<int>(out_task_id))); in Collector()
188 … RETURN_IF_NOT_OK(out_connector_->Add(std::move(in_pair.first), static_cast<int>(out_task_id))); in Collector()
194 RETURN_IF_NOT_OK(out_connector_->SendEOF(static_cast<int>(out_task_id))); in Collector()
Dcache_merge_op.cc87 RETURN_IF_NOT_OK(out_connector_->Add(std::move(new_row), worker_id)); in WorkerEntry()
222 return out_connector_->SendEOE(worker_id); in EoeReceived()
229 return out_connector_->SendEOF(worker_id); in EofReceived()
Dbuild_sentence_piece_vocab_op.cc96 RETURN_IF_NOT_OK(out_connector_->SendEOE()); in SentenceThread()
97 RETURN_IF_NOT_OK(out_connector_->SendEOF()); in SentenceThread()
Dcache_base_op.cc176 RETURN_IF_NOT_OK(out_connector_->SendEOF(worker_id)); in FetchFromCache()
178 RETURN_IF_NOT_OK(out_connector_->SendEOE(worker_id)); in FetchFromCache()
198 RETURN_IF_NOT_OK(out_connector_->Add(std::move(row), worker_id)); in FetchFromCache()
Dbuild_vocab_op.cc188 RETURN_IF_NOT_OK(out_connector_->SendEOE()); in CollectorThread()
189 RETURN_IF_NOT_OK(out_connector_->SendEOF()); in CollectorThread()
Dbatch_op.cc235 RETURN_IF_NOT_OK(out_connector_->SendEOE(workerId)); in WorkerEntry()
237 RETURN_IF_NOT_OK(out_connector_->SendEOF(workerId)); in WorkerEntry()
241 RETURN_IF_NOT_OK(out_connector_->Add(std::move(new_row), workerId)); in WorkerEntry()
/third_party/mindspore/mindspore/ccsrc/minddata/dataset/engine/datasetops/source/
Dmappable_leaf_op.cc103 RETURN_IF_NOT_OK(out_connector_->SendEOE(worker_id)); in WorkerEntry()
105 RETURN_IF_NOT_OK(out_connector_->SendEOF(worker_id)); in WorkerEntry()
112 RETURN_IF_NOT_OK(out_connector_->Add(std::move(trow), worker_id)); in WorkerEntry()
Dgenerator_op.cc212 if (!new_row.empty()) RETURN_IF_NOT_OK(out_connector_->Add(std::move(new_row))); in operator ()()
217 RETURN_IF_NOT_OK(out_connector_->SendEOE()); in operator ()()
221 RETURN_IF_NOT_OK(out_connector_->SendEOF()); in operator ()()
Dnonmappable_leaf_op.cc84 RETURN_IF_NOT_OK(out_connector_->Add(std::move(fetched_row), 0)); in operator ()()
110 RETURN_IF_NOT_OK(out_connector_->SendEOE()); in operator ()()
123 RETURN_IF_NOT_OK(out_connector_->SendEOF()); in operator ()()
Drandom_data_op.cc193 RETURN_IF_NOT_OK(out_connector_->SendEOE(eoe_worker_id_)); in EpochSync()
200 RETURN_IF_NOT_OK(out_connector_->SendEOF(eof_worker_id)); in EpochSync()
256 RETURN_IF_NOT_OK(out_connector_->Add(std::move(new_row), worker_id)); in WorkerEntry()
Dmindrecord_op.cc161 RETURN_IF_NOT_OK(out_connector_->SendEOE(worker_id)); in WorkerEntry()
166 RETURN_IF_NOT_OK(out_connector_->SendEOF(worker_id)); in WorkerEntry()
191 RETURN_IF_NOT_OK(out_connector_->Add(std::move(fetched_row), worker_id)); in WorkerEntry()
/third_party/mindspore/mindspore/ccsrc/minddata/dataset/engine/datasetops/map_op/
Dmap_op.cc224 RETURN_IF_NOT_OK(out_connector_->SendEOE(worker_id)); in WorkerEntry()
227 RETURN_IF_NOT_OK(out_connector_->SendEOF(worker_id)); in WorkerEntry()
239 RETURN_IF_NOT_OK(out_connector_->Add(std::move(out_row), static_cast<int>(worker_id))); in WorkerEntry()