Home
last modified time | relevance | path

Searched defs:SetThreadPool (Results 1 – 4 of 4) sorted by relevance

/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/kernel/pyboost/
Dpyboost_kernel_extra_func.h32 virtual void SetThreadPool(const kernel::KernelModPtr &kernel) {} in SetThreadPool() function
Dpyboost_utils.h266 void SetThreadPool(const std::string &device_name, const kernel::KernelModPtr &kernel) { in SetThreadPool() function
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/cpu/kernel/pyboost/
Dpyboost_cpu_kernel_extra_func.cc26 void PyboostCPUKernelExtraFunc::SetThreadPool(const kernel::KernelModPtr &kernel) { in SetThreadPool() function in mindspore::kernel::pyboost::PyboostCPUKernelExtraFunc
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/cpu/kernel/
Dcpu_kernel.h155 void SetThreadPool(ThreadPool *pool) { pool_ = pool; } in SetThreadPool() function