From 0294152f4177ae8a711c750f5bc2bf50d8fdc2a7 Mon Sep 17 00:00:00 2001
From: xiaogang <txg4794@163.com>
Date: Mon, 20 Apr 2020 20:06:47 +0800
Subject: [PATCH] fix: fix lstm compute bug (#3447)

---
 lite/backends/arm/math/lstm.cc | 8 +-------
 1 file changed, 1 insertion(+), 7 deletions(-)

diff --git a/lite/backends/arm/math/lstm.cc b/lite/backends/arm/math/lstm.cc
index 5a2a263bb4..cd8e012a28 100644
--- a/lite/backends/arm/math/lstm.cc
+++ b/lite/backends/arm/math/lstm.cc
@@ -33,6 +33,7 @@ void add_bias_rowwise(Tensor* input,
     for (int w = start_w; w < w_adds; ++w) {
       i_data[w] += b_data[w];
     }
+    i_data += width;
   }
 }
 void vector_dot(
@@ -67,15 +68,8 @@ void vector_dot(
   for (int i = 0; i < remain; ++i) {
     if (!v2) {
       out_ptr[i] = in_ptr[i] * v1_ptr[i];
-      ++out_ptr;
-      ++in_ptr;
-      ++v1_ptr;
     } else {
       out_ptr[i] = in_ptr[i] + v1_ptr[i] * v2_ptr[i];
-      ++out_ptr;
-      ++in_ptr;
-      ++v1_ptr;
-      ++v2_ptr;
     }
   }
 }
-- 
GitLab