Home
last modified time | relevance | path

Searched refs:task_start (Results 1 – 3 of 3) sorted by relevance

/third_party/mindspore/mindspore/lite/src/delegate/tensorrt/
Dtensorrt_utils.cc328 size_t task_start = 0; in PackNHWCToNCHWFp16() local
332 task_start = offset_hw * task_id; in PackNHWCToNCHWFp16()
333 size_t count = plane - task_start; in PackNHWCToNCHWFp16()
337 task_end = (task_id + 1) == thread_count ? plane : MSMIN(plane, task_start + offset_hw); in PackNHWCToNCHWFp16()
338 hw8 = task_start + ((task_end - task_start) >= offset_hw ? offset_hw : 0); in PackNHWCToNCHWFp16()
347 size_t hw = task_start; in PackNHWCToNCHWFp16()
/third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/cpu/nnacl/fp16/
Dpack_fp16.c194 int task_start = 0; in PackNHWCToNCHWFp16() local
198 task_start = offset_hw * task_id; in PackNHWCToNCHWFp16()
199 int count = plane - task_start; in PackNHWCToNCHWFp16()
203 task_end = (task_id + 1) == thread_count ? plane : MSMIN(plane, task_start + offset_hw); in PackNHWCToNCHWFp16()
204 hw_align = task_start + ((task_end - task_start) >= offset_hw ? offset_hw : 0); in PackNHWCToNCHWFp16()
213 int hw = task_start; in PackNHWCToNCHWFp16()
/third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/cpu/nnacl/fp32/
Dpack_fp32.c547 int task_start = 0; in PackNHWCToNCHWFp32() local
551 task_start = offset_hw * task_id; in PackNHWCToNCHWFp32()
552 int count = plane - task_start; in PackNHWCToNCHWFp32()
556 task_end = (task_id + 1) == thread_count ? plane : MSMIN(plane, task_start + offset_hw); in PackNHWCToNCHWFp32()
557 hw8 = task_start + ((task_end - task_start) >= offset_hw ? offset_hw : 0); in PackNHWCToNCHWFp32()
566 int hw = task_start; in PackNHWCToNCHWFp32()