Преглед изворни кода

CMSIS-NN: Add back missing prototype (#1438)

* CMSIS-NN: Add back missing prototype

Adds back arm_nn_mat_mult_kernel_q7_q15_reordered.
Måns Nilsson пре 3 година
родитељ
комит
0edf39abb8
1 измењених фајлова са 25 додато и 2 уклоњено
  1. 25 2
      CMSIS/NN/Include/arm_nnsupportfunctions.h

+ 25 - 2
CMSIS/NN/Include/arm_nnsupportfunctions.h

@@ -22,8 +22,8 @@
  * Description:  Public header file of support functions for CMSIS NN Library
  *
 
- * $Date:        24. February 2022
- * $Revision:    V.6.2.0
+ * $Date:        16. March 2022
+ * $Revision:    V.6.2.1
  *
  * Target Processor:  Cortex-M CPUs
  * -------------------------------------------------------------------- */
@@ -550,6 +550,29 @@ q7_t *arm_nn_depthwise_conv_nt_t_s8(const q7_t *lhs,
                                     const int32_t *const output_bias,
                                     q7_t *out);
 
+/**
+ *@brief Matrix-multiplication function for convolution with reordered columns
+ *@param[in]       pA          pointer to operand A
+ *@param[in]       pInBuffer   pointer to operand B, always conssists of 2 vectors
+ *@param[in]       ch_im_out   numRow of A
+ *@param[in]       numCol_A    numCol of A
+ *@param[in]       bias_shift  amount of left-shift for bias
+ *@param[in]       out_shift   amount of right-shift for output
+ *@param[in]       bias        the bias
+ *@param[in,out]   pOut        pointer to output
+ *@return     The function returns the incremented output pointer
+ *
+ *@details  This function assumes that data in pInBuffer are reordered
+ */
+q7_t *arm_nn_mat_mult_kernel_q7_q15_reordered(const q7_t *pA,
+                                              const q15_t *pInBuffer,
+                                              const uint16_t ch_im_out,
+                                              const uint16_t numCol_A,
+                                              const uint16_t bias_shift,
+                                              const uint16_t out_shift,
+                                              const q7_t *bias,
+                                              q7_t *pOut);
+
 /**
   @brief         Read 2 q15 elements and post increment pointer.
   @param[in]     in_q15   Pointer to pointer that holds address of input.