fix

lxwlaq · lxwlaq · commit 4d373a8961d9 · 2022-03-24T10:35:54.000Z
diff --git a/lite/kernels/arm/rnn_compute.cc b/lite/kernels/arm/rnn_compute.cc
@@ -801,11 +801,9 @@ void RnnCompute::Run() {
     }
   }
 
-  lite::arm::math::concat_func<float>(
-      last_h_unbind_t, stride1.size(), state[0]);
+  lite::arm::math::concat_func<float>(last_h_unbind_t, 0, state[0]);
   if ("LSTM" == mode) {
-    lite::arm::math::concat_func<float>(
-        last_c_unbind_t, stride2.size(), state[1]);
+    lite::arm::math::concat_func<float>(last_c_unbind_t, 0, state[1]);
   }
 
   // output_holder != output

Original file line number	Diff line number	Diff line change
`@@ -801,11 +801,9 @@ void RnnCompute::Run() {`
`801`	`801`	`}`
`802`	`802`	`}`
`803`	`803`
`804`		`- lite::arm::math::concat_func<float>(`
`805`		`- last_h_unbind_t, stride1.size(), state[0]);`
	`804`	`+ lite::arm::math::concat_func<float>(last_h_unbind_t, 0, state[0]);`
`806`	`805`	`if ("LSTM" == mode) {`
`807`		`- lite::arm::math::concat_func<float>(`
`808`		`- last_c_unbind_t, stride2.size(), state[1]);`
	`806`	`+ lite::arm::math::concat_func<float>(last_c_unbind_t, 0, state[1]);`
`809`	`807`	`}`
`810`	`808`
`811`	`809`	`// output_holder != output`