| /third_party/mindspore/mindspore/ccsrc/minddata/dataset/engine/ |
| D | tree_adapter.h | 65 TaskGroup *const AllTasks() const { return tree_ ? tree_->AllTasks() : nullptr; } in AllTasks() function
|
| D | execution_tree.h | 168 TaskGroup *const AllTasks() const { return tg_.get(); } in AllTasks() function
|
| /third_party/mindspore/mindspore/ccsrc/minddata/dataset/engine/datasetops/source/ |
| D | image_folder_op.cc | 222 RETURN_IF_NOT_OK(io_block_queues_.Register(tree_->AllTasks())); in LaunchThreadsAndInitOp()
223 RETURN_IF_NOT_OK(folder_name_queue_->Register(tree_->AllTasks())); in LaunchThreadsAndInitOp()
224 RETURN_IF_NOT_OK(image_name_queue_->Register(tree_->AllTasks())); in LaunchThreadsAndInitOp()
225 RETURN_IF_NOT_OK(wait_for_workers_post_.Register(tree_->AllTasks())); in LaunchThreadsAndInitOp()
231 …tree_->AllTasks()->CreateAsyncTask("walk dir", std::bind(&ImageFolderOp::StartAsyncWalk, this), nu… in LaunchThreadsAndInitOp()
|
| D | celeba_op.cc | 55 RETURN_IF_NOT_OK(io_block_queues_.Register(tree_->AllTasks())); in LaunchThreadsAndInitOp()
56 RETURN_IF_NOT_OK(attr_info_queue_->Register(tree_->AllTasks())); in LaunchThreadsAndInitOp()
57 RETURN_IF_NOT_OK(wait_for_workers_post_.Register(tree_->AllTasks())); in LaunchThreadsAndInitOp()
60 …tree_->AllTasks()->CreateAsyncTask("Walking attr file", std::bind(&CelebAOp::ParseAttrFile, this),… in LaunchThreadsAndInitOp()
|
| D | flickr_op.cc | 48 RETURN_IF_NOT_OK(io_block_queues_.Register(tree_->AllTasks())); in LaunchThreadsAndInitOp()
49 RETURN_IF_NOT_OK(wait_for_workers_post_.Register(tree_->AllTasks())); in LaunchThreadsAndInitOp()
|
| D | cifar_op.cc | 53 RETURN_IF_NOT_OK(io_block_queues_.Register(tree_->AllTasks())); in LaunchThreadsAndInitOp()
54 RETURN_IF_NOT_OK(wait_for_workers_post_.Register(tree_->AllTasks())); in LaunchThreadsAndInitOp()
55 RETURN_IF_NOT_OK(tree_->AllTasks()->CreateAsyncTask( in LaunchThreadsAndInitOp()
|
| D | sbu_op.cc | 130 RETURN_IF_NOT_OK(io_block_queues_.Register(tree_->AllTasks())); in LaunchThreadsAndInitOp()
131 RETURN_IF_NOT_OK(wait_for_workers_post_.Register(tree_->AllTasks())); in LaunchThreadsAndInitOp()
|
| D | cityscapes_op.cc | 54 RETURN_IF_NOT_OK(io_block_queues_.Register(tree_->AllTasks())); in LaunchThreadsAndInitOp()
55 RETURN_IF_NOT_OK(wait_for_workers_post_.Register(tree_->AllTasks())); in LaunchThreadsAndInitOp()
|
| D | manifest_op.cc | 51 RETURN_IF_NOT_OK(io_block_queues_.Register(tree_->AllTasks())); in LaunchThreadsAndInitOp()
52 RETURN_IF_NOT_OK(wait_for_workers_post_.Register(tree_->AllTasks())); in LaunchThreadsAndInitOp()
|
| D | div2k_op.cc | 75 RETURN_IF_NOT_OK(io_block_queues_.Register(tree_->AllTasks())); in LaunchThreadsAndInitOp()
76 RETURN_IF_NOT_OK(wait_for_workers_post_.Register(tree_->AllTasks())); in LaunchThreadsAndInitOp()
|
| D | mnist_op.cc | 259 RETURN_IF_NOT_OK(io_block_queues_.Register(tree_->AllTasks())); in LaunchThreadsAndInitOp()
260 RETURN_IF_NOT_OK(wait_for_workers_post_.Register(tree_->AllTasks())); in LaunchThreadsAndInitOp()
|
| D | mindrecord_op.cc | 306 RETURN_IF_NOT_OK(io_block_queues_.Register(tree_->AllTasks())); in LaunchThreadsAndInitOp()
307 RETURN_IF_NOT_OK(wait_for_workers_post_.Register(tree_->AllTasks())); in LaunchThreadsAndInitOp()
|
| D | voc_op.cc | 253 RETURN_IF_NOT_OK(io_block_queues_.Register(tree_->AllTasks())); in LaunchThreadsAndInitOp()
254 RETURN_IF_NOT_OK(wait_for_workers_post_.Register(tree_->AllTasks())); in LaunchThreadsAndInitOp()
|
| D | generator_op.cc | 164 RETURN_IF_NOT_OK(wp_.Register(tree_->AllTasks())); in operator ()()
|
| /third_party/mindspore/mindspore/ccsrc/minddata/dataset/engine/datasetops/ |
| D | filter_op.cc | 44 RETURN_IF_NOT_OK(filter_queues_.Register(tree_->AllTasks())); in LaunchThreadsAndInitOp()
45 RETURN_IF_NOT_OK(worker_queues_.Register(tree_->AllTasks())); in LaunchThreadsAndInitOp()
50 …tree_->AllTasks()->CreateAsyncTask("FilterCollector", std::bind(&FilterOp::Collector, this), nullp… in LaunchThreadsAndInitOp()
|
| D | build_vocab_op.cc | 84 RETURN_IF_NOT_OK(distributor_queue_->Register(tree_->AllTasks())); in operator ()()
85 RETURN_IF_NOT_OK(collector_queue_->Register(tree_->AllTasks())); in operator ()()
90 …tree_->AllTasks()->CreateAsyncTask("collector", std::bind(&BuildVocabOp::CollectorThread, this), n… in operator ()()
|
| D | cache_op.cc | 172 RETURN_IF_NOT_OK(rows_cache_done_.Register(tree_->AllTasks())); in RegisterResources()
173 RETURN_IF_NOT_OK(keys_miss_->Register(tree_->AllTasks())); in RegisterResources()
|
| D | build_sentence_piece_vocab_op.cc | 46 RETURN_IF_NOT_OK(sentence_queue_->Register(tree_->AllTasks())); in operator ()()
47 RETURN_IF_NOT_OK(tree_->AllTasks()->CreateAsyncTask( in operator ()()
|
| D | cache_base_op.cc | 207 RETURN_IF_NOT_OK(wait_for_workers_post_.Register(tree_->AllTasks())); in RegisterResources()
208 RETURN_IF_NOT_OK(io_block_queues_.Register(tree_->AllTasks())); in RegisterResources()
209 RETURN_IF_NOT_OK(prefetch_queues_.Register(tree_->AllTasks())); in RegisterResources()
|
| D | device_queue_op.cc | 130 RETURN_IF_NOT_OK(tree_->AllTasks()->CreateAsyncTask( in operator ()()
150 RETURN_IF_NOT_OK(data_info_queue_ptr_->Register(tree_->AllTasks())); in operator ()()
353 RETURN_IF_NOT_OK(data_info_queue_ptr_->Register(tree_->AllTasks())); in GetDataInfo()
393 RETURN_IF_NOT_OK(receive_queues_.Register(tree_->AllTasks())); in LaunchParallelCopyThread()
396 RETURN_IF_NOT_OK(tree_->AllTasks()->CreateAsyncTask("Push data to GPU queue", in LaunchParallelCopyThread()
|
| D | parallel_op.cc | 80 return (worker_connector_->Register(tree_->AllTasks())); in RegisterWorkerConnectors()
|
| D | cache_merge_op.cc | 55 RETURN_IF_NOT_OK(io_que_->Register(tree_->AllTasks())); in operator ()()
64 …tree_->AllTasks()->CreateAsyncTask("Cleaner", std::bind(&CacheMergeOp::Cleaner, this), nullptr, id… in operator ()()
|
| D | cache_lookup_op.cc | 93 RETURN_IF_NOT_OK(leaf_op_wp_.Register(tree_->AllTasks())); in RegisterResources()
|
| /third_party/mindspore/mindspore/ccsrc/minddata/dataset/engine/datasetops/map_op/ |
| D | map_op.cc | 123 Status rc = local_queues_.Register(tree_->AllTasks()); in operator ()()
124 RETURN_IF_NOT_OK(wait_for_workers_post_.Register(tree_->AllTasks())); in operator ()()
|
| /third_party/mindspore/mindspore/ccsrc/minddata/dataset/engine/consumers/ |
| D | tree_consumer.cc | 41 …CHECK_FAIL_RETURN_UNEXPECTED(tree_adapter_->AllTasks() != nullptr, " Execution tree has not been b… in Terminate()
42 return tree_adapter_->AllTasks()->ServiceStop(); in Terminate()
737 RETURN_UNEXPECTED_IF_NULL(tree->AllTasks()); in Terminate()
738 RETURN_IF_NOT_OK(tree->AllTasks()->ServiceStop()); in Terminate()
|