提交 4b2e2b14 编写于 作者: M mindspore-ci-bot 提交者: Gitee

!4611 [MS][LITE][Develop]fix bias data assignment of fp16 1x1 convolution

Merge pull request !4611 from lixian/master
......@@ -81,7 +81,8 @@ int Convolution1x1FP16CPUKernel::InitWeightBias() {
return RET_ERROR;
}
memset(bias_data_, 0, matmul_param_->col_8_ * sizeof(float16_t));
memcpy(bias_data_, in_tensors_[2]->Data(), conv_param_->output_channel_ * sizeof(float16_t));
Float32ToFloat16(reinterpret_cast<float *>(in_tensors_[2]->Data()), reinterpret_cast<float16_t *>(bias_data_),
conv_param_->output_channel_);
} else {
bias_data_ = nullptr;
}
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册