diff --git a/src/operators/kernel/arm/dequantize_bn_kernel.cpp b/src/operators/kernel/arm/dequantize_bn_kernel.cpp index 5000ae644a5f14c712f44fb2c407b10c451b3b66..d4b41b8b872ebe4fa21fd78f20a74a57d7ed0bc1 100644 --- a/src/operators/kernel/arm/dequantize_bn_kernel.cpp +++ b/src/operators/kernel/arm/dequantize_bn_kernel.cpp @@ -103,10 +103,10 @@ void DequantBNCompute(const FusionDequantBNParam *param) { f1 = vmlaq_f32(__bias, __scale, f1); f2 = vmlaq_f32(__bias, __scale, f2); f3 = vmlaq_f32(__bias, __scale, f3); - f0 = math::vActiveq_f32(f0); - f1 = math::vActiveq_f32(f1); - f2 = math::vActiveq_f32(f2); - f3 = math::vActiveq_f32(f3); + f0 = math::vActiveq_f32(f0); + f1 = math::vActiveq_f32(f1); + f2 = math::vActiveq_f32(f2); + f3 = math::vActiveq_f32(f3); vst1q_f32(y, f0); vst1q_f32(y + 4, f1); vst1q_f32(y + 8, f2); @@ -241,10 +241,10 @@ void DequantBNQuantCompute(const FusionDequantAddBNQuantParam *param) { f1 = vmlaq_f32(__bias, __scale, f1); f2 = vmlaq_f32(__bias, __scale, f2); f3 = vmlaq_f32(__bias, __scale, f3); - f0 = math::vActiveq_f32(f0); - f1 = math::vActiveq_f32(f1); - f2 = math::vActiveq_f32(f2); - f3 = math::vActiveq_f32(f3); + f0 = math::vActiveq_f32(f0); + f1 = math::vActiveq_f32(f1); + f2 = math::vActiveq_f32(f2); + f3 = math::vActiveq_f32(f3); f0 = vmulq_f32(__quant_scale, f0); f1 = vmulq_f32(__quant_scale, f1); f2 = vmulq_f32(__quant_scale, f2); @@ -273,7 +273,7 @@ void DequantBNQuantCompute(const FusionDequantAddBNQuantParam *param) { } } } else { - // TODO + // TODO(hjchen2) max_abs = std::max(max_abs, 1e-6f); } param->online_scale_->mutable_data()[0] = max_abs;