From 4dcaae97063af3bcbde9569c2b9f6cb63c64843e Mon Sep 17 00:00:00 2001 From: zhaozhenlong Date: Tue, 10 Nov 2020 18:20:38 +0800 Subject: [PATCH] add reg for fp16 pad to save time for ml_object_detect --- mindspore/lite/src/runtime/kernel/arm/fp16/pad_fp16.cc | 1 + 1 file changed, 1 insertion(+) diff --git a/mindspore/lite/src/runtime/kernel/arm/fp16/pad_fp16.cc b/mindspore/lite/src/runtime/kernel/arm/fp16/pad_fp16.cc index 808935b449..60fa6e83a7 100644 --- a/mindspore/lite/src/runtime/kernel/arm/fp16/pad_fp16.cc +++ b/mindspore/lite/src/runtime/kernel/arm/fp16/pad_fp16.cc @@ -110,4 +110,5 @@ kernel::LiteKernel *CpuPadFp16KernelCreator(const std::vector &i } return kernel; } +REG_KERNEL(kCPU, kNumberTypeFloat16, PrimitiveType_Pad, CpuPadFp16KernelCreator) } // namespace mindspore::kernel