wip

Tencent · Apr 18, 2024 · 45f9161 · 45f9161
1 parent 9275f17
commit 45f9161
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/src/layer/arm/rnn_arm.cpp b/src/layer/arm/rnn_arm.cpp
@@ -594,9 +594,9 @@ int RNN_arm::create_pipeline_int8(const Option& opt)
             float32x4_t _reciprocal_hc = vdivq_f32(_one, _hc);
 #else
             float32x4_t _reciprocal_xc = vrecpeq_f32(_xc);
-            _reciprocal_xc = vmulq_f32(vrecpsq_f32(_xc, _reciprocal_xc_R), _reciprocal_xc_R);
+            _reciprocal_xc = vmulq_f32(vrecpsq_f32(_xc, _reciprocal_xc), _reciprocal_xc);
             float32x4_t _reciprocal_hc = vrecpeq_f32(_hc);
-            _reciprocal_hc = vmulq_f32(vrecpsq_f32(_hc, _reciprocal_hc_R), _reciprocal_hc_R);
+            _reciprocal_hc = vmulq_f32(vrecpsq_f32(_hc, _reciprocal_hc), _reciprocal_hc);
 #endif
 
             vst1q_f32(weight_xc_int8_descales_ptr, _reciprocal_xc);