|
|
|
@@ -242,7 +242,8 @@ kernel::LiteKernel *CpuConvFp16KernelCreator(const std::vector<lite::tensor::Ten |
|
|
|
|
|
|
|
kernel::LiteKernel *kernel = nullptr; |
|
|
|
if (kernel_h == 1 && kernel_w == 1) { |
|
|
|
kernel = new (std::nothrow) kernel::Convolution1x1FP16CPUKernel(opParameter, inputs, outputs, ctx, primitive); |
|
|
|
// kernel = new (std::nothrow) kernel::Convolution1x1FP16CPUKernel(opParameter, inputs, outputs, ctx, primitive); |
|
|
|
kernel = new (std::nothrow) kernel::ConvolutionFP16CPUKernel(opParameter, inputs, outputs, ctx, primitive); |
|
|
|
} else { |
|
|
|
bool use_winograd = false; |
|
|
|
int out_unit; |
|
|
|
@@ -273,5 +274,5 @@ kernel::LiteKernel *CpuConvFp16KernelCreator(const std::vector<lite::tensor::Ten |
|
|
|
} |
|
|
|
return kernel; |
|
|
|
} |
|
|
|
// REG_KERNEL(kCPU, kNumberTypeFloat16, PrimitiveType_Conv2D, CpuConvFp16KernelCreator) |
|
|
|
REG_KERNEL(kCPU, kNumberTypeFloat16, PrimitiveType_Conv2D, CpuConvFp16KernelCreator) |
|
|
|
} // namespace mindspore::kernel |