Home
last modified time | relevance | path

Searched refs:softmax_params (Results 1 – 3 of 3) sorted by relevance

/cmsis-nn-3.7.0/Source/SoftmaxFunctions/
Darm_softmax_s16.c44 const cmsis_nn_softmax_lut_s16 *softmax_params, in arm_softmax_s16() argument
50 if (softmax_params->exp_lut == NULL || softmax_params->one_by_one_lut == NULL) in arm_softmax_s16()
78 const int16_t base = softmax_params->exp_lut[index]; in arm_softmax_s16()
79 … const int16_t slope = softmax_params->exp_lut[index + 1] - softmax_params->exp_lut[index]; in arm_softmax_s16()
100 const int16_t base = softmax_params->one_by_one_lut[index]; in arm_softmax_s16()
101 …const int16_t slope = softmax_params->one_by_one_lut[index + 1] - softmax_params->one_by_one_lut[i… in arm_softmax_s16()
/cmsis-nn-3.7.0/Tests/UnitTest/TestCases/test_arm_softmax_s16/
Dtest_arm_softmax_s16.c36 const cmsis_nn_softmax_lut_s16 softmax_params = {.exp_lut = softmax_s16_exp_lut, in softmax_s16_arm_softmax_s16() local
42 arm_softmax_s16(input_data, num_rows, row_size, mult, shift, &softmax_params, output); in softmax_s16_arm_softmax_s16()
/cmsis-nn-3.7.0/Include/
Darm_nnfunctions.h2152 const cmsis_nn_softmax_lut_s16 *softmax_params,