Home
last modified time | relevance | path

Searched refs:dst_c_offset (Results 1 – 4 of 4) sorted by relevance

/third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/cpu/nnacl/fp32/
Dpack_fp32.c311 int dst_c_offset = dst_offset + c4_block_num * C4NUM + c4_block_res; in PackNC4HW4ToNHWC4Fp32() local
314 int dst_kernel_offset = dst_c_offset + k * c4 * C4NUM; in PackNC4HW4ToNHWC4Fp32()
331 int dst_c_offset = dst_kernel_offset + c * C4NUM; in PackNC4HW4ToNHWCFp32() local
333 MS_STQ_F32((float *)dst + dst_c_offset, MS_LDQ_F32((float *)src + src_c_offset)); in PackNC4HW4ToNHWCFp32()
335 ((float *)dst + dst_c_offset)[0] = ((float *)src + src_c_offset)[0]; in PackNC4HW4ToNHWCFp32()
336 ((float *)dst + dst_c_offset)[1] = ((float *)src + src_c_offset)[1]; in PackNC4HW4ToNHWCFp32()
337 ((float *)dst + dst_c_offset)[2] = ((float *)src + src_c_offset)[2]; in PackNC4HW4ToNHWCFp32()
338 ((float *)dst + dst_c_offset)[3] = ((float *)src + src_c_offset)[3]; in PackNC4HW4ToNHWCFp32()
362 int dst_c_offset = dst_kernel_offset + c * C8NUM; in PackNC8HW8ToNHWCFp32() local
364 MS_ST256_F32((float *)dst + dst_c_offset, MS_LD256_F32((float *)src + src_c_offset)); in PackNC8HW8ToNHWCFp32()
[all …]
/third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/cpu/nnacl/fp16/
Dpack_fp16.c174 int dst_c_offset = dst_offset + c4_block_num * plane * C4NUM; in PackNCHWToNC4HW4Fp16() local
177 int dst_kernel_offset = dst_c_offset + C4NUM * k + c4_block_rem; in PackNCHWToNC4HW4Fp16()
332 int dst_c_offset = nhwc4_batch_offset + c; in PackNCHWToNHWC4Fp16() local
335 int dst_plane_offset = dst_c_offset + i * ic4 * C4NUM; in PackNCHWToNHWC4Fp16()
352 int dst_c_offset = dst_offset + c4_block_num * C4NUM + c4_block_res; in PackNC4HW4ToNHWC4Fp16() local
355 int dst_kernel_offset = dst_c_offset + k * c4 * C4NUM; in PackNC4HW4ToNHWC4Fp16()
371 int dst_c_offset = dst_offset + c; in PackNC4HW4ToNHWCFp16() local
374 int dst_kernel_offset = dst_c_offset + k * channel; in PackNC4HW4ToNHWCFp16()
390 int dst_c_offset = dst_offset + c * plane; in PackNC4HW4ToNCHWFp16() local
393 int dst_kernel_offset = dst_c_offset + k; in PackNC4HW4ToNCHWFp16()
[all …]
/third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/cpu/nnacl/int8/
Dpack_int8.c829 int dst_c_offset = dst_plane_offset + i * in_plane; in PackInputToC8Int8() local
831 vst1q_s16(packed_input + dst_c_offset, vmovl_s8(vld1_s8(input_data + src_c_offset))); in PackInputToC8Int8()
834 (packed_input + dst_c_offset)[j] = (int16_t)(input_data + src_c_offset)[j]; in PackInputToC8Int8()
842 int dst_c_offset = dst_plane_offset + tmp_ic_offset + l; in PackInputToC8Int8() local
843 (packed_input + dst_c_offset)[0] = (int16_t)(input_data + src_c_offset)[0]; in PackInputToC8Int8()
847 int dst_c_offset = dst_plane_offset + tmp_ic_offset + res_c + l; in PackInputToC8Int8() local
848 (packed_input + dst_c_offset)[0] = 0; in PackInputToC8Int8()
1093 int dst_c_offset = dst_offset + c8_block_num * plane * C8NUM; in PackNCHWToNC8HW8Int8() local
1096 int dst_kernel_offset = dst_c_offset + C8NUM * k + c8_block_rem; in PackNCHWToNC8HW8Int8()
1113 int dst_c_offset = dst_kernel_offset + c * C4NUM; in PackNC4HW4ToNHWCInt8() local
[all …]
/third_party/mindspore/mindspore/lite/src/runtime/kernel/opencl/
Dutils.h78 int dst_c_offset = dst_offset + c4_block_num * plane_out; in PackNCHWToNC4HW4() local
81 int dst_kernel_offset = dst_c_offset + C4NUM * k + c4_block_rem; in PackNCHWToNC4HW4()