Home
last modified time | relevance | path

Searched defs:thread_num (Results 1 – 25 of 91) sorted by relevance

1234

/third_party/mindspore/mindspore/lite/test/ut/src/runtime/kernel/arm/int8/
Dresize_nearest_neighbor_int8_tests.cc53 const int thread_num) { in Prepare()
98 int thread_num = 1; in TEST_F() local
119 int thread_num = 1; in TEST_F() local
145 int thread_num = 2; in TEST_F() local
172 int thread_num = 2; in TEST_F() local
198 int thread_num = 2; in TEST_F() local
Dresize_bilinear_int8_tests.cc58 const int thread_num) { in Prepare()
96 int thread_num = 1; in TEST_F() local
114 int thread_num = 1; in TEST_F() local
140 int thread_num = 2; in TEST_F() local
166 int thread_num = 2; in TEST_F() local
/third_party/mindspore/mindspore/lite/test/ut/src/runtime/kernel/arm/fp32/
Dpad_fp32_test.cc62 const int thread_num) { in Prepare()
106 int thread_num = 2; in TEST_F() local
148 int thread_num = 2; in TEST_F() local
184 int thread_num = 2; in TEST_F() local
220 int thread_num = 2; in TEST_F() local
256 int thread_num = 2; in TEST_F() local
Dl2norm_fp32_test.cc50 … 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
Dscale_fp32_tests.cc67 ActivationType act_type, const int thread_num) { in Prepare()
108 int thread_num = 2; in TEST_F() local
130 int thread_num = 2; in TEST_F() local
151 int thread_num = 2; in TEST_F() local
/third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/gpu/cuda_impl/
Dinstance_norm_impl.cu20 __global__ void CopyMemKernel(const size_t thread_num, const size_t N, const size_t C, in CopyMemKernel()
56 size_t thread_num = N * C * 4; in CopyMemDevice2Device() local
62 __global__ void ComputeMeanKernel(const size_t thread_num, const size_t N, const size_t C, in ComputeMeanKernel()
85 size_t thread_num = C * 2; in ComputeMean() local
Dprelu_grad_impl.cu43 __global__ void ComputeDwData(size_t weight_size, size_t thread_num, const float *dw_array, T *dw) { in ComputeDwData()
56 size_t thread_num = static_cast<size_t>(GET_BLOCKS(size) * GET_THREADS); in CalPReLUGrad() local
/third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/cpu/nnacl/fp32/
Dreduce_fp32.c27 int thread_num) { in ReduceMean()
52 int thread_num) { in IntReduceMean()
101 int thread_num) { in ReduceSum()
142 int thread_num) { in IntReduceSum()
183 int thread_num) { in ReduceMax()
208 int thread_num) { in IntReduceMax()
233 int thread_num) { in ReduceMin()
258 int thread_num) { in IntReduceMin()
283 int thread_num) { in ReduceAll()
308 int thread_num) { in ReduceProd()
[all …]
Dprior_box_fp32.h28 const int thread_num) { in PriorBox()
Done_hot_fp32.c21 const OneHotParameter *one_hot_param, const int tid, const int thread_num) { in OneHotToFp32()
/third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/cpu/nnacl/fp16/
Dreduce_fp16.c22 int tid, int thread_num) { in ReduceMeanFp16()
47 int tid, int thread_num) { in ReduceMaxFp16()
69 int tid, int thread_num) { in ReduceSumFp16()
Done_hot_fp16.c20 const OneHotParameter *one_hot_param, const int tid, const int thread_num) { in OneHotToFp16()
/third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/cpu/nnacl/int8/
Dreduce_int8.c177 … 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/ccsrc/backend/kernel_compiler/cpu/
Dstridedslice_cpu_kernel.cc99 int thread_num = 1; in InitParallelParam() local
194 …ridedSliceCPUKernel::ParallelRun(const uint8_t *input_addr, uint8_t *output_addr, int thread_num) { in ParallelRun()
225 int thread_num = slice_param_.op_parameter_.thread_num_; in Launch() local
Dunique_cpu_kernel.h103 size_t thread_num = params->thread_num_; in SplitAndCalculateBucketSize() local
185 auto thread_num = segments.size(); in GatherSegmentsToBuckets() local
294 size_t thread_num = buckets.size(); in UniqueEachBucket() local
334 size_t thread_num = buckets.size(); in MergeBuckets() local
/third_party/mindspore/mindspore/lite/test/ut/src/runtime/kernel/arm/fp16/
Dreduce_fp16_tests.cc52 const int thread_num) { in Prepare()
98 int thread_num = 1; in TEST_F() local
/third_party/mindspore/mindspore/ccsrc/ps/random_normal/
Drandom_normal.cc40 uint32_t thread_num = std::max(kMaxThreadNum, kCPUCoreNum); in UpdateStandardDeviation() local
71 uint32_t thread_num = std::max(kMaxThreadNum, kCPUCoreNum); in InitRandomNormal() local
/third_party/mindspore/mindspore/lite/micro/coder/opcoders/base/
Ddtype_cast_coder.cc31 int thread_num = MSMIN(thread_num_, static_cast<int>(data_num_)); in Prepare() local
/third_party/mindspore/mindspore/lite/micro/coder/wrapper/thread/
Dthread_wrapper.cc26 int CreateThreadPool(int thread_num) { in CreateThreadPool()
/third_party/ltp/testcases/open_posix_testsuite/conformance/interfaces/pthread_barrier_wait/
D2-1.c45 int thread_num = *(int *)arg; in fn_chld() local
/third_party/mindspore/mindspore/ccsrc/ps/core/communicator/
Dtask_executor.cc22 TaskExecutor::TaskExecutor(size_t thread_num, size_t max_task_num, size_t submit_timeout) in TaskExecutor()
/third_party/mindspore/mindspore/lite/java/native/runtime/
Dms_config.cpp22 …JNIEnv *env, jobject thiz, jint device_type, jint thread_num, jint cpu_bind_mode, jboolean enable_… in Java_com_mindspore_lite_config_MSConfig_createMSConfig()
/third_party/mindspore/mindspore/lite/micro/coder/opcoders/
Dop_coder.cc50 void OperatorCoder::set_thread_num(int thread_num) { in set_thread_num()
/third_party/mindspore/mindspore/lite/src/cxx_api/
Dcontext.h34 int32_t thread_num = 2; member
/third_party/grpc/test/cpp/qps/
Dinterarrival.h92 int64_t next(int thread_num) { in next()

1234