Searched refs:task_offset (Results 1 – 8 of 8) sorted by relevance
| /third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/cpu/ |
| D | embedding_look_up_cpu_kernel.cc | 114 size_t task_offset = 0; in LaunchKernel() local
117 if (task_offset >= indices_lens_) { in LaunchKernel()
120 MS_LOG(DEBUG) << "task_offset: " << task_offset << " task_proc_lenss:" << task_proc_lens; in LaunchKernel()
121 auto task = [input_addr, indices_addr, output_addr, task_offset, task_proc_lens, this]() { in LaunchKernel()
122 …LookUpTableTask<T>(input_addr, indices_addr + task_offset, output_addr + task_offset * outer_dim_s… in LaunchKernel()
127 task_offset += task_proc_lens; in LaunchKernel()
128 if (task_offset + task_proc_lens > indices_lens_) { in LaunchKernel()
129 task_proc_lens = indices_lens_ - task_offset; in LaunchKernel()
|
| /third_party/mindspore/mindspore/ccsrc/ps/ps_cache/ |
| D | ps_cache_manager.cc | 472 size_t task_offset = 0; in CheckCacheHitOrOutRange() local
475 if (task_offset >= batch_ids_len) { in CheckCacheHitOrOutRange()
480 …std::thread(&PsCacheManager::CheckCacheHitOrOutRangeTask, this, batch_ids + task_offset, task_proc… in CheckCacheHitOrOutRange()
481 … hash_index + task_offset, in_device + task_offset, out_range + task_offset, hash_hit_count + i); in CheckCacheHitOrOutRange()
482 task_offset += task_proc_lens; in CheckCacheHitOrOutRange()
484 if (task_offset != batch_ids_len) { in CheckCacheHitOrOutRange()
485 …<< "Ps cache check id in device inadequate, total:" << batch_ids_len << " checked:" << task_offset; in CheckCacheHitOrOutRange()
724 size_t task_offset = 0; in LookUpHostHashTable() local
727 if (task_offset >= indices_lens) { in LookUpHostHashTable()
730 MS_LOG(DEBUG) << "Task offset: " << task_offset << ", task process lens:" << task_proc_lens; in LookUpHostHashTable()
[all …]
|
| /third_party/mindspore/mindspore/ccsrc/ps/random_normal/ |
| D | random_normal.cc | 46 size_t task_offset = 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/backend/kernel_compiler/cpu/nnacl/int8/ |
| D | transpose_int8.c | 237 size_t task_offset = offset_size * task_id; in TransposeDimsInt8() local
238 size_t count = data_size - task_offset; in TransposeDimsInt8()
239 if (data_size < task_offset) { in TransposeDimsInt8()
243 for (size_t idx = task_offset; idx < task_offset + count; ++idx) { in TransposeDimsInt8()
|
| /third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/cpu/nnacl/fp32/ |
| D | transpose_fp32.c | 187 int task_offset = offset_size * task_id; in TransposeDimsFp32() local
188 int count = data_size - task_offset; in TransposeDimsFp32()
193 for (int idx = task_offset; idx < task_offset + count; ++idx) { in TransposeDimsFp32()
|
| /third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/cpu/nnacl/fp16/ |
| D | transpose_fp16.c | 189 size_t task_offset = offset_size * task_id; in TransposeDimsFp16() local
190 int count = data_size - task_offset; in TransposeDimsFp16()
195 for (size_t idx = task_offset; idx < task_offset + count; ++idx) { in TransposeDimsFp16()
|
| /third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/cpu/nnacl/base/ |
| D | transpose_base.c | 191 … size_t task_offset = offset_size * task_id; \
192 … int count = data_size - task_offset; \
197 … for (int idx = task_offset; idx < task_offset + count; ++idx) { \
|
| /third_party/mindspore/mindspore/ccsrc/ps/ |
| D | parameter_server.cc | 182 size_t task_offset = 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()
|