Browse Source

!8419 add reg for pad fp16 to save time

From: @zhaozhenlong
Reviewed-by: @zhang_xue_tong,@zhanghaibo5
Signed-off-by: @zhang_xue_tong
tags/v1.1.0
mindspore-ci-bot Gitee 5 years ago
parent
commit
c1e13eb5c3
1 changed files with 1 additions and 0 deletions
  1. +1
    -0
      mindspore/lite/src/runtime/kernel/arm/fp16/pad_fp16.cc

+ 1
- 0
mindspore/lite/src/runtime/kernel/arm/fp16/pad_fp16.cc View File

@@ -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

Loading…
Cancel
Save