| /cmsis-nn-3.6.0-3.5.0/Tests/UnitTest/TestCases/Common/Softmax/ |
| D | softmax_int8_to_int16_template.json | 16 row_size
33 row_size
|
| /cmsis-nn-3.6.0-3.5.0/Source/SoftmaxFunctions/ |
| D | arm_softmax_u8.c | 46 const int32_t row_size, in arm_softmax_u8() argument
62 for (col = 1; col < row_size; ++col) in arm_softmax_u8()
70 for (col = 0; col < row_size; ++col) in arm_softmax_u8()
83 for (col = 0; col < row_size; ++col) in arm_softmax_u8()
97 input += row_size; in arm_softmax_u8()
98 output += row_size; in arm_softmax_u8()
|
| D | arm_nn_softmax_common_s8.c | 58 const int32_t row_size, in arm_nn_softmax_common_s8() argument
75 for (col = 1; col < row_size; ++col) in arm_nn_softmax_common_s8()
83 for (col = 0; col < row_size; ++col) in arm_nn_softmax_common_s8()
98 int16_t *output_s16 = (int16_t *)output + row_idx * row_size; in arm_nn_softmax_common_s8()
102 for (col = 0; col < row_size; ++col) in arm_nn_softmax_common_s8()
121 int8_t *output_s8 = (int8_t *)output + row_idx * row_size; in arm_nn_softmax_common_s8()
125 for (col = 0; col < row_size; ++col) in arm_nn_softmax_common_s8()
142 input += row_size; in arm_nn_softmax_common_s8()
|
| D | arm_softmax_s8_s16.c | 45 const int32_t row_size, 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_s16.c | 41 const int32_t row_size, in arm_softmax_s16() argument
59 for (col = 1; col < row_size; ++col) in arm_softmax_s16()
68 for (col = 0; col < row_size; ++col) in arm_softmax_s16()
105 for (col = 0; col < row_size; ++col) in arm_softmax_s16()
113 output += row_size; in arm_softmax_s16()
114 input += row_size; in arm_softmax_s16()
|
| D | arm_softmax_s8.c | 88 const int32_t row_size, in arm_softmax_s8() argument
105 int32_t vec_count = (row_size + 15) / 16; in arm_softmax_s8()
106 uint32_t r_count = (uint32_t)row_size; in arm_softmax_s8()
115 vec_count = row_size / 4; in arm_softmax_s8()
140 const int32_t tail_idx = row_size & ~3; in arm_softmax_s8()
141 for (int i = 0; i < (row_size & 3); i++) in arm_softmax_s8()
154 vec_count = row_size / 4; in arm_softmax_s8()
189 for (int i = 0; i < (row_size & 3); i++) in arm_softmax_s8()
205 input += row_size; in arm_softmax_s8()
206 output += row_size; in arm_softmax_s8()
[all …]
|
| /cmsis-nn-3.6.0-3.5.0/Tests/UnitTest/TestCases/test_arm_softmax_s8/ |
| D | test_arm_softmax_s8.c | 30 const int32_t row_size = SOFTMAX_ROW_SIZE; 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()
47 const int32_t row_size = SOFTMAX_ROW_SIZE; 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 | 30 const int32_t row_size = SOFTMAX_S8_S16_ROW_SIZE; 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()
47 const int32_t row_size = SOFTMAX_S8_S16_ROW_SIZE; 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 | 32 const int32_t row_size = SOFTMAX_S16_ROW_SIZE; 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/Include/ |
| D | arm_nnfunctions.h | 1563 const int32_t row_size,
1585 const int32_t row_size,
1614 const int32_t row_size,
1637 const int32_t row_size,
|
| D | arm_nnsupportfunctions.h | 774 const int32_t row_size,
|
