Home
last modified time | relevance | path

Searched refs:input (Results 1 – 25 of 40) sorted by relevance

12

/cmsis-nn-latest/Source/ActivationFunctions/
Darm_relu_q15.c57 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()
Darm_relu_q7.c57 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()
Darm_nn_activation_s16.c51 arm_cmsis_nn_status arm_nn_activation_s16(const int16_t *input, in arm_nn_activation_s16() argument
77 for (int i = 0; i < size; ++i, input++, output++) in arm_nn_activation_s16()
79 const int32_t input_data = ((*input) * input_multiplier + rounding) >> abs_left_shift; in arm_nn_activation_s16()
/cmsis-nn-latest/Source/ReshapeFunctions/
Darm_reshape_s8.c50 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-latest/Source/SoftmaxFunctions/
Darm_softmax_u8.c44 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()
Darm_nn_softmax_common_s8.c56 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()
Darm_softmax_s8_s16.c43 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()
Darm_softmax_s16.c39 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()
Darm_softmax_s8.c86 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-latest/Source/ConcatenationFunctions/
Darm_concatenation_s8_x.c49 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()
Darm_concatenation_s8_z.c49 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()
Darm_concatenation_s8_y.c49 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()
Darm_concatenation_s8_w.c49 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-latest/Source/ConvolutionFunctions/
Darm_depthwise_conv_wrapper_s16.c54 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()
Darm_depthwise_conv_wrapper_s8.c53 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()
Darm_depthwise_conv_wrapper_s4.c53 const int8_t *input, in arm_depthwise_conv_wrapper_s4() argument
69 input, in arm_depthwise_conv_wrapper_s4()
83 input, in arm_depthwise_conv_wrapper_s4()
Darm_depthwise_conv_s8.c47 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()
Darm_depthwise_conv_s4.c43 static void depthwise_conv_s4_generic(const int8_t *input, in depthwise_conv_s4_generic() argument
150 acc_0 += (input[idx_0] + input_offset) * ker_val0; in depthwise_conv_s4_generic()
151 acc_1 += (input[idx_0 + 1] + input_offset) * ker_val1; in depthwise_conv_s4_generic()
248 acc_0 += (input[idx_0] + input_offset) * ker_val0; in depthwise_conv_s4_generic()
335 acc_0 += (input[idx_0] + input_offset) * ker_val0; in depthwise_conv_s4_generic()
336 acc_1 += (input[idx_0] + input_offset) * ker_val1; in depthwise_conv_s4_generic()
364 input += (input_x * input_y * input_ch); in depthwise_conv_s4_generic()
454 acc_0 += (input[idx_0] + input_offset) * ker_val; in depthwise_conv_s4_generic()
480 input += (input_x * input_y * input_ch); in depthwise_conv_s4_generic()
496 const int8_t *input, in arm_depthwise_conv_s4() argument
[all …]
Darm_depthwise_conv_s16.c43 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()
/cmsis-nn-latest/Source/FullyConnectedFunctions/
Darm_fully_connected_s16.c53 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()
Darm_fully_connected_s4.c54 const int8_t *input, in arm_fully_connected_s4() argument
71 arm_nn_vec_mat_mult_t_s4(input, in arm_fully_connected_s4()
84 input += filter_dims->n; in arm_fully_connected_s4()
Darm_fully_connected_s8.c54 const int8_t *input, in arm_fully_connected_s8() argument
78 arm_nn_vec_mat_mult_t_s8(input, in arm_fully_connected_s8()
94 input += filter_dims->n; in arm_fully_connected_s8()
/cmsis-nn-latest/Source/LSTMFunctions/
Darm_lstm_unidirectional_s16.c49 arm_cmsis_nn_status arm_lstm_unidirectional_s16(const int16_t *input, in arm_lstm_unidirectional_s16() argument
62 const int16_t *data_in = input + (t * params->batch_size * params->input_size); in arm_lstm_unidirectional_s16()
78 const int16_t *data_in = input + (t * params->input_size); in arm_lstm_unidirectional_s16()
Darm_lstm_unidirectional_s8.c49 arm_cmsis_nn_status arm_lstm_unidirectional_s8(const int8_t *input, in arm_lstm_unidirectional_s8() argument
62 const int8_t *data_in = input + (t * params->batch_size * params->input_size); in arm_lstm_unidirectional_s8()
78 const int8_t *data_in = input + (t * params->input_size); in arm_lstm_unidirectional_s8()
/cmsis-nn-latest/Source/NNSupportFunctions/
Darm_nn_vec_mat_mul_result_acc_s8_s16.c88 const int8x16_t input = vldrbq_z_s8(lhs_vec, p); in arm_nn_vec_mat_mul_result_acc_s8_s16() local
91 acc_0 = vmladavaq_s8(acc_0, ker_0, input); in arm_nn_vec_mat_mul_result_acc_s8_s16()
94 acc_1 = vmladavaq_s8(acc_1, ker_1, input); in arm_nn_vec_mat_mul_result_acc_s8_s16()
97 acc_2 = vmladavaq_s8(acc_2, ker_2, input); in arm_nn_vec_mat_mul_result_acc_s8_s16()
100 acc_3 = vmladavaq_s8(acc_3, ker_3, input); in arm_nn_vec_mat_mul_result_acc_s8_s16()
134 const int8x16_t input = vldrbq_z_s8(lhs_vec, p); in arm_nn_vec_mat_mul_result_acc_s8_s16() local
137 acc_0 = vmladavaq_s8(acc_0, ker_0, input); in arm_nn_vec_mat_mul_result_acc_s8_s16()

12