Browse Source

add reg for fp16 pad to save time for ml_object_detect

tags/v1.1.0
zhaozhenlong 5 years ago
parent
commit
4dcaae9706
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