[Arm] fix viterbi_decode compute best_path (#10101)

PaddlePaddle · Mar 15, 2023 · f0dd78c · f0dd78c
1 parent 0dd6280
commit f0dd78c
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/lite/backends/arm/math/viterbi_decode.cc b/lite/backends/arm/math/viterbi_decode.cc
@@ -359,7 +359,7 @@ void viterbi_decode(
              batch_path_ptr + (actual_len - last_ids_index) * batch,
              batch);
   arange(batch_offset_ptr, batch, n_labels);
-  for (int sl = 0; sl < max_seq_len - 1; sl++) {
+  for (int sl = max_seq_len - 2; sl >= 0; sl--) {
     ++last_ids_index;
     vector_add(1, left_length_ptr, left_length_ptr, batch);
     vector_add(batch_offset_ptr, last_ids_ptr, gather_idx_ptr, batch);