Searched refs:cx_num (Results 1 – 2 of 2) sorted by relevance
| /third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/cpu/nnacl/fp32/ |
| D | pack_fp32.h | 37 void PackNHWCXToNHWCFp32(const void *src, void *dst, int batch, int plane, int channel, int cx_num);
|
| D | pack_fp32.c | 284 void PackNHWCXToNHWCFp32(const void *src, void *dst, int batch, int plane, int channel, int cx_num)… in PackNHWCXToNHWCFp32() argument
285 int c_algin = UP_DIV(channel, cx_num); in PackNHWCXToNHWCFp32()
286 int ic_remainder_ = channel % cx_num; in PackNHWCXToNHWCFp32()
290 int batch_offset = b * c_algin * cx_num * plane; in PackNHWCXToNHWCFp32()
293 (float *)src + batch_offset + i * c_algin * cx_num, channel * sizeof(float)); in PackNHWCXToNHWCFp32()
|