| /third_party/mindspore/mindspore-src/source/mindspore/lite/test/ut/src/runtime/kernel/arm/int8/ |
| D | resize_nearest_neighbor_int8_tests.cc | 52 const int thread_num) { in Prepare()
97 int thread_num = 1; in TEST_F() local
118 int thread_num = 1; in TEST_F() local
144 int thread_num = 2; in TEST_F() local
171 int thread_num = 2; in TEST_F() local
197 int thread_num = 2; in TEST_F() local
|
| D | resize_bilinear_int8_tests.cc | 57 const int thread_num) { in Prepare()
95 int thread_num = 1; in TEST_F() local
113 int thread_num = 1; in TEST_F() local
139 int thread_num = 2; in TEST_F() local
165 int thread_num = 2; in TEST_F() local
|
| /third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/gpu/kernel/cuda_impl/cuda_ops/ |
| D | instance_norm_impl.cu | 20 __global__ void CopyMemKernel(const size_t thread_num, const size_t N, const size_t C, float *gamma… in CopyMemKernel()
54 size_t thread_num = N * C * 4; in CopyMemDevice2Device() local
61 __global__ void ComputeMeanKernel(const size_t thread_num, const size_t N, const size_t C, float *d… in ComputeMeanKernel()
83 size_t thread_num = C * 2; in ComputeMean() local
|
| D | prelu_grad_impl.cu | 42 __global__ void ComputeDwData(size_t weight_size, size_t thread_num, const float *dw_array, T *dw) { in ComputeDwData()
55 size_t thread_num = static_cast<size_t>(GET_BLOCKS(size) * GET_THREADS); in CalPReLUGrad() local
|
| D | assign_add_impl.cu | 32 int thread_num = size > 512 ? 512 : size; in CalAssignAdd() local
|
| D | eps_impl.cu | 34 int thread_num = size > 512 ? 512 : size; in CalEps() local
|
| D | bias_add_nhwc.cu | 34 size_t thread_num = num_value > 256 ? 256 : num_value; in CalBiasAddNHWC() local
|
| /third_party/mindspore/mindspore-src/source/mindspore/lite/test/ut/src/runtime/kernel/arm/fp32/ |
| D | l2norm_fp32_test.cc | 50 … float *output_data, const int axis_num, ActType activation_type, const int thread_num) { in Init()
91 int thread_num = 2; in TEST_F() local
112 int thread_num = 2; in TEST_F() local
133 int thread_num = 4; in TEST_F() local
154 int thread_num = 1; in TEST_F() local
|
| D | scale_fp32_tests.cc | 69 ActivationType act_type, const int thread_num) { in Prepare()
113 int thread_num = 2; in TEST_F() local
135 int thread_num = 2; in TEST_F() local
156 int thread_num = 2; in TEST_F() local
|
| /third_party/mindspore/mindspore-src/source/mindspore/lite/src/litert/ |
| D | thread_pool_reuse_manager.cc | 40 …dPoolReuseManager::GetThreadPool(size_t actor_num, size_t inter_op_parallel_num, size_t thread_num, in GetThreadPool()
64 …ReuseManager::RetrieveThreadPool(size_t actor_num, size_t inter_op_parallel_num, size_t thread_num, in RetrieveThreadPool()
79 …eadPoolReuseManager::ComputeHash(size_t actor_num, size_t inter_op_parallel_num, size_t thread_num, in ComputeHash()
|
| D | thread_cost_model.cc | 97 …ostModel::GetOptimalThreadNum(const ThreadCostContext *thread_cost_context, const int thread_num) { in GetOptimalThreadNum()
144 int thread_num) { in UpdateThreadNum()
|
| /third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/cpu/kernel/nnacl/fp16/ |
| D | reduce_fp16.c | 24 int tid, int thread_num) { in ReduceMeanFp16()
49 int tid, int thread_num) { in ReduceMaxFp16()
71 int tid, int thread_num) { in ReduceMinFp16()
96 int tid, int thread_num) { in ReduceProdFp16()
121 int tid, int thread_num) { in ReduceSumFp16()
161 int tid, int thread_num) { in ReduceL2NormFp16()
|
| D | where_fp16.c | 20 int task_id, int thread_num) { in WhereWithTripleInputsFp16()
|
| D | one_hot_fp16.c | 20 const OneHotStruct *one_hot_param, const int tid, const int thread_num) { in OneHotToFp16()
|
| /third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/cpu/kernel/nnacl/int8/ |
| D | reduce_int8.c | 177 … int32_t *dst_data, const ReduceQuantArg *quant, const int tid, const int thread_num) { in ReduceMeanInt8()
211 … int8_t *dst_data, const ReduceQuantArg *quant, const int tid, const int thread_num) { in ReduceMeanLastAxis()
252 … int32_t *dst_data, const ReduceQuantArg *quant, const int tid, const int thread_num) { in ReduceSumInt8()
284 … int8_t *dst_data, const ReduceQuantArg *quant, const int tid, const int thread_num) { in ReduceSumLastAxis()
324 … int8_t *dst_data, const ReduceQuantArg *quant, const int tid, const int thread_num) { in ReduceMaxLastAxis()
360 … int32_t *dst_data, const ReduceQuantArg *quant, const int tid, const int thread_num) { in ReduceMaxInt8()
383 … int8_t *dst_data, const ReduceQuantArg *quant, const int tid, const int thread_num) { in ReduceMinLastAxis()
421 … int32_t *dst_data, const ReduceQuantArg *quant, const int tid, const int thread_num) { in ReduceMinInt8()
443 … int8_t *dst_data, const ReduceQuantArg *quant, const int tid, const int thread_num) { in ReduceProdLastAxis()
486 … int32_t *dst_data, const ReduceQuantArg *quant, const int tid, const int thread_num) { in ReduceProdInt8()
[all …]
|
| /third_party/mindspore/mindspore-src/source/mindspore/lite/tools/converter/micro/coder/wrapper/thread/ |
| D | micro_core_affinity.c | 153 int InitBindCoreId(size_t thread_num, enum BindMode bind_mode, int *bind_id) { in InitBindCoreId()
207 int thread_num = g_pool->max_thread_num; in FreeScheduleThreads() local
228 int thread_num = g_pool->max_thread_num; in BindThreadsToCore() local
|
| /third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/cpu/kernel/nnacl/fp32/ |
| D | prior_box_fp32.h | 28 const int thread_num) { in PriorBox()
|
| D | where_fp32.c | 21 int thread_num) { in WhereWithTripleInputs()
|
| D | one_hot_fp32.c | 21 const OneHotStruct *one_hot_param, const int tid, const int thread_num) { in OneHotToFp32()
|
| /third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/cpu/kernel/ |
| D | stridedslice_cpu_kernel.cc | 122 int thread_num = 1; in InitParallelParam() local
209 …edSliceCpuKernelMod::ParallelRun(const uint8_t *input_addr, uint8_t *output_addr, int thread_num) { in ParallelRun()
259 int thread_num = slice_struct_.base_.thread_nr_; in LaunchKernel() local
|
| /third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/cpu/kernel/nnacl/fp32/online_fusion/ |
| D | reduce_concat_fp32.c | 24 … int64_t batch_tile_size, int64_t inner_tile, int64_t thread_num, in Fp32ReduceSumConcatAxisSizeAVX512Fusion()
86 int64_t thread_num, int64_t task_id) { in Fp32ReduceSumConcatFusion()
|
| /third_party/mindspore/mindspore-src/source/mindspore/lite/tools/converter/micro/coder/opcoders/nnacl/fp32/ |
| D | pooling_fp32_coder.cc | 32 int thread_num = pooling_parameter->op_parameter_.thread_num_; in DoCode() local
|
| /third_party/ltp/testcases/open_posix_testsuite/conformance/interfaces/pthread_barrier_wait/ |
| D | 2-1.c | 45 int thread_num = *(int *)arg; in fn_chld() local
|
| /third_party/mindspore/mindspore-src/source/mindspore/ccsrc/ps/core/communicator/ |
| D | task_executor.cc | 22 TaskExecutor::TaskExecutor(size_t thread_num, size_t max_task_num, size_t submit_timeout) in TaskExecutor()
|
| /third_party/mindspore/mindspore-src/source/mindspore/lite/tools/converter/micro/coder/opcoders/ |
| D | op_coder.cc | 50 void OperatorCoder::set_thread_num(int thread_num) { in set_thread_num()
|