Home
last modified time | relevance | path

Searched defs:thread_pool (Results 1 – 17 of 17) sorted by relevance

/third_party/mindspore/mindspore-src/source/mindspore/lite/src/litert/
Dthread_pool_reuse_manager.cc30 for (auto &thread_pool : pair.second) { in ~ThreadPoolReuseManager() local
52 auto thread_pool = thread_pool_container_[hash_key].back(); in GetThreadPool() local
66 ThreadPool *thread_pool) { in RetrieveThreadPool()
Dexecutor.cc28 auto thread_pool = ctx_->thread_pool_; in Run() local
Dparallel_lite_actor.cc132 auto thread_pool = reinterpret_cast<ActorThreadPool *>(ctx_->thread_pool_); in KernelActorInit() local
Dmindrt_executor.cc339 auto thread_pool = ctx_->thread_pool_; in Run() local
Dlite_mindrt.cc460 ActorThreadPool *thread_pool = reinterpret_cast<ActorThreadPool *>(ctx->thread_pool_); in CreateOpActor() local
Dlite_session.cc1800 ThreadPool *thread_pool = this->context_->thread_pool_; in InitGPURuntime() local
1839 ThreadPool *thread_pool = this->context_->thread_pool_; in InitGPURuntime() local
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/cpu/kernel/pyboost/
Dpyboost_cpu_kernel_extra_func.cc31 auto thread_pool = kernel::GetActorMgrInnerThreadPool(); in SetThreadPool() local
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/cpu/optimizer/
Dcpu_pass_utils.cc79 auto thread_pool = kernel::GetActorMgrInnerThreadPool(); in AddCastOpNodeToGraph() local
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/cpu/hal/profiler/
Dcpu_profiling.cc74 auto thread_pool = actor_manager->GetActorThreadPool(); in SetRuntimeStart() local
92 auto thread_pool = actor_manager->GetActorThreadPool(); in SetRuntimeEnd() local
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/cpu/kernel/
Dcpu_kernel.cc186 auto thread_pool = actor_manager->GetActorThreadPool(); in GetActorMgrInnerThreadPool() local
206 auto thread_pool = pool == nullptr ? GetActorMgrInnerThreadPool() : pool; in ParallelLaunch() local
232 auto thread_pool = pool == nullptr ? GetActorMgrInnerThreadPool() : pool; in ParallelLaunch() local
258 auto thread_pool = pool == nullptr ? GetActorMgrInnerThreadPool() : pool; in ParallelLaunchAutoSearch() local
Dstridedslice_cpu_kernel.cc120 auto thread_pool = GetActorMgrInnerThreadPool(); in InitParallelParam() local
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/cpu/kernel/mkldnn/
Dmkl_cpu_kernel.h211 auto thread_pool = pool_ == nullptr ? GetActorMgrInnerThreadPool() : pool_; in MKLCpuKernelMod() local
/third_party/mindspore/mindspore-src/source/mindspore/core/mindrt/src/thread/
Dparallel_thread_pool_manager.cc84 ParallelThreadPool *thread_pool = static_cast<ParallelThreadPool *>(pool); in GetThreadPoolSize() local
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/cpu/hal/hardware/
Dcpu_device_context.cc464 auto thread_pool = kernel::GetActorMgrInnerThreadPool(); in CreateKernel() local
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/runtime/graph_scheduler/
Dgraph_scheduler.cc667 …Scheduler::SpawnMultiPipelineActor(ActorSet *const actor_set, ActorThreadPool *const thread_pool) { in SpawnMultiPipelineActor()
752 …duler::RefreshContextAndThreadPool(ActorSet *const actor_set, ActorThreadPool *const thread_pool) { in RefreshContextAndThreadPool()
829 auto thread_pool = ActorMgr::GetActorMgrRef()->GetActorThreadPool(); in Run() local
946 auto thread_pool = ActorMgr::GetActorMgrRef()->GetActorThreadPool(); in SetActorExecutionStrategy() local
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/include/common/
Drandom.h73 auto &thread_pool = common::ThreadPool::GetInstance(); in GenerateRandomsParallel() local
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/minddata/dataset/kernels/data/
Dparse_example_op.cc936 const std::unique_ptr<Eigen::ThreadPool> &thread_pool) { in ParallelFor()