| /cmsis-nn-3.6.0-3.5.0/Source/SoftmaxFunctions/ |
| D | arm_softmax_s8_s16.c | 48 const int32_t diff_min, 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 | 49 const int32_t diff_min, in arm_softmax_u8() argument
73 if (diff >= diff_min) in arm_softmax_u8()
86 if (diff >= diff_min) in arm_softmax_u8()
|
| D | arm_nn_softmax_common_s8.c | 61 const int32_t diff_min, in arm_nn_softmax_common_s8() argument
86 if (diff >= diff_min) in arm_nn_softmax_common_s8()
106 if (diff >= diff_min) in arm_nn_softmax_common_s8()
128 if (diff >= diff_min) in arm_nn_softmax_common_s8()
|
| D | arm_softmax_s8.c | 91 const int32_t diff_min, in arm_softmax_s8() argument
123 mve_pred16_t p = vcmpgeq_n_s32(ip, diff_min); in arm_softmax_s8()
144 if (diff >= diff_min) in arm_softmax_s8()
162 mve_pred16_t p = vcmpgeq_n_s32(ip, diff_min); in arm_softmax_s8()
192 if (diff >= diff_min) 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()
|
| /cmsis-nn-3.6.0-3.5.0/Tests/UnitTest/TestCases/test_arm_softmax_s8/ |
| D | test_arm_softmax_s8.c | 33 const int32_t diff_min = SOFTMAX_DIFF_MIN; 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()
50 const int32_t diff_min = 0x7FFFFFFF; 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 | 33 const int32_t diff_min = SOFTMAX_S8_S16_DIFF_MIN; 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()
50 const int32_t diff_min = 0x7FFFFFFF; 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/Include/ |
| D | arm_nnfunctions.h | 1566 const int32_t diff_min,
1588 const int32_t diff_min,
1640 const int32_t diff_min,
|
| D | arm_nnsupportfunctions.h | 777 const int32_t diff_min,
|
| /cmsis-nn-3.6.0-3.5.0/Tests/UnitTest/ |
| D | model_extractor.py | 41 self.diff_min, self.input_multiplier, self.input_left_shift = 0, 0, 0
111 f.write("#define {}_DIFF_MIN {}\n".format(prefix, -self.diff_min))
|
| D | generate_test_data.py | 1069 self.diff_min = ((1 << self.softmax_input_integer_bits) - 1) * \
1072 self.diff_min = math.floor(self.diff_min)
1087 f.write("#define {}_DIFF_MIN {}\n".format(prefix, -self.diff_min))
|
