| /cmsis-nn-3.6.0-3.5.0/Source/SoftmaxFunctions/ |
| D | arm_softmax_s8_s16.c | 46 const int32_t mult, in arm_softmax_s8_s16() argument
51 … arm_nn_softmax_common_s8(input, num_rows, row_size, mult, shift, diff_min, true, (void *)output); in arm_softmax_s8_s16()
|
| D | arm_softmax_u8.c | 47 const int32_t mult, in arm_softmax_u8() argument
75 sum += DIV_POW2(EXP_ON_NEG(MUL_SAT(diff * mask, mult)), ACCUM_BITS); in arm_softmax_u8()
89 … DIV_POW2(MUL_SAT(shifted_scale, EXP_ON_NEG(MUL_SAT(diff * mask, mult))), bits_over_unit); in arm_softmax_u8()
|
| D | arm_nn_softmax_common_s8.c | 59 const int32_t mult, in arm_nn_softmax_common_s8() argument
88 sum += DIV_POW2(EXP_ON_NEG(MUL_SAT(diff * mask, mult)), ACCUM_BITS); in arm_nn_softmax_common_s8()
109 … DIV_POW2(MUL_SAT(shifted_scale, EXP_ON_NEG(MUL_SAT(diff * mask, mult))), bits_over_unit) + in arm_nn_softmax_common_s8()
131 … DIV_POW2(MUL_SAT(shifted_scale, EXP_ON_NEG(MUL_SAT(diff * mask, mult))), bits_over_unit) + in arm_nn_softmax_common_s8()
|
| D | arm_softmax_s8.c | 89 const int32_t mult, in arm_softmax_s8() argument
128 int32x4_t res = MUL_SAT_MVE(ip, vdupq_n_s32(mult)); in arm_softmax_s8()
146 sum += DIV_POW2(EXP_ON_NEG(MUL_SAT(diff * mask, mult)), ACCUM_BITS); in arm_softmax_s8()
170 tmp_res = MUL_SAT_MVE(ip, vdupq_n_s32(mult)); in arm_softmax_s8()
195 … DIV_POW2(MUL_SAT(shifted_scale, EXP_ON_NEG(MUL_SAT(diff * mask, mult))), bits_over_unit) + in arm_softmax_s8()
209 … arm_nn_softmax_common_s8(input, num_rows, row_size, mult, shift, diff_min, false, (void *)output); in arm_softmax_s8()
|
| D | arm_softmax_s16.c | 42 const int32_t mult, in arm_softmax_s16() argument
71 const int32_t scaled_diff = arm_nn_requantize(diff, mult, shift); in arm_softmax_s16()
|
| /cmsis-nn-3.6.0-3.5.0/Tests/UnitTest/TestCases/test_arm_softmax_s8/ |
| D | test_arm_softmax_s8.c | 31 const int32_t mult = SOFTMAX_INPUT_MULT; in softmax_arm_softmax_s8() local
39 arm_softmax_s8(input_data, num_rows, row_size, mult, shift, diff_min, output); in softmax_arm_softmax_s8()
48 const int32_t mult = SOFTMAX_INPUT_MULT; in softmax_invalid_diff_min_arm_softmax_s8() local
62 arm_softmax_s8(input_data, num_rows, row_size, mult, shift, diff_min, output); in softmax_invalid_diff_min_arm_softmax_s8()
|
| /cmsis-nn-3.6.0-3.5.0/Tests/UnitTest/TestCases/test_arm_softmax_s8_s16/ |
| D | test_arm_softmax_s8_s16.c | 31 const int32_t mult = SOFTMAX_S8_S16_INPUT_MULT; in softmax_s8_s16_arm_softmax_s8_s16() local
39 arm_softmax_s8_s16(input_data, num_rows, row_size, mult, shift, diff_min, output); in softmax_s8_s16_arm_softmax_s8_s16()
48 const int32_t mult = SOFTMAX_S8_S16_INPUT_MULT; in softmax_s8_s16_invalid_diff_min_arm_softmax_s8_s16() local
62 arm_softmax_s8_s16(input_data, num_rows, row_size, mult, shift, diff_min, output); in softmax_s8_s16_invalid_diff_min_arm_softmax_s8_s16()
|
| /cmsis-nn-3.6.0-3.5.0/Tests/UnitTest/TestCases/test_arm_softmax_s16/ |
| D | test_arm_softmax_s16.c | 33 const int32_t mult = SOFTMAX_S16_INPUT_MULT; 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.6.0-3.5.0/Source/NNSupportFunctions/ |
| D | arm_nn_depthwise_conv_nt_t_padded_s8.c | 123 const int32x4_t mult = vldrwq_s32(out_mult); in arm_nn_depthwise_conv_nt_t_padded_s8() local
128 out_0 = arm_requantize_mve_32x4(out_0, mult, shift); in arm_nn_depthwise_conv_nt_t_padded_s8()
135 out_1 = arm_requantize_mve_32x4(out_1, mult, shift); in arm_nn_depthwise_conv_nt_t_padded_s8()
141 out_2 = arm_requantize_mve_32x4(out_2, mult, shift); in arm_nn_depthwise_conv_nt_t_padded_s8()
147 out_3 = arm_requantize_mve_32x4(out_3, mult, shift); in arm_nn_depthwise_conv_nt_t_padded_s8()
|
| D | arm_nn_depthwise_conv_nt_t_s8.c | 117 const int32x4_t mult = vldrwq_s32(out_mult); in arm_nn_depthwise_conv_nt_t_s8() local
123 out_0 = arm_requantize_mve_32x4(out_0, mult, shift); in arm_nn_depthwise_conv_nt_t_s8()
129 out_1 = arm_requantize_mve_32x4(out_1, mult, shift); in arm_nn_depthwise_conv_nt_t_s8()
135 out_2 = arm_requantize_mve_32x4(out_2, mult, shift); in arm_nn_depthwise_conv_nt_t_s8()
141 out_3 = arm_requantize_mve_32x4(out_3, mult, shift); in arm_nn_depthwise_conv_nt_t_s8()
|
| /cmsis-nn-3.6.0-3.5.0/Source/ConvolutionFunctions/ |
| D | arm_nn_depthwise_conv_s8_core.c | 133 const int32x4_t mult = vldrwq_s32(out_mult); in arm_nn_depthwise_conv_s8_core() local
138 out_0 = arm_requantize_mve_32x4(out_0, mult, shift); in arm_nn_depthwise_conv_s8_core()
139 out_1 = arm_requantize_mve_32x4(out_1, mult, shift); in arm_nn_depthwise_conv_s8_core()
184 const int32x4_t mult = vldrwq_z_s32(out_mult, p); in arm_nn_depthwise_conv_s8_core() local
187 col_0_sum = arm_requantize_mve_32x4(col_0_sum, mult, shift); in arm_nn_depthwise_conv_s8_core()
188 col_1_sum = arm_requantize_mve_32x4(col_1_sum, mult, shift); in arm_nn_depthwise_conv_s8_core()
|
| D | arm_depthwise_conv_s8_opt.c | 191 const int32x4_t mult = vldrwq_s32(&output_mult[offset]); in arm_depthwise_conv_s8_opt() local
194 out_0 = arm_requantize_mve_32x4(out_0, mult, shift); in arm_depthwise_conv_s8_opt()
|
| /cmsis-nn-3.6.0-3.5.0/Include/ |
| D | arm_nnsupportfunctions.h | 775 const int32_t mult,
813 int64_t mult = 1 << 30; in arm_nn_doubling_high_mult() local
817 mult = 1 - mult; in arm_nn_doubling_high_mult()
820 mult = mult + (int64_t)m1 * m2; in arm_nn_doubling_high_mult()
824 result = (int32_t)(mult / (1ll << 31)); in arm_nn_doubling_high_mult()
850 union arm_nn_long_long mult; in arm_nn_doubling_high_mult_no_sat() local
853 mult.word.low = 1 << 30; in arm_nn_doubling_high_mult_no_sat()
854 mult.word.high = 0; in arm_nn_doubling_high_mult_no_sat()
857 mult.long_long = mult.long_long + (int64_t)m1 * m2; in arm_nn_doubling_high_mult_no_sat()
861 result = (int32_t)(mult.long_long >> 31); in arm_nn_doubling_high_mult_no_sat()
|
| D | arm_nnfunctions.h | 1564 const int32_t mult,
1586 const int32_t mult,
1615 const int32_t mult,
1638 const int32_t mult,
|
