Home
last modified time | relevance | path

Searched defs:src_kernel_offset (Results 1 – 4 of 4) sorted by relevance

/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/cpu/kernel/nnacl/fp16/
Dpack_fp16.c54 int src_kernel_offset = src_oc_offset + k * channel; in PackNHWCToNC4HW4Fp16() local
79 int src_kernel_offset = batch_oc_offset + k * channel; in PackNHWCToNC8HW8NotAlignedFp16() local
104 int src_kernel_offset = src_c_offset + k; in PackNCHWToNC4HW4Fp16() local
282 int src_kernel_offset = src_c_offset + k * C4NUM; in PackNC4HW4ToNHWC4Fp16() local
301 int src_kernel_offset = src_c_offset + k * C4NUM; in PackNC4HW4ToNHWCFp16() local
320 int src_kernel_offset = src_c_offset + k * C4NUM; in PackNC4HW4ToNCHWFp16() local
341 int src_kernel_offset = src_c_offset + k; in PackNCHWFp32ToNC8HW8Fp16() local
362 int src_kernel_offset = src_c_offset + k; in PackNCHWFp16ToNC8HW8Fp16() local
426 int src_kernel_offset = src_offset + k * C8NUM; in PackNC8HW8ToNHWCFp16() local
/third_party/mindspore/mindspore-src/source/mindspore/lite/src/litert/kernel/opencl/
Dutils.h83 int src_kernel_offset = src_c_offset + k; in PackNCHWToNC4HW4() local
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/cpu/kernel/nnacl/int8/
Dpack_int8.c72 int src_kernel_offset = k * input_channel; in PackWeightToC8Int8() local
278 int src_kernel_offset = src_offset + k * C4NUM; in PackNC4HW4ToNHWCInt8() local
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/cpu/kernel/nnacl/fp32/
Dpack_fp32.c43 int src_kernel_offset = batch_oc_offset + k * channel; in PackNHWCToNC4HW4NotAlignedFp32() local
74 int src_kernel_offset = batch_oc_offset + k * channel; in PackNHWCToNC8HW8NotAlignedFp32() local
1076 int src_kernel_offset = src_oc_offset + k * channel; in PackNHWCToNC4HW4Fp32() local
1329 int src_kernel_offset = src_c_offset + k * C4NUM; in PackNC4HW4ToNHWC4Fp32() local
1366 int src_kernel_offset = src_offset + k * C4NUM; in PackNC4HW4ToNHWCFp32() local
1418 int src_kernel_offset = src_oc_offset + k * channel; in PackNHWCToNC8HW8Fp32() local
1448 int src_kernel_offset = src_offset + k * C8NUM; in PackNC8HW8ToNHWCFp32() local