Searched refs:task_len (Results 1 – 2 of 2) sorted by relevance
| /third_party/mindspore/mindspore/ccsrc/ps/random_normal/ |
| D | random_normal.cc | 34 auto update_stddev_task = [](float stddev, size_t task_len, float *data) { in UpdateStandardDeviation() argument
35 for (size_t i = 0; i < task_len; i++) { in UpdateStandardDeviation()
49 size_t task_len = total_count / thread_num + (i < (total_count % thread_num) ? 1 : 0); in UpdateStandardDeviation() local
50 threads[i] = std::thread(update_stddev_task, stddev, task_len, output + task_offset); in UpdateStandardDeviation()
51 task_offset += task_len; in UpdateStandardDeviation()
|
| /third_party/mindspore/mindspore/ccsrc/ps/ |
| D | parameter_server.cc | 170 auto init_task = [](float value, size_t task_len, float *data) { in InitAccumParallel() argument
171 for (size_t i = 0; i < task_len; i++) { in InitAccumParallel()
186 size_t task_len = total_len / thread_num + (i < (total_len % thread_num) ? 1 : 0); in InitAccumParallel() local
187 threads[i] = std::thread(init_task, init_value, task_len, embedding_data + task_offset); in InitAccumParallel()
188 task_offset += task_len; in InitAccumParallel()
|