diff --git a/mindspore/lite/src/runtime/kernel/arm/fp16/convolution_fp16.cc b/mindspore/lite/src/runtime/kernel/arm/fp16/convolution_fp16.cc index e4fbb687f0..f35d27c3f3 100644 --- a/mindspore/lite/src/runtime/kernel/arm/fp16/convolution_fp16.cc +++ b/mindspore/lite/src/runtime/kernel/arm/fp16/convolution_fp16.cc @@ -296,7 +296,7 @@ lite::Tensor *CreateOutputTensor(std::vector out_shape, const std::vectorset_data_type(outputs.at(index)->data_type()); + out_tensor->set_data_type(mindspore::kNumberTypeFloat16); out_tensor->set_format(outputs.at(index)->format()); if (infered_flag) { out_tensor->set_shape(out_shape); @@ -356,7 +356,7 @@ kernel::LiteKernel *CpuGroupConvFp16KernelCreator(const std::vectordata_type(), in_shape, infered_flag); + auto in_tensor = CreateInputTensor(mindspore::kNumberTypeFloat16, in_shape, infered_flag); if (in_tensor == nullptr) { delete new_conv_parameter; FreeMemoryFp16(group_convs, new_inputs, new_outputs);