!8419 add reg for pad fp16 to save time

From: @zhaozhenlong Reviewed-by: @zhang_xue_tong,@zhanghaibo5 Signed-off-by: @zhang_xue_tong
4 years ago · c1e13eb5c3
parent 52b13752ad 4dcaae9706
commit c1e13eb5c3
1 changed files with 1 additions and 0 deletions
--- 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<lite::Tensor *> &i
  }
  return kernel;
 }
+REG_KERNEL(kCPU, kNumberTypeFloat16, PrimitiveType_Pad, CpuPadFp16KernelCreator)
 }  // namespace mindspore::kernel