p47068915
/
mindspore
forked from mindspore-Ecosystem/mindspore
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

!5128 [MS][LITE] 20200822 ml_liveness_detect_landmark model fp16 version exec fail 

Merge pull request !5128 from sunsuodong/fix_fp16_conv1x1
This commit is contained in:
mindspore-ci-bot 2020-08-25 16:51:22 +08:00 committed by Gitee
parent 2de216961f e6dea90625
commit 2ac18edafd
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
mindspore/lite/src/runtime/kernel/arm/fp16/convolution_1x1_fp16.cc
View File

@ -83,7 +83,6 @@ int Convolution1x1FP16CPUKernel::InitConv1x1Param() {
}
int Convolution1x1FP16CPUKernel::InitWeightBias() {
auto bias_tensor = in_tensors_.at(kBiasIndex);
auto weight_tensor = in_tensors_.at(kWeightIndex);
auto input_channel = weight_tensor->Channel();
auto output_channel = weight_tensor->Batch();
@ -96,6 +95,7 @@ int Convolution1x1FP16CPUKernel::InitWeightBias() {
}
memset(bias_data_, 0, size);
if (in_tensors_.size() == 3) {
auto bias_tensor = in_tensors_.at(kBiasIndex);
if (bias_tensor->data_type() == kNumberTypeFloat16) {
memcpy(bias_data_, bias_tensor->Data(), output_channel * sizeof(float16_t));
} else {