/third_party/mindspore/mindspore/ccsrc/minddata/dataset/util/ |
D | task_manager.h | 87 static Task *FindMe(); 171 Task *my_task = TaskManager::FindMe(); in is_interrupted() 176 Task *my_task = TaskManager::FindMe(); in is_master_thread() 181 Task *my_task = TaskManager::FindMe(); in GetInterruptStatus()
|
D | task_manager.cc | 93 Task *TaskManager::FindMe() { in FindMe() function in mindspore::dataset::TaskManager 186 TaskManager::FindMe()->Post(); in WatchDog() 282 auto pMytask = TaskManager::FindMe(); in CreateAsyncTask()
|
/third_party/mindspore/mindspore/ccsrc/minddata/dataset/engine/gnn/ |
D | graph_data_server.cc | 65 TaskManager::FindMe()->Post(); in InitGraphDataImpl() 77 TaskManager::FindMe()->Post(); in StartAsyncRpcService() 84 TaskManager::FindMe()->Post(); in JudgeAutoShutdownServer()
|
/third_party/mindspore/tests/ut/cpp/dataset/ |
D | interrupt_test.cc | 42 TaskManager::FindMe()->Post(); in TEST_F() 60 TaskManager::FindMe()->Post(); in TEST_F()
|
D | task_manager_test.cc | 35 TaskManager::FindMe()->Post(); in TEST_F() 69 TaskManager::FindMe()->Post(); in TEST_F()
|
D | connector_test.cc | 250 TaskManager::FindMe()->Post(); in SerialWorkerPull() 279 TaskManager::FindMe()->Post(); in FirstWorkerPush() 299 TaskManager::FindMe()->Post(); in MidWorkerJob()
|
D | btree_test.cc | 114 TaskManager::FindMe()->Post(); in TEST_F() 125 TaskManager::FindMe()->Post(); in TEST_F()
|
D | circular_pool_test.cc | 49 TaskManager::FindMe()->Post(); in TestMem()
|
/third_party/mindspore/mindspore/ccsrc/minddata/dataset/engine/datasetops/ |
D | device_queue_op.cc | 133 TaskManager::FindMe()->Post(); in operator ()() 289 if (stop_send_ || !TaskManager::FindMe()->Interrupted()) { in SendDataToAscend() 406 TaskManager::FindMe()->Post(); in PushDataToGPU() 484 if (!TaskManager::FindMe()->Interrupted() && !GpuBufferMgr::GetInstance().IsClosed()) { in PushDataToGPU() 498 if (!TaskManager::FindMe()->Interrupted() && !GpuBufferMgr::GetInstance().IsClosed()) { in PushDataToGPU() 511 while (!GpuBufferMgr::GetInstance().IsClosed() && !TaskManager::FindMe()->Interrupted()) { in RetryPushData() 540 TaskManager::FindMe()->Post(); in WorkerEntry() 592 if (!TaskManager::FindMe()->Interrupted() && !GpuBufferMgr::GetInstance().IsClosed()) { in SendDataToGPU() 599 if (!TaskManager::FindMe()->Interrupted() && !GpuBufferMgr::GetInstance().IsClosed()) { in SendDataToGPU() 700 TaskManager::FindMe()->Post(); in DetectFirstBatch()
|
D | cache_merge_op.cc | 66 TaskManager::FindMe()->Post(); in operator ()() 73 TaskManager::FindMe()->Post(); in WorkerEntry() 96 TaskManager::FindMe()->Post(); in CacheMissWorkerEntry() 154 TaskManager::FindMe()->Post(); in Cleaner()
|
D | cache_lookup_op.cc | 39 TaskManager::FindMe()->Post(); in operator ()() 46 TaskManager::FindMe()->Post(); in WorkerEntry()
|
D | filter_op.cc | 58 TaskManager::FindMe()->Post(); in operator ()() 120 TaskManager::FindMe()->Post(); in WorkerEntry() 173 TaskManager::FindMe()->Post(); in Collector()
|
D | build_vocab_op.cc | 48 TaskManager::FindMe()->Post(); in WorkerEntry() 91 TaskManager::FindMe()->Post(); in operator ()() 128 TaskManager::FindMe()->Post(); in CollectorThread()
|
D | cache_op.cc | 53 TaskManager::FindMe()->Post(); in operator ()() 164 TaskManager::FindMe()->Post(); in WorkerEntry()
|
D | build_sentence_piece_vocab_op.cc | 49 TaskManager::FindMe()->Post(); in operator ()() 71 TaskManager::FindMe()->Post(); in SentenceThread()
|
D | barrier_op.cc | 39 TaskManager::FindMe()->Post(); in operator ()()
|
D | shuffle_op.cc | 113 TaskManager::FindMe()->Post(); in operator ()()
|
/third_party/mindspore/mindspore/ccsrc/minddata/dataset/engine/datasetops/source/ |
D | nonmappable_leaf_op.cc | 66 TaskManager::FindMe()->Post(); in operator ()() 133 TaskManager::FindMe()->Post(); in WorkerEntry() 256 TaskManager::FindMe()->Post(); in WaitToFillIOBlockQueue()
|
D | image_folder_op.cc | 150 TaskManager::FindMe()->Post(); in PrescanWorkerEntry() 204 TaskManager::FindMe()->Post(); in StartAsyncWalk() 237 TaskManager::FindMe()->Post(); in LaunchThreadsAndInitOp()
|
D | random_data_op.cc | 164 TaskManager::FindMe()->Post(); in operator ()() 233 TaskManager::FindMe()->Post(); in WorkerEntry()
|
D | celeba_op.cc | 63 TaskManager::FindMe()->Post(); in LaunchThreadsAndInitOp() 71 TaskManager::FindMe()->Post(); in ParseAttrFile()
|
D | mappable_leaf_op.cc | 92 TaskManager::FindMe()->Post(); in WorkerEntry()
|
/third_party/mindspore/mindspore/ccsrc/minddata/dataset/engine/perf/ |
D | monitor.cc | 29 TaskManager::FindMe()->Post(); in operator ()()
|
/third_party/mindspore/mindspore/ccsrc/minddata/dataset/engine/datasetops/map_op/ |
D | map_op.cc | 126 TaskManager::FindMe()->Post(); in operator ()() 134 TaskManager::FindMe()->Post(); in operator ()() 202 TaskManager::FindMe()->Post(); in WorkerEntry()
|
/third_party/mindspore/mindspore/ccsrc/minddata/dataset/engine/cache/perf/ |
D | cache_pipeline_run.cc | 136 TaskManager::FindMe()->Post(); in ListenToParent() 262 TaskManager::FindMe()->Post(); in WriterWorkerEntry() 403 TaskManager::FindMe()->Post(); in ReaderWorkerEntry()
|