Просмотр исходного кода

CMSIS-NN: Fix for batch greater than 1 case in Fully Connected

The ouptut pointer is updated for batch greater than 1 for
Fully Connected operator. The fix is applicable for all Cortex-M CPUs
except for Cortex-M55.

Change-Id: I1d8b632ca9b11ede60ed601b67b0eea4df698ea2
Felix Johnny 5 лет назад
Родитель
Сommit
62ed6cd35b
1 измененных файлов с 3 добавлено и 2 удалено
  1. 3 2
      CMSIS/NN/Source/FullyConnectedFunctions/arm_fully_connected_s8.c

+ 3 - 2
CMSIS/NN/Source/FullyConnectedFunctions/arm_fully_connected_s8.c

@@ -21,8 +21,8 @@
  * Title:        arm_fully_connected_s8
  * Description:  Fully connected function compatible with TF Lite.
  *
- * $Date:        30. January 2020
- * $Revision:    V.1.0.0
+ * $Date:        April 1, 2020
+ * $Revision:    V.1.5.0
  *
  * Target Processor:  Cortex-M and Cortex-A cores
  *
@@ -271,6 +271,7 @@ arm_fully_connected_s8(const int8_t *input,
                                  output_activation_min,
                                  output_activation_max);
         input += col_dim;
+        output += row_dim;
         batch_cnt--;
     }
     return (ARM_MATH_SUCCESS);