| /third_party/mindspore/mindspore/ccsrc/minddata/dataset/engine/datasetops/ |
| D | cache_op.cc | 63 Status CacheOp::CacheAllRows(int32_t worker_id) { in CacheAllRows()
163 Status CacheOp::WorkerEntry(int32_t worker_id) { in WorkerEntry()
180 Status CacheOp::EoeReceived(int32_t worker_id) { in EoeReceived()
185 Status CacheOp::EofReceived(int32_t worker_id) { in EofReceived()
|
| D | repeat_op.cc | 60 Status RepeatOp::GetNextRow(TensorRow *row, int32_t worker_id, bool retry_if_eoe) { in GetNextRow()
82 Status RepeatOp::EoeReceived(int32_t worker_id) { in EoeReceived()
113 Status RepeatOp::EofReceived(int32_t worker_id) { in EofReceived()
|
| D | project_op.cc | 51 Status ProjectOp::GetNextRow(TensorRow *row, int32_t worker_id, bool retry_if_eoe) { in GetNextRow()
100 Status ProjectOp::EoeReceived(int32_t worker_id) { in EoeReceived()
105 Status ProjectOp::EofReceived(int32_t worker_id) { return Status::OK(); } in EofReceived()
|
| D | epoch_ctrl_op.cc | 43 Status EpochCtrlOp::GetNextRow(TensorRow *row, int32_t worker_id, bool retry_if_eoe) { in GetNextRow()
61 Status EpochCtrlOp::EoeReceived(int32_t worker_id) { in EoeReceived()
|
| D | zip_op.cc | 32 Status ZipOp::getNextZippedRow(TensorRow *const new_zip_row, int32_t *skip_child, int32_t worker_id, in getNextZippedRow()
52 Status ZipOp::drainPipeline(int32_t skip_child, int32_t worker_id, bool retry_if_eoe) { in drainPipeline()
122 Status ZipOp::GetNextRow(TensorRow *row, int32_t worker_id, bool retry_if_eoe) { in GetNextRow()
|
| D | shuffle_op.cc | 117 int32_t worker_id = 0; in operator ()() local
251 Status ShuffleOp::EoeReceived(int32_t worker_id) { in EoeReceived()
|
| D | cache_merge_op.cc | 72 Status CacheMergeOp::WorkerEntry(int32_t worker_id) { in WorkerEntry()
219 Status CacheMergeOp::EoeReceived(int32_t worker_id) { in EoeReceived()
226 Status CacheMergeOp::EofReceived(int32_t worker_id) { in EofReceived()
|
| D | cache_base_op.cc | 76 std::vector<row_id_type> &keys) -> Status { in FetchSamplesToWorkers()
165 Status CacheBase::FetchFromCache(int32_t worker_id) { in FetchFromCache()
272 Status CacheBase::Prefetcher(int32_t worker_id) { in Prefetcher()
|
| D | dataset_op.cc | 262 Status DatasetOp::GetNextRow(TensorRow *row, int32_t worker_id, bool 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()
|
| D | take_op.cc | 44 Status TakeOp::GetNextRow(TensorRow *row, int32_t worker_id, bool retry_if_eoe) { in GetNextRow()
|
| D | skip_op.cc | 49 Status SkipOp::GetNextRow(TensorRow *row, int32_t worker_id, bool retry_if_eoe) { in GetNextRow()
|
| D | device_queue_op.cc | 80 void DeviceQueueOp::ReleaseData(void *addr, int32_t worker_id) { in ReleaseData()
87 Status DeviceQueueOp::EoeReceived(int32_t worker_id) { in EoeReceived()
537 Status DeviceQueueOp::WorkerEntry(int32_t worker_id) { in WorkerEntry()
616 const int32_t &worker_id) { in MallocForGPUData()
|
| D | cache_lookup_op.cc | 45 Status CacheLookupOp::WorkerEntry(int32_t worker_id) { in WorkerEntry()
|
| D | barrier_op.cc | 42 const int32_t worker_id = 0; in operator ()() local
|
| D | rename_op.cc | 36 Status RenameOp::GetNextRow(TensorRow *row, int32_t worker_id, bool retry_if_eoe) { in GetNextRow()
|
| /third_party/mindspore/mindspore/ccsrc/minddata/dataset/engine/ |
| D | connector.h | 97 …virtual Status Pop(int32_t worker_id, // The worker-id of the caller. See the requirement at the … in Pop()
118 Status Push(int32_t worker_id, const T &el) noexcept { in Push()
131 virtual Status Push(int32_t worker_id, T &&el) noexcept { in Push()
|
| D | gpu_item_connector.h | 48 Status Pop(int32_t worker_id, std::vector<device::DataItemGpu> *result) noexcept override { in Pop()
|
| D | jagged_connector.h | 45 Status Pop(int32_t worker_id, TensorRow *result) noexcept override { in Pop()
|
| D | dataset_iterator.cc | 166 ChildIterator::ChildIterator(DatasetOp *current_op, int32_t worker_id, int32_t child_idx) in ChildIterator()
|
| /third_party/mindspore/mindspore/ccsrc/minddata/dataset/engine/datasetops/source/ |
| D | random_data_op.cc | 171 Status RandomDataOp::EpochSync(int32_t worker_id, bool *quitting) { in EpochSync()
229 Status RandomDataOp::WorkerEntry(int32_t worker_id) { in WorkerEntry()
269 Status RandomDataOp::CreateRandomRow(int32_t worker_id, TensorRow *new_row) { in CreateRandomRow()
|
| D | mappable_leaf_op.cc | 91 Status MappableLeafOp::WorkerEntry(int32_t worker_id) { in WorkerEntry()
|
| D | mindrecord_op.cc | 146 Status MindRecordOp::WorkerEntry(int32_t worker_id) { in WorkerEntry()
197 Status MindRecordOp::GetRowFromReader(TensorRow *fetched_row, uint64_t row_id, int32_t worker_id) { in GetRowFromReader()
|
| /third_party/mindspore/mindspore/ccsrc/minddata/dataset/engine/cache/perf/ |
| D | cache_pipeline_run.cc | 226 int32_t worker_id = 0; in RunFirstEpoch() local
260 Status CachePipelineRun::WriterWorkerEntry(int32_t worker_id) { in WriterWorkerEntry()
371 int32_t worker_id = 0; in RunReadEpoch() local
401 Status CachePipelineRun::ReaderWorkerEntry(int32_t worker_id) { in ReaderWorkerEntry()
|
| /third_party/mindspore/mindspore/ccsrc/minddata/dataset/engine/datasetops/map_op/ |
| D | map_op.cc | 77 Status MapOp::FetchNextWork(uint32_t worker_id, TensorRow *row, std::vector<std::shared_ptr<MapJob>… in FetchNextWork()
200 Status MapOp::WorkerEntry(int32_t worker_id) { in WorkerEntry()
|
| /third_party/mindspore/mindspore/ccsrc/fl/server/kernel/round/ |
| D | reconstruct_secrets_kernel.cc | 131 std::string worker_id = std::to_string(DistributedCountService::GetInstance().local_rank()); in OnLastCountEvent() local
|