Home
last modified time | relevance | path

Searched refs:kMaxThreadNum (Results 1 – 8 of 8) sorted by relevance

/third_party/mindspore/mindspore/ccsrc/common/
Dthread_pool.cc29 const size_t kMaxThreadNum = 23; variable
44 if (max_thread_num_ > kMaxThreadNum) { in ThreadPool()
45 max_thread_num_ = kMaxThreadNum; in ThreadPool()
/third_party/mindspore/mindspore/ccsrc/ps/random_normal/
Drandom_normal.cc26 static const uint32_t kMaxThreadNum = 16; variable
40 uint32_t thread_num = std::max(kMaxThreadNum, kCPUCoreNum); in UpdateStandardDeviation()
71 uint32_t thread_num = std::max(kMaxThreadNum, kCPUCoreNum); in InitRandomNormal()
/third_party/mindspore/mindspore/ccsrc/runtime/framework/actor/
Dactor_common.cc28 const size_t kMaxThreadNum = 23; in ComputeThreadNums() local
40 if (*max_thread_num > kMaxThreadNum) { in ComputeThreadNums()
41 *max_thread_num = kMaxThreadNum; in ComputeThreadNums()
/third_party/mindspore/mindspore/lite/src/runtime/kernel/arm/fp32/
Dl2_norm_fp32.cc31 const int kMaxThreadNum = 8; variable
50 CHECK_LESS_RETURN(MAX_MALLOC_SIZE, kMaxThreadNum * sizeof(float)); in MallocTmpBuffer()
51 tmp_sum_ = reinterpret_cast<float *>(malloc(kMaxThreadNum * sizeof(float))); in MallocTmpBuffer()
/third_party/mindspore/mindspore/ccsrc/ps/ps_cache/
Dps_cache_manager.cc468 thread_num = thread_num > kMaxThreadNum ? kMaxThreadNum : thread_num; in CheckCacheHitOrOutRange()
469 std::thread threads[kMaxThreadNum]; in CheckCacheHitOrOutRange()
470 size_t hash_hit_count[kMaxThreadNum] = {0}; in CheckCacheHitOrOutRange()
720 thread_num = thread_num > kMaxThreadNum ? kMaxThreadNum : thread_num; in LookUpHostHashTable()
721 std::thread threads[kMaxThreadNum]; in LookUpHostHashTable()
751 thread_num = thread_num > kMaxThreadNum ? kMaxThreadNum : thread_num; in InsertHostHashTable()
752 std::thread threads[kMaxThreadNum]; in InsertHostHashTable()
Dps_cache_manager.h42 constexpr size_t kMaxThreadNum = 16; variable
/third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/cpu/
Dcpu_kernel.cc153 const size_t kMaxThreadNum = 23; in GetActorMgrInnerThreadPool() local
162 max_thread_num = max_thread_num < kMaxThreadNum ? max_thread_num : kMaxThreadNum; in GetActorMgrInnerThreadPool()
/third_party/mindspore/mindspore/ccsrc/ps/
Dparameter_server.cc23 static const uint32_t kMaxThreadNum = 16; variable
176 size_t thread_num = std::max(kMaxThreadNum, kCPUCoreNum); in InitAccumParallel()