|
|
@@ -112,7 +112,7 @@ kernel::LiteKernel *CpuConvDwFp16KernelCreator(const std::vector<lite::Tensor *> |
|
|
TypeId origin_weight_data_type, TypeId origin_bias_data_type) { |
|
|
TypeId origin_weight_data_type, TypeId origin_bias_data_type) { |
|
|
MS_ASSERT(opParameter != nullptr); |
|
|
MS_ASSERT(opParameter != nullptr); |
|
|
auto conv_param = reinterpret_cast<ConvParameter *>(opParameter); |
|
|
auto conv_param = reinterpret_cast<ConvParameter *>(opParameter); |
|
|
kernel::LiteKernel *kernel; |
|
|
|
|
|
|
|
|
kernel::LiteKernel *kernel = nullptr; |
|
|
if (conv_param->input_channel_ < 32) { |
|
|
if (conv_param->input_channel_ < 32) { |
|
|
kernel = new (std::nothrow) kernel::ConvolutionDepthwiseSWFp16CPUKernel( |
|
|
kernel = new (std::nothrow) kernel::ConvolutionDepthwiseSWFp16CPUKernel( |
|
|
opParameter, inputs, outputs, ctx, origin_weight, origin_bias, origin_weight_data_type, origin_bias_data_type); |
|
|
opParameter, inputs, outputs, ctx, origin_weight, origin_bias, origin_weight_data_type, origin_bias_data_type); |
|
|
|