arm neon optimization for rnn (Tencent#2601)

CensorKo · Jan 15, 2021 · 4f2907d · 4f2907d
1 parent 6672b09
commit 4f2907d
Show file tree

Hide file tree

Showing 5 changed files with 1,492 additions and 4 deletions.
diff --git a/src/layer/arm/gru_arm.h b/src/layer/arm/gru_arm.h
@@ -41,9 +41,9 @@ class GRU_arm : virtual public GRU
     int forward_bf16s(const std::vector<Mat>& bottom_blobs, std::vector<Mat>& top_blobs, const Option& opt) const;
 
 public:
-    Mat weight_hc_data_packed;
-    Mat bias_c_data_packed;
     Mat weight_xc_data_packed;
+    Mat bias_c_data_packed;
+    Mat weight_hc_data_packed;
 };
 
 } // namespace ncnn

diff --git a/src/layer/arm/lstm_arm.h b/src/layer/arm/lstm_arm.h
@@ -41,9 +41,9 @@ class LSTM_arm : virtual public LSTM
     int forward_bf16s(const std::vector<Mat>& bottom_blobs, std::vector<Mat>& top_blobs, const Option& opt) const;
 
 public:
-    Mat weight_hc_data_packed;
-    Mat bias_c_data_packed;
     Mat weight_xc_data_packed;
+    Mat bias_c_data_packed;
+    Mat weight_hc_data_packed;
 };
 
 } // namespace ncnn