Searched refs:ori_input_size (Results 1 – 3 of 3) sorted by relevance
| /third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/cpu/nnacl/fp32/ |
| D | pack_fp32.c | 135 size_t ori_input_size = batch * plane * channel * sizeof(float); in PackNHWCToNHWC4Fp32() local
136 memcpy((float *)dst, (float *)src, ori_input_size); in PackNHWCToNHWC4Fp32()
155 size_t ori_input_size = batch * plane * channel * sizeof(float); in PackNHWCToNHWCXFp32() local
156 memcpy((float *)dst, (float *)src, ori_input_size); in PackNHWCToNHWCXFp32()
279 size_t ori_input_size = batch * plane * channel * sizeof(float); in PackNHWCToNHWC8Fp32() local
280 memcpy((float *)dst, (float *)src, ori_input_size); in PackNHWCToNHWC8Fp32()
297 size_t ori_input_size = batch * plane * channel * sizeof(float); in PackNHWCXToNHWCFp32() local
298 memcpy((float *)dst, (float *)src, ori_input_size); in PackNHWCXToNHWCFp32()
|
| /third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/cpu/nnacl/fp16/ |
| D | pack_fp16.c | 276 size_t ori_input_size = batch * plane * channel * sizeof(float16_t); in PackNHWCToNHWC4Fp16() local
277 memcpy(dst, src, ori_input_size); in PackNHWCToNHWC4Fp16()
300 size_t ori_input_size = batch * plane * channel * sizeof(float16_t); in PackNHWCToNHWC8Fp16() local
301 memcpy(dst, src, ori_input_size); in PackNHWCToNHWC8Fp16()
318 size_t ori_input_size = batch * plane * channel * sizeof(float16_t); in PackNHWC4ToNHWCFp16() local
319 memcpy((float16_t *)dst, (float16_t *)src, ori_input_size); in PackNHWC4ToNHWCFp16()
|
| /third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/cpu/nnacl/int8/ |
| D | pack_int8.c | 1021 size_t ori_input_size = batch * plane * channel; in PackNHWCToNHWC4Int8() local
1022 memcpy((int8_t *)dst, (int8_t *)src, ori_input_size); in PackNHWCToNHWC4Int8()
1040 size_t ori_input_size = batch * plane * channel; in PackNHWC4ToNHWCInt8() local
1041 memcpy((int8_t *)dst, (int8_t *)src, ori_input_size); in PackNHWC4ToNHWCInt8()
1060 size_t ori_input_size = batch * plane * channel; in PackNHWCToNHWC8Int8() local
1061 memcpy((int8_t *)dst, (int8_t *)src, ori_input_size); in PackNHWCToNHWC8Int8()
1079 size_t ori_input_size = batch * plane * channel; in PackNHWC8ToNHWCInt8() local
1080 memcpy((int8_t *)dst, (int8_t *)src, ori_input_size); in PackNHWC8ToNHWCInt8()
|