Home
last modified time | relevance | path

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

/third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/cpu/nnacl/fp16/
Dpack_fp16.c113 int dst_ic_offset = dst_oc_offset + c8_block_num * kernel_plane * C8NUM + c8_block_rem; in PackWeightToC8Fp16() local
114 (packed_weight_data + dst_ic_offset)[0] = (origin_weight_data + src_ic_offset)[0]; in PackWeightToC8Fp16()
139 int dst_ic_offset = dst_oc_offset + c4_block_num * kernel_plane * C4NUM + c4_block_rem; in PackWeightToC4Fp16() local
140 (packed_weight_data + dst_ic_offset)[0] = (origin_weight_data + src_ic_offset)[0]; in PackWeightToC4Fp16()
158 int dst_ic_offset = dst_kernel_offset + c4_block_num * plane * C4NUM + c4_block_rem; in PackNHWCToNC4HW4Fp16() local
159 ((float16_t *)dst + dst_ic_offset)[0] = ((float16_t *)src + src_ic_offset)[0]; in PackNHWCToNC4HW4Fp16()
/third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/cpu/nnacl/int8/
Dpack_int8.c879 int dst_ic_offset = dst_oc_offset + i * kernel_plane; in PackWeightToC8Int8() local
892 vst1_s16(packed_weight_data + dst_ic_offset, dst1_s16); in PackWeightToC8Int8()
893 vst1_s16(packed_weight_data + dst_ic_offset + 4, dst2_s16); in PackWeightToC8Int8()
896 …(packed_weight_data + dst_ic_offset + ci)[0] = (int16_t)((origin_weight_data + src_ic_offset + ci)… in PackWeightToC8Int8()
904 int dst_ic_offset = dst_oc_offset + c8_block_rem; in PackWeightToC8Int8() local
905 … (packed_weight_data + dst_ic_offset)[0] = (int16_t)((origin_weight_data + src_ic_offset)[0] - zp); in PackWeightToC8Int8()
/third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/cpu/nnacl/fp32/
Dpack_fp32.c87 int dst_ic_offset = dst_kernel_offset + j * plane * C4NUM; in PackNHWCToNC4HW4Fp32() local
89 vst1q_f32((float *)dst + dst_ic_offset, vld1q_f32((float *)src + src_ic_offset)); in PackNHWCToNC4HW4Fp32()
92 ((float *)dst + dst_ic_offset)[i] = ((float *)src + src_ic_offset)[i]; in PackNHWCToNC4HW4Fp32()
104 int dst_ic_offset = dst_kernel_offset + tmp_c_offset + l; in PackNHWCToNC4HW4Fp32() local
105 ((float *)dst + dst_ic_offset)[0] = ((float *)src + src_ic_offset)[0]; in PackNHWCToNC4HW4Fp32()