Searched refs:dst_per_plane (Results 1 – 3 of 3) sorted by relevance
| /third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/cpu/nnacl/fp32/ |
| D | pack_fp32.c | 129 float *dst_per_plane = (float *)dst + dst_batch_offset + i * oc_block_channel; in PackNHWCToNHWC4Fp32() local
130 memcpy(dst_per_plane, (float *)src + batch_offset + i * channel, channel * sizeof(float)); in PackNHWCToNHWC4Fp32()
131 memset(dst_per_plane + channel, 0, (oc_block_channel - channel) * sizeof(float)); in PackNHWCToNHWC4Fp32()
149 float *dst_per_plane = (float *)dst + dst_batch_offset + i * oc_block_channel; in PackNHWCToNHWCXFp32() local
150 memcpy(dst_per_plane, (float *)src + batch_offset + i * channel, channel * sizeof(float)); in PackNHWCToNHWCXFp32()
151 memset(dst_per_plane + channel, 0, (oc_block_channel - channel) * sizeof(float)); in PackNHWCToNHWCXFp32()
270 float *dst_per_plane = (float *)dst + nhwc8_batch_offset + i * c8_channel; in PackNHWCToNHWC8Fp32() local
271 memcpy(dst_per_plane, (float *)src + batch_offset + i * channel, channel * sizeof(float)); in PackNHWCToNHWC8Fp32()
273 dst_per_plane[j] = 0; in PackNHWCToNHWC8Fp32()
|
| /third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/cpu/nnacl/fp16/ |
| D | pack_fp16.c | 267 float16_t *dst_per_plane = (float16_t *)dst + nhwc4_batch_offset + i * c4_channel; in PackNHWCToNHWC4Fp16() local
268 … memcpy(dst_per_plane, (float16_t *)src + batch_offset + i * channel, channel * sizeof(float16_t)); in PackNHWCToNHWC4Fp16()
270 dst_per_plane[j] = 0; in PackNHWCToNHWC4Fp16()
291 float16_t *dst_per_plane = (float16_t *)dst + nhwc8_batch_offset + i * c8_channel; in PackNHWCToNHWC8Fp16() local
292 … memcpy(dst_per_plane, (float16_t *)src + batch_offset + i * channel, channel * sizeof(float16_t)); in PackNHWCToNHWC8Fp16()
294 dst_per_plane[j] = 0; in PackNHWCToNHWC8Fp16()
|
| /third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/cpu/nnacl/int8/ |
| D | pack_int8.c | 1012 int8_t *dst_per_plane = (int8_t *)dst + nhwc4_batch_offset + i * c4_channel; in PackNHWCToNHWC4Int8() local
1013 memcpy(dst_per_plane, (int8_t *)src + batch_offset + i * channel, channel); in PackNHWCToNHWC4Int8()
1015 dst_per_plane[j] = 0; in PackNHWCToNHWC4Int8()
|