!6955 [MSLITE][Develop] fix bug of arm fp16 cpu op: reduce, arithmetic

Merge pull request !6955 from yangruoqi713/lite
pull/6955/MERGE
mindspore-ci-bot 4 years ago committed by Gitee
commit 55dadf7a67

@ -30,11 +30,11 @@ int ReduceMeanFp16(const int outer_size, const int inner_size, const int axis_si
for (k = 0; k < inner_size; k++) {
const float16_t *inner_src = outer_src + k;
float16_t *inner_dst = outer_dst + k;
float16_t tmp = 0.0;
float tmp = 0.0;
for (i = 0; i < axis_size; i++) {
tmp += inner_src[i * inner_size];
}
*inner_dst = tmp / (float16_t)axis_size;
*inner_dst = (float16_t)(tmp / axis_size);
}
}
return NNACL_OK;

@ -36,7 +36,7 @@ int ArithmeticCPUKernel::Init() {
if (!InferShapeDone()) {
return RET_OK;
}
if (in_tensors_[0]->data_type() == kNumberTypeFloat32) {
if (in_tensors_[0]->data_type() == kNumberTypeFloat32 || in_tensors_[0]->data_type() == kNumberTypeFloat16) {
data_type_ = kDataTypeFloat;
} else {
data_type_ = kDataTypeInt;

Loading…
Cancel
Save