From 45f9161922afe12691c57f1e5dbcaa8b0811aa40 Mon Sep 17 00:00:00 2001 From: nihuini Date: Thu, 18 Apr 2024 20:05:10 +0800 Subject: [PATCH] wip --- src/layer/arm/rnn_arm.cpp | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/src/layer/arm/rnn_arm.cpp b/src/layer/arm/rnn_arm.cpp index d2e3c3e8779..db92e87b885 100644 --- 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);