kernel should not change the cell tensor shape

Author: Hakim7267

lib/src/kernels/common/impl/mli_krn_lstm_cell_ref.h

@@ -113,19 +113,11 @@ MLI_FORCE_INLINE void lstm_cell_prepare_and_run(
 
     mli_tensor rnn_out;
     rnn_out.data = out->data;
-    rnn_out.rank = 2;
-    rnn_out.shape[0] = 1;
-    rnn_out.shape[1] = lstm_out_elements;
-    rnn_out.mem_stride[0] = rnn_out.shape[1];
-    rnn_out.mem_stride[1] = 1;
+    rnn_out.rank = 1;
+    rnn_out.shape[0] = lstm_out_elements;
+    rnn_out.mem_stride[0] = 1;
     rnn_out.el_type = in->el_type;
 
-    cell->rank = 2;
-    cell->shape[0] = forget_gate.shape[0];
-    cell->shape[1] = forget_gate.shape[1];
-    cell->mem_stride[0] = forget_gate.mem_stride[0];
-    cell->mem_stride[1] = forget_gate.mem_stride[1];
-
     for (int timestep = 0; timestep < seq_len; timestep++) {
 
         // Step 1: Applying Dense
@@ -173,9 +165,7 @@ MLI_FORCE_INLINE void lstm_cell_prepare_and_run(
         temp.data = rnn_out.data;
         temp.rank = rnn_out.rank;
         temp.shape[0] = rnn_out.shape[0];
-        temp.shape[1] = rnn_out.shape[1];
         temp.mem_stride[0] = rnn_out.mem_stride[0];
-        temp.mem_stride[1] = rnn_out.mem_stride[1];
         temp.el_type = rnn_out.el_type;
         temp.el_params = out->el_params;
 
@@ -249,4 +239,4 @@ MLI_FORCE_INLINE void lstm_cell_prepare_and_run(
 } // namespace mli
 } // namespace krn
 
-#endif  //_MLI_KRN_LSTM_CELL_REF_H_
+#endif  //_MLI_KRN_LSTM_CELL_REF_H_

lib/src/kernels/common/impl/mli_krn_lstm_cell_vdsp.h

@@ -123,19 +123,11 @@ MLI_FORCE_INLINE void lstm_cell_prepare_and_run(
 
     mli_tensor rnn_out;
     rnn_out.data = out->data;
-    rnn_out.rank = 2;
-    rnn_out.shape[0] = 1;
-    rnn_out.shape[1] = lstm_out_elements;
-    rnn_out.mem_stride[0] = rnn_out.shape[1];
-    rnn_out.mem_stride[1] = 1;
+    rnn_out.rank = 1;
+    rnn_out.shape[0] = lstm_out_elements;
+    rnn_out.mem_stride[0] = 1;
     rnn_out.el_type = in->el_type;
 
-    cell->rank = 2;
-    cell->shape[0] = tmp_gate.shape[0];
-    cell->shape[1] = tmp_gate.shape[1];
-    cell->mem_stride[0] = tmp_gate.mem_stride[0];
-    cell->mem_stride[1] = tmp_gate.mem_stride[1];
-
     struct s8asym_quant_params out_params_sigm;
     struct s8asym_quant_params out_params_tanh;
 
@@ -236,9 +228,7 @@ MLI_FORCE_INLINE void lstm_cell_prepare_and_run(
         temp.data = rnn_out.data;
         temp.rank = rnn_out.rank;
         temp.shape[0] = rnn_out.shape[0];
-        temp.shape[1] = rnn_out.shape[1];
         temp.mem_stride[0] = rnn_out.mem_stride[0];
-        temp.mem_stride[1] = rnn_out.mem_stride[1];
         temp.el_type = rnn_out.el_type;
         temp.el_params = out->el_params;
 
@@ -314,4 +304,4 @@ MLI_FORCE_INLINE void lstm_cell_prepare_and_run(
 } // namespace mli
 } // namespace krn
 
-#endif  //_MLI_KRN_LSTM_CELL_VDSP_H_
+#endif  //_MLI_KRN_LSTM_CELL_VDSP_H_