Home
last modified time | relevance | path

Searched refs:num_rows (Results 1 – 11 of 11) sorted by relevance

/cmsis-nn-3.6.0-3.5.0/Tests/UnitTest/TestCases/Common/Softmax/
Dsoftmax_int8_to_int16_template.json15 num_rows,
32 num_rows,
/cmsis-nn-3.6.0-3.5.0/Source/SoftmaxFunctions/
Darm_softmax_s8_s16.c44 const int32_t num_rows, 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.c45 const int32_t num_rows, in arm_softmax_u8() argument
57 for (row_idx = 0; row_idx < num_rows; ++row_idx) in arm_softmax_u8()
Darm_nn_softmax_common_s8.c57 const int32_t num_rows, in arm_nn_softmax_common_s8() argument
70 for (row_idx = 0; row_idx < num_rows; ++row_idx) in arm_nn_softmax_common_s8()
Darm_softmax_s16.c40 const int32_t num_rows, in arm_softmax_s16() argument
55 for (row_idx = 0; row_idx < num_rows; ++row_idx) in arm_softmax_s16()
Darm_softmax_s8.c87 const int32_t num_rows, in arm_softmax_s8() argument
101 for (int i_num_rows = 0; i_num_rows < num_rows; ++i_num_rows) 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/
Dtest_arm_softmax_s8.c29 const int32_t num_rows = SOFTMAX_NUM_ROWS; 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()
46 const int32_t num_rows = SOFTMAX_NUM_ROWS; 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/
Dtest_arm_softmax_s8_s16.c29 const int32_t num_rows = SOFTMAX_S8_S16_NUM_ROWS; 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()
46 const int32_t num_rows = SOFTMAX_S8_S16_NUM_ROWS; 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/
Dtest_arm_softmax_s16.c31 const int32_t num_rows = SOFTMAX_S16_NUM_ROWS; 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/
Darm_nnfunctions.h1562 const int32_t num_rows,
1584 const int32_t num_rows,
1613 const int32_t num_rows,
1636 const int32_t num_rows,
Darm_nnsupportfunctions.h773 const int32_t num_rows,