Searched refs:vec_in (Results 1 – 4 of 4) sorted by relevance
| /hal_nxp-latest/mcux/mcux-sdk/CMSIS/NN/Source/SoftmaxFunctions/ |
| D | arm_softmax_q15.c | 61 void arm_softmax_q15(const q15_t *vec_in, const uint16_t dim_vec, q15_t *p_out) in arm_softmax_q15() argument
70 if (vec_in[i] > base) in arm_softmax_q15()
72 base = vec_in[i]; in arm_softmax_q15()
86 if (vec_in[i] > base) in arm_softmax_q15()
88 shift = (uint8_t)__USAT(vec_in[i] - base, 5); in arm_softmax_q15()
103 if (vec_in[i] > base) in arm_softmax_q15()
106 shift = (uint8_t)__USAT(17 + base - vec_in[i], 5); in arm_softmax_q15()
|
| D | arm_softmax_q7.c | 61 void arm_softmax_q7(const q7_t *vec_in, const uint16_t dim_vec, q7_t *p_out) in arm_softmax_q7() argument
72 if (vec_in[i] > base) in arm_softmax_q7()
74 base = vec_in[i]; in arm_softmax_q7()
89 shift = (uint8_t)__USAT(vec_in[i] - base, 3); in arm_softmax_q7()
100 shift = (uint8_t)__USAT(13 + base - vec_in[i], 5); in arm_softmax_q7()
|
| D | arm_softmax_with_batch_q7.c | 62 void arm_softmax_with_batch_q7(const q7_t *vec_in, const uint16_t nb_batches, const uint16_t dim_ve… 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()
|
| /hal_nxp-latest/mcux/mcux-sdk/CMSIS/NN/Include/ |
| D | arm_nnfunctions.h | 1709 void arm_softmax_q7(const q7_t *vec_in, const uint16_t dim_vec, q7_t *p_out);
1724 void arm_softmax_with_batch_q7(const q7_t *vec_in, const uint16_t nb_batches, const uint16_t dim_ve…
1737 void arm_softmax_q15(const q15_t *vec_in, const uint16_t dim_vec, q15_t *p_out);
|
