| /cmsis-nn-3.6.0-3.5.0/Source/ActivationFunctions/ |
| D | arm_relu_q15.c | 57 int16_t *input = data; in arm_relu_q15() local
65 in = arm_nn_read_q15x2_ia((const int16_t **)&input); in arm_relu_q15()
79 if (*input < 0) in arm_relu_q15()
81 *input = 0; in arm_relu_q15()
83 input++; in arm_relu_q15()
|
| D | arm_relu_q7.c | 57 int8_t *input = data; in arm_relu_q7() local
65 in = arm_nn_read_s8x4_ia((const int8_t **)&input); in arm_relu_q7()
81 if (*input < 0) in arm_relu_q7()
83 *input = 0; in arm_relu_q7()
85 input++; in arm_relu_q7()
|
| D | arm_nn_activation_s16.c | 50 void arm_nn_activation_s16(const int16_t *input, in arm_nn_activation_s16() argument
74 for (int i = 0; i < size; ++i, input++, output++) in arm_nn_activation_s16()
76 int32_t input_data = ((*input) * input_multiplier); in arm_nn_activation_s16()
|
| /cmsis-nn-3.6.0-3.5.0/Source/ReshapeFunctions/ |
| D | arm_reshape_s8.c | 50 void arm_reshape_s8(const int8_t *input, int8_t *output, const uint32_t total_size) in arm_reshape_s8() argument
52 arm_memcpy_s8(output, input, total_size); in arm_reshape_s8()
|
| /cmsis-nn-3.6.0-3.5.0/Source/SoftmaxFunctions/ |
| D | arm_softmax_u8.c | 44 void arm_softmax_u8(const uint8_t *input, in arm_softmax_u8() argument
60 uint8_t max = *input; in arm_softmax_u8()
64 max = MAX(max, input[col]); in arm_softmax_u8()
72 diff = input[col] - max; in arm_softmax_u8()
85 diff = input[col] - max; in arm_softmax_u8()
97 input += row_size; in arm_softmax_u8()
|
| D | arm_nn_softmax_common_s8.c | 56 void arm_nn_softmax_common_s8(const int8_t *input, in arm_nn_softmax_common_s8() argument
73 int8_t max = *input; in arm_nn_softmax_common_s8()
77 max = MAX(max, input[col]); in arm_nn_softmax_common_s8()
85 diff = input[col] - max; in arm_nn_softmax_common_s8()
104 diff = input[col] - max; in arm_nn_softmax_common_s8()
127 diff = input[col] - max; in arm_nn_softmax_common_s8()
142 input += row_size; in arm_nn_softmax_common_s8()
|
| D | arm_softmax_s8_s16.c | 43 void arm_softmax_s8_s16(const int8_t *input, 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 | 39 arm_cmsis_nn_status arm_softmax_s16(const int16_t *input, in arm_softmax_s16() argument
58 int16_t max = *input; in arm_softmax_s16()
61 max = MAX(max, input[col]); in arm_softmax_s16()
70 diff = input[col] - max; in arm_softmax_s16()
114 input += row_size; in arm_softmax_s16()
|
| D | arm_softmax_s8.c | 86 void arm_softmax_s8(const int8_t *input, in arm_softmax_s8() argument
110 const int8x16_t ip = vldrbq_z_s8(&input[i * 16], p); in arm_softmax_s8()
121 int32x4_t ip = vldrbq_s32(&input[idx * 4]); in arm_softmax_s8()
143 const int32_t diff = input[tail_idx + i] - max; in arm_softmax_s8()
159 int32x4_t ip = vldrbq_s32(&input[idx]); in arm_softmax_s8()
191 int32_t diff = input[tail_idx + i] - max; in arm_softmax_s8()
205 input += row_size; 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/Source/ConcatenationFunctions/ |
| D | arm_concatenation_s8_x.c | 49 void arm_concatenation_s8_x(const int8_t *input, in arm_concatenation_s8_x() argument
67 arm_memcpy_s8(output, input, input_x); in arm_concatenation_s8_x()
68 input += input_x; in arm_concatenation_s8_x()
|
| D | arm_concatenation_s8_z.c | 49 void arm_concatenation_s8_z(const int8_t *input, in arm_concatenation_s8_z() argument
67 arm_memcpy_s8(output, input, input_copy_size); in arm_concatenation_s8_z()
68 input += input_copy_size; in arm_concatenation_s8_z()
|
| D | arm_concatenation_s8_y.c | 49 void arm_concatenation_s8_y(const int8_t *input, in arm_concatenation_s8_y() argument
68 arm_memcpy_s8(output, input, input_copy_size); in arm_concatenation_s8_y()
69 input += input_copy_size; in arm_concatenation_s8_y()
|
| D | arm_concatenation_s8_w.c | 49 void arm_concatenation_s8_w(const int8_t *input, in arm_concatenation_s8_w() argument
61 arm_memcpy_s8(output, input, input_copy_size); in arm_concatenation_s8_w()
|
| /cmsis-nn-3.6.0-3.5.0/Source/ConvolutionFunctions/ |
| D | arm_depthwise_conv_wrapper_s16.c | 54 const int16_t *input, in arm_depthwise_conv_wrapper_s16() argument
70 input, in arm_depthwise_conv_wrapper_s16()
84 input, in arm_depthwise_conv_wrapper_s16()
|
| D | arm_depthwise_conv_wrapper_s8.c | 53 const int8_t *input, in arm_depthwise_conv_wrapper_s8() argument
73 input, in arm_depthwise_conv_wrapper_s8()
88 input, in arm_depthwise_conv_wrapper_s8()
103 input, in arm_depthwise_conv_wrapper_s8()
|
| D | arm_depthwise_conv_s8.c | 47 depthwise_conv_s8_mult_4(const int8_t *input, in depthwise_conv_s8_mult_4() argument
108 int32_t in_val = input[in_idx + ker_w * input_ch] + input_offset; in depthwise_conv_s8_mult_4()
154 static void depthwise_conv_s8_generic(const int8_t *input, in depthwise_conv_s8_generic() argument
246 acc_0 += (input[idx_0] + input_offset) * kernel[ker_idx_0]; in depthwise_conv_s8_generic()
262 input += (input_x * input_y * input_ch); in depthwise_conv_s8_generic()
277 const int8_t *input, in arm_depthwise_conv_s8() argument
294 depthwise_conv_s8_mult_4(input, in arm_depthwise_conv_s8()
320 depthwise_conv_s8_generic(input, in arm_depthwise_conv_s8()
|
| D | arm_depthwise_conv_s16.c | 43 static void __attribute__((unused)) depthwise_conv_s16_mult_4_s16(const int16_t *input, in depthwise_conv_s16_mult_4_s16() argument
101 int32_t in_val = input[in_idx + ker_w * input_ch]; in depthwise_conv_s16_mult_4_s16()
133 static void depthwise_conv_s16_generic_s16(const int16_t *input, in depthwise_conv_s16_generic_s16() argument
220 acc_0 += input[idx_0] * kernel[ker_idx_0]; in depthwise_conv_s16_generic_s16()
234 input += (input_x * input_y * input_ch); in depthwise_conv_s16_generic_s16()
248 const int16_t *input, in arm_depthwise_conv_s16() argument
262 depthwise_conv_s16_generic_s16(input, in arm_depthwise_conv_s16()
|
| D | arm_depthwise_conv_fast_s16.c | 55 const int16_t *input, in arm_depthwise_conv_fast_s16() argument
125 … (int16_t *)(input + (i_ker_y * input_x + i_ker_x) * input_ch), in arm_depthwise_conv_fast_s16()
150 input += input_x * input_y * input_ch; in arm_depthwise_conv_fast_s16()
266 input + (idx_y * input_x + idx_x) * input_ch, in arm_depthwise_conv_fast_s16()
427 input += (input_x * input_y * input_ch); in arm_depthwise_conv_fast_s16()
436 input, in arm_depthwise_conv_fast_s16()
|
| /cmsis-nn-3.6.0-3.5.0/Source/FullyConnectedFunctions/ |
| D | arm_fully_connected_s16.c | 53 const int16_t *input, in arm_fully_connected_s16() argument
71 arm_nn_vec_mat_mult_t_s16(input, in arm_fully_connected_s16()
81 input += filter_dims->n; in arm_fully_connected_s16()
|
| D | arm_fully_connected_s8.c | 54 const int8_t *input, in arm_fully_connected_s8() argument
70 arm_nn_vec_mat_mult_t_s8(input, in arm_fully_connected_s8()
83 input += filter_dims->n; in arm_fully_connected_s8()
|
| /cmsis-nn-3.6.0-3.5.0/Source/NNSupportFunctions/ |
| D | arm_nn_lstm_step_s8_s16.c | 44 arm_cmsis_nn_status arm_nn_lstm_step_s8_s16(const int8_t *input, in arm_nn_lstm_step_s8_s16() argument
66 arm_nn_lstm_calculate_gate_s8_s16(input, in arm_nn_lstm_step_s8_s16()
82 arm_nn_lstm_calculate_gate_s8_s16(input, in arm_nn_lstm_step_s8_s16()
98 arm_nn_lstm_calculate_gate_s8_s16(input, in arm_nn_lstm_step_s8_s16()
122 arm_nn_lstm_calculate_gate_s8_s16(input, in arm_nn_lstm_step_s8_s16()
|
| D | arm_nn_lstm_calculate_gate_s8_s16.c | 55 void arm_nn_lstm_calculate_gate_s8_s16(const int8_t *input, in arm_nn_lstm_calculate_gate_s8_s16() argument
73 arm_nn_vec_mat_mul_result_acc_s8(input, in arm_nn_lstm_calculate_gate_s8_s16()
|
| D | arm_nn_vec_mat_mul_result_acc_s8.c | 84 const int8x16_t input = vldrbq_z_s8(lhs_vec, p); in arm_nn_vec_mat_mul_result_acc_s8() local
87 acc_0 = vmladavaq_p_s8(acc_0, ker_0, input, p); in arm_nn_vec_mat_mul_result_acc_s8()
90 acc_1 = vmladavaq_p_s8(acc_1, ker_1, input, p); in arm_nn_vec_mat_mul_result_acc_s8()
93 acc_2 = vmladavaq_p_s8(acc_2, ker_2, input, p); in arm_nn_vec_mat_mul_result_acc_s8()
96 acc_3 = vmladavaq_p_s8(acc_3, ker_3, input, p); in arm_nn_vec_mat_mul_result_acc_s8()
135 const int8x16_t input = vldrbq_z_s8(lhs_vec, p); in arm_nn_vec_mat_mul_result_acc_s8() local
138 acc_0 = vmladavaq_p_s8(acc_0, ker_0, input, p); in arm_nn_vec_mat_mul_result_acc_s8()
|
| D | arm_nn_vec_mat_mult_t_s8.c | 106 const int8x16_t input = vldrbq_z_s8(lhs_vec, p); in arm_nn_vec_mat_mult_t_s8() local
110 acc_0 = vmladavaq_s8(acc_0, ker_0, input); in arm_nn_vec_mat_mult_t_s8()
114 acc_1 = vmladavaq_s8(acc_1, ker_1, input); in arm_nn_vec_mat_mult_t_s8()
118 acc_2 = vmladavaq_s8(acc_2, ker_2, input); in arm_nn_vec_mat_mult_t_s8()
162 const int8x16_t input = vldrbq_z_s8(lhs_vec, p); in arm_nn_vec_mat_mult_t_s8() local
166 acc_0 = vmladavaq_s8(acc_0, ker_0, input); in arm_nn_vec_mat_mult_t_s8()
|
| /cmsis-nn-3.6.0-3.5.0/Include/ |
| D | arm_nnfunctions.h | 1335 void arm_nn_activation_s16(const int16_t *input,
1561 void arm_softmax_s8(const int8_t *input,
1583 void arm_softmax_s8_s16(const int8_t *input,
1612 arm_cmsis_nn_status arm_softmax_s16(const int16_t *input,
1635 void arm_softmax_u8(const uint8_t *input,
1657 void arm_reshape_s8(const int8_t *input, int8_t *output, const uint32_t total_size);
1702 void arm_concatenation_s8_x(const int8_t *input,
1749 void arm_concatenation_s8_y(const int8_t *input,
1796 void arm_concatenation_s8_z(const int8_t *input,
1839 void arm_concatenation_s8_w(const int8_t *input,
|
