Searched refs:col (Results 1 – 9 of 9) sorted by relevance
| /cmsis-nn-3.6.0-3.5.0/Source/SoftmaxFunctions/ |
| D | arm_nn_softmax_common_s8.c | 67 int32_t col = 0; in arm_nn_softmax_common_s8() local
75 for (col = 1; col < row_size; ++col) in arm_nn_softmax_common_s8()
77 max = MAX(max, input[col]); in arm_nn_softmax_common_s8()
83 for (col = 0; col < row_size; ++col) in arm_nn_softmax_common_s8()
85 diff = input[col] - max; in arm_nn_softmax_common_s8()
102 for (col = 0; col < row_size; ++col) in arm_nn_softmax_common_s8()
104 diff = input[col] - max; in arm_nn_softmax_common_s8()
111 output_s16[col] = (int16_t)CLAMP(res, (int32_t)NN_Q15_MAX, (int32_t)NN_Q15_MIN); in arm_nn_softmax_common_s8()
115 output_s16[col] = NN_Q15_MIN; in arm_nn_softmax_common_s8()
125 for (col = 0; col < row_size; ++col) in arm_nn_softmax_common_s8()
[all …]
|
| D | arm_softmax_u8.c | 54 int32_t col = 0; in arm_softmax_u8() local
62 for (col = 1; col < row_size; ++col) in arm_softmax_u8()
64 max = MAX(max, input[col]); in arm_softmax_u8()
70 for (col = 0; col < row_size; ++col) in arm_softmax_u8()
72 diff = input[col] - max; in arm_softmax_u8()
83 for (col = 0; col < row_size; ++col) in arm_softmax_u8()
85 diff = input[col] - max; in arm_softmax_u8()
90 output[col] = (uint8_t)CLAMP(res, (int32_t)255, (int32_t)0); in arm_softmax_u8()
94 output[col] = 0; in arm_softmax_u8()
|
| D | arm_softmax_s16.c | 47 int32_t col = 0; in arm_softmax_s16() local
59 for (col = 1; col < row_size; ++col) in arm_softmax_s16()
61 max = MAX(max, input[col]); in arm_softmax_s16()
68 for (col = 0; col < row_size; ++col) in arm_softmax_s16()
70 diff = input[col] - max; in arm_softmax_s16()
82 cached_exp_results[col] = result; in arm_softmax_s16()
84 sum += cached_exp_results[col]; in arm_softmax_s16()
105 for (col = 0; col < row_size; ++col) in arm_softmax_s16()
108 int32_t result = (cached_exp_results[col] * one_by_one_result) >> right_shift; in arm_softmax_s16()
110 output[col] = (int16_t)result; in arm_softmax_s16()
|
| /cmsis-nn-3.6.0-3.5.0/Source/NNSupportFunctions/ |
| D | arm_nn_mat_mul_core_4x_s8.c | 76 int32_t col = col_base[j]; in arm_nn_mat_mul_core_4x_s8() local
77 sum_tmp += col; in arm_nn_mat_mul_core_4x_s8()
78 acc_n0 += ip_row_0[j] * col; in arm_nn_mat_mul_core_4x_s8()
79 acc_n1 += ip_row_1[j] * col; in arm_nn_mat_mul_core_4x_s8()
80 acc_n2 += ip_row_2[j] * col; in arm_nn_mat_mul_core_4x_s8()
81 acc_n3 += ip_row_3[j] * col; in arm_nn_mat_mul_core_4x_s8()
100 : [col] "+r"(col_base), in arm_nn_mat_mul_core_4x_s8()
|
| D | arm_nn_mat_mul_core_1x_s8.c | 76 int32_t col = col_base[j]; in arm_nn_mat_mul_core_1x_s8() local
77 sum_tmp += col; in arm_nn_mat_mul_core_1x_s8()
78 acc_n0 += row_base[j] * col; in arm_nn_mat_mul_core_1x_s8()
91 … : [col] "+r"(col_base), [sum] "+Te"(sum_tmp), [row0] "+r"(row_base), [out0] "+Te"(acc_n0) in arm_nn_mat_mul_core_1x_s8()
|
| D | arm_nn_mat_mult_nt_t_s8.c | 85 int32_t col = col_base[j]; in arm_nn_mat_mult_nt_t_s8() local
86 sum_tmp += col; in arm_nn_mat_mult_nt_t_s8()
87 acc_n0 += lhs_vec[j] * col; in arm_nn_mat_mult_nt_t_s8()
88 acc_n1 += ip_row_1[j] * col; in arm_nn_mat_mult_nt_t_s8()
89 acc_n2 += ip_row_2[j] * col; in arm_nn_mat_mult_nt_t_s8()
90 acc_n3 += ip_row_3[j] * col; in arm_nn_mat_mult_nt_t_s8()
116 : [col] "+r"(col_base), in arm_nn_mat_mult_nt_t_s8()
166 int32_t col = col_base[j]; in arm_nn_mat_mult_nt_t_s8() local
167 sum_tmp += col; in arm_nn_mat_mult_nt_t_s8()
168 acc_n0 += lhs_vec[j] * col; in arm_nn_mat_mult_nt_t_s8()
[all …]
|
| /cmsis-nn-3.6.0-3.5.0/Source/ConvolutionFunctions/ |
| D | arm_nn_depthwise_conv_s8_core.c | 41 const int16_t *col, in arm_nn_depthwise_conv_s8_core() argument
71 const int16_t *col_0 = col + offset; in arm_nn_depthwise_conv_s8_core()
72 const int16_t *col_1 = col + kernel_size * num_ch + offset; in arm_nn_depthwise_conv_s8_core()
165 const int16_t *col_pos_0 = col + ch_idx; in arm_nn_depthwise_conv_s8_core()
206 (void)col; in arm_nn_depthwise_conv_s8_core()
|
| /cmsis-nn-3.6.0-3.5.0/Include/ |
| D | arm_nnsupportfunctions.h | 185 const int16_t *col,
|
| /cmsis-nn-3.6.0-3.5.0/Tests/UnitTest/ |
| D | generate_test_data.py | 1818 col = dims[1]
1828 for i_col in range(col):
|
