Home
last modified time | relevance | path

Searched refs:dim_vec (Results 1 – 10 of 10) sorted by relevance

/hal_nxp-latest/mcux/mcux-sdk/CMSIS/NN/Source/FullyConnectedFunctions/
Darm_fully_connected_q7.c69 const uint16_t dim_vec, in arm_fully_connected_q7() argument
89 arm_q7_to_q15_reordered_no_shift(pV, vec_buffer, dim_vec); in arm_fully_connected_q7()
95 uint16_t colCnt = dim_vec >> 2; in arm_fully_connected_q7()
98 pB2 = pB + dim_vec; in arm_fully_connected_q7()
118 colCnt = dim_vec & 0x3; in arm_fully_connected_q7()
133 pB += dim_vec; in arm_fully_connected_q7()
142 uint16_t colCnt = dim_vec >> 2; in arm_fully_connected_q7()
163 colCnt = dim_vec & 0x3; in arm_fully_connected_q7()
185 for (j = 0; j < dim_vec; j++) in arm_fully_connected_q7()
187 ip_out += pV[j] * pM[i * dim_vec + j]; in arm_fully_connected_q7()
Darm_fully_connected_q15.c67 const uint16_t dim_vec, in arm_fully_connected_q15() argument
80 const q15_t *pB2 = pB + dim_vec; in arm_fully_connected_q15()
92 uint16_t colCnt = dim_vec >> 2; in arm_fully_connected_q15()
95 pB2 = pB + dim_vec; in arm_fully_connected_q15()
114 colCnt = dim_vec & 0x3; in arm_fully_connected_q15()
129 pB = pB + dim_vec; in arm_fully_connected_q15()
139 uint16_t colCnt = dim_vec >> 2; in arm_fully_connected_q15()
158 colCnt = dim_vec & 0x3; in arm_fully_connected_q15()
180 for (j = 0; j < dim_vec; j++) in arm_fully_connected_q15()
182 ip_out += pV[j] * pM[i * dim_vec + j]; in arm_fully_connected_q15()
Darm_fully_connected_mat_q7_vec_q15.c70 const uint16_t dim_vec, in arm_fully_connected_mat_q7_vec_q15() argument
94 uint16_t colCnt = dim_vec >> 2; in arm_fully_connected_mat_q7_vec_q15()
97 pB2 = pB + dim_vec; in arm_fully_connected_mat_q7_vec_q15()
117 colCnt = dim_vec & 0x3; in arm_fully_connected_mat_q7_vec_q15()
132 pB += dim_vec; in arm_fully_connected_mat_q7_vec_q15()
142 uint16_t colCnt = dim_vec >> 2; in arm_fully_connected_mat_q7_vec_q15()
162 colCnt = dim_vec & 0x3; in arm_fully_connected_mat_q7_vec_q15()
182 for (j = 0; j < dim_vec; j++) in arm_fully_connected_mat_q7_vec_q15()
184 ip_out += pV[j] * pM[i * dim_vec + j]; in arm_fully_connected_mat_q7_vec_q15()
Darm_fully_connected_q15_opt.c98 const uint16_t dim_vec, in arm_fully_connected_q15_opt() argument
124 uint16_t colCnt = dim_vec >> 1; in arm_fully_connected_q15_opt()
181 colCnt = dim_vec & 0x1; in arm_fully_connected_q15_opt()
213 uint16_t colCnt = dim_vec >> 2; in arm_fully_connected_q15_opt()
233 colCnt = dim_vec & 0x3; in arm_fully_connected_q15_opt()
262 uint16_t colCnt = dim_vec >> 1; in arm_fully_connected_q15_opt()
288 colCnt = dim_vec & 0x1; in arm_fully_connected_q15_opt()
317 for (j = 0; j < dim_vec; j++) in arm_fully_connected_q15_opt()
Darm_fully_connected_q7_opt.c130 const uint16_t dim_vec, in arm_fully_connected_q7_opt() argument
148 arm_q7_to_q15_reordered_no_shift(pV, vec_buffer, dim_vec); in arm_fully_connected_q7_opt()
158 uint16_t colCnt = dim_vec >> 2; in arm_fully_connected_q7_opt()
319 colCnt = dim_vec & 0x3; in arm_fully_connected_q7_opt()
349 uint16_t colCnt = dim_vec >> 2; in arm_fully_connected_q7_opt()
369 colCnt = dim_vec & 0x3; in arm_fully_connected_q7_opt()
398 uint16_t colCnt = dim_vec >> 2; in arm_fully_connected_q7_opt()
443 colCnt = dim_vec & 0x3; in arm_fully_connected_q7_opt()
475 for (j = 0; j < dim_vec; j++) in arm_fully_connected_q7_opt()
Darm_fully_connected_mat_q7_vec_q15_opt.c118 const uint16_t dim_vec, in arm_fully_connected_mat_q7_vec_q15_opt() argument
145 uint16_t colCnt = dim_vec >> 1; in arm_fully_connected_mat_q7_vec_q15_opt()
260 colCnt = dim_vec & 0x1; in arm_fully_connected_mat_q7_vec_q15_opt()
291 uint16_t colCnt = dim_vec >> 2; in arm_fully_connected_mat_q7_vec_q15_opt()
311 colCnt = dim_vec & 0x3; in arm_fully_connected_mat_q7_vec_q15_opt()
339 uint16_t colCnt = dim_vec >> 1; in arm_fully_connected_mat_q7_vec_q15_opt()
367 colCnt = dim_vec & 0x1; in arm_fully_connected_mat_q7_vec_q15_opt()
398 for (j = 0; j < dim_vec; j++) in arm_fully_connected_mat_q7_vec_q15_opt()
/hal_nxp-latest/mcux/mcux-sdk/CMSIS/NN/Source/SoftmaxFunctions/
Darm_softmax_with_batch_q7.c62 …x_with_batch_q7(const q7_t *vec_in, const uint16_t nb_batches, const uint16_t dim_vec, q7_t *p_out) in arm_softmax_with_batch_q7() argument
66 arm_softmax_q7(vec_in, dim_vec, p_out); in arm_softmax_with_batch_q7()
67 vec_in += dim_vec; in arm_softmax_with_batch_q7()
68 p_out += dim_vec; in arm_softmax_with_batch_q7()
Darm_softmax_q7.c61 void arm_softmax_q7(const q7_t *vec_in, const uint16_t dim_vec, q7_t *p_out) in arm_softmax_q7() argument
70 for (i = 0; i < dim_vec; i++) in arm_softmax_q7()
87 for (i = 0; i < dim_vec; i++) in arm_softmax_q7()
96 for (i = 0; i < dim_vec; i++) in arm_softmax_q7()
Darm_softmax_q15.c61 void arm_softmax_q15(const q15_t *vec_in, const uint16_t dim_vec, q15_t *p_out) in arm_softmax_q15() argument
68 for (i = 0; i < dim_vec; i++) in arm_softmax_q15()
84 for (i = 0; i < dim_vec; i++) in arm_softmax_q15()
101 for (i = 0; i < dim_vec; i++) in arm_softmax_q15()
/hal_nxp-latest/mcux/mcux-sdk/CMSIS/NN/Include/
Darm_nnfunctions.h1108 const uint16_t dim_vec,
1189 const uint16_t dim_vec,
1214 const uint16_t dim_vec,
1239 const uint16_t dim_vec,
1264 const uint16_t dim_vec,
1289 const uint16_t dim_vec,
1709 void arm_softmax_q7(const q7_t *vec_in, const uint16_t dim_vec, q7_t *p_out);
1724 …_with_batch_q7(const q7_t *vec_in, const uint16_t nb_batches, const uint16_t dim_vec, q7_t *p_out);
1737 void arm_softmax_q15(const q15_t *vec_in, const uint16_t dim_vec, q15_t *p_out);