Home
last modified time | relevance | path

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

/cmsis-nn-latest/Source/SoftmaxFunctions/
Darm_softmax_s8_s16.c48 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()
Darm_softmax_u8.c49 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()
Darm_nn_softmax_common_s8.c61 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()
Darm_softmax_s8.c91 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-latest/Tests/UnitTest/TestCases/test_arm_softmax_s8/
Dtest_arm_softmax_s8.c33 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-latest/Tests/UnitTest/TestCases/test_arm_softmax_s8_s16/
Dtest_arm_softmax_s8_s16.c33 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-latest/Tests/UnitTest/
Dsoftmax_settings.py85 self.diff_min = ((1 << self.softmax_input_integer_bits) - 1) * \
88 self.diff_min = math.floor(self.diff_min)
103 f.write("#define {}_DIFF_MIN {}\n".format(prefix, -self.diff_min))
Dmodel_extractor.py43 self.diff_min, self.input_multiplier, self.input_left_shift = 0, 0, 0
107 f.write("#define {}_DIFF_MIN {}\n".format(prefix, -self.diff_min))
/cmsis-nn-latest/Include/
Darm_nnfunctions.h2101 const int32_t diff_min,
2123 const int32_t diff_min,
2175 const int32_t diff_min,
Darm_nnsupportfunctions.h1260 const int32_t diff_min,