Searched refs:PackNCHWToNHWCFp16 (Results 1 – 4 of 4) sorted by relevance
| /third_party/mindspore/mindspore/lite/src/runtime/kernel/arm/fp16/ |
| D | transpose_fp16.cc | 32 …ransposeFp16CPUKernel::GetNchwToNhwcFunc(TypeId dtype) { NHNCTransposeFunc_ = PackNCHWToNHWCFp16; } in GetNchwToNhwcFunc()
|
| D | convolution_depthwise_fp16.cc | 30 …PackNCHWToNHWCFp16(reinterpret_cast<float16_t *>(origin_weight), reinterpret_cast<float16_t *>(pac… in PackWeight()
|
| /third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/cpu/nnacl/fp16/ |
| D | pack_fp16.h | 42 void PackNCHWToNHWCFp16(const void *src, void *dst, int batch, int plane, int channel, int task_id,…
|
| D | pack_fp16.c | 253 void PackNCHWToNHWCFp16(const void *src, void *dst, int batch, int plane, int channel, int task_id,… in PackNCHWToNHWCFp16() function
|