Home
last modified time | relevance | path

Searched refs:input_ch (Results 1 – 7 of 7) sorted by relevance

/cmsis-3.4.0/CMSIS/NN/Source/ConvolutionFunctions/
Darm_convolve_1_x_n_s8.c77 const uint16_t input_ch = input_dims->c; in arm_convolve_1_x_n_s8() local
110 (void)arm_nn_mat_mul_core_1x_s8((ker_end_idx[0] - ker_begin_idx[0]) * input_ch, in arm_convolve_1_x_n_s8()
111 input_data + input_begin_idx[0] * input_ch, in arm_convolve_1_x_n_s8()
112 filter_data + (input_ch * kernel_x * i_out_ch) + in arm_convolve_1_x_n_s8()
113 (ker_begin_idx[0] * input_ch), in arm_convolve_1_x_n_s8()
116 (void)arm_nn_mat_mul_core_1x_s8((ker_end_idx[1] - ker_begin_idx[1]) * input_ch, in arm_convolve_1_x_n_s8()
117 input_data + input_begin_idx[1] * input_ch, in arm_convolve_1_x_n_s8()
118 filter_data + (input_ch * kernel_x * i_out_ch) + in arm_convolve_1_x_n_s8()
119 (ker_begin_idx[1] * input_ch), in arm_convolve_1_x_n_s8()
123 (void)arm_nn_mat_mul_core_1x_s8((ker_end_idx[2] - ker_begin_idx[2]) * input_ch, in arm_convolve_1_x_n_s8()
[all …]
Darm_depthwise_conv_3x3_s8.c69 const int32_t input_ch = input_dims->c; in arm_depthwise_conv_3x3_s8() local
85 if (input_ch != output_ch) in arm_depthwise_conv_3x3_s8()
102 for (; in_ch <= (input_ch - 4); in_ch += 4) in arm_depthwise_conv_3x3_s8()
109 …const int8_t *input_ptr = input + (in_h + ker_h_start) * (input_ch * input_x) + in_w * input_ch + … in arm_depthwise_conv_3x3_s8()
110 const int8_t *kernel_ptr = kernel + ker_h_start * (input_ch * 3) + in_ch; in arm_depthwise_conv_3x3_s8()
128 in_val = arm_nn_read_q7x4(input_ptr + input_ch); in arm_depthwise_conv_3x3_s8()
129 ker_val = arm_nn_read_q7x4(kernel_ptr + input_ch); in arm_depthwise_conv_3x3_s8()
138 in_val = arm_nn_read_q7x4(input_ptr + (input_ch << 1)); in arm_depthwise_conv_3x3_s8()
139 ker_val = arm_nn_read_q7x4(kernel_ptr + (input_ch << 1)); in arm_depthwise_conv_3x3_s8()
147 input_ptr += (input_ch * input_x); in arm_depthwise_conv_3x3_s8()
[all …]
Darm_depthwise_conv_s8_opt.c64 const int32_t input_ch = input_dims->c; in arm_depthwise_conv_s8_opt() local
68 if (input_ch != output_ch) in arm_depthwise_conv_s8_opt()
111 arm_memset_q7(lhs_buffer, (int8_t)-input_offset, (uint32_t)input_ch); in arm_depthwise_conv_s8_opt()
116 … arm_memcpy_q7(lhs_buffer, input + (i_ker_y * input_x + i_ker_x) * input_ch, (uint32_t)input_ch); in arm_depthwise_conv_s8_opt()
118 lhs_buffer += input_ch; in arm_depthwise_conv_s8_opt()
131 input_ch, in arm_depthwise_conv_s8_opt()
146 input_ch, in arm_depthwise_conv_s8_opt()
167 int32_t loop_count = (input_ch + 3) / 4; in arm_depthwise_conv_s8_opt()
169 int32_t num_ch_to_process = input_ch; in arm_depthwise_conv_s8_opt()
172 const int8_t *col_0 = lhs_buffer + (kernel_size * input_ch * i_buf) + offset; in arm_depthwise_conv_s8_opt()
[all …]
Darm_convolve_s8.c69 const uint16_t input_ch = input_dims->c; in arm_convolve_s8() local
97 const int32_t num_elem = kernel_x * kernel_y * input_ch; in arm_convolve_s8()
112 memset(im2col_buf, (int8_t)-input_offset, sizeof(q7_t) * input_ch); in arm_convolve_s8()
117 … arm_memcpy_q7(im2col_buf, input_data + (i_ker_y * input_x + i_ker_x) * input_ch, input_ch); in arm_convolve_s8()
119 im2col_buf += input_ch; in arm_convolve_s8()
220 memset(two_column_buf, 0, sizeof(q15_t) * input_ch); in arm_convolve_s8()
225 … arm_q7_to_q15_with_offset(input_data + (i_ker_y * input_x + i_ker_x) * input_ch, in arm_convolve_s8()
227 input_ch, in arm_convolve_s8()
230 two_column_buf += input_ch; in arm_convolve_s8()
235 if (two_column_buf == buffer_a + 2 * input_ch * kernel_y * kernel_x) in arm_convolve_s8()
[all …]
Darm_convolve_1x1_s8_fast.c79 const int32_t input_ch = input_dims->c; in arm_convolve_1x1_s8_fast() local
94 (void)arm_nn_mat_mul_core_4x_s8(input_ch, in arm_convolve_1x1_s8_fast()
95 input_ch, in arm_convolve_1x1_s8_fast()
96 input_data + i_items * input_ch, in arm_convolve_1x1_s8_fast()
97 filter_data + i_out_ch * input_ch, in arm_convolve_1x1_s8_fast()
130 … input_ch, input_data + i_items * input_ch, filter_data + i_out_ch * input_ch, &sum_row, &acc); in arm_convolve_1x1_s8_fast()
Darm_depthwise_conv_s8.c46 const int32_t input_ch, in depthwise_conv_s8_mult_4() argument
86 … int32_t in_idx = (in_h + ker_h) * (input_ch * input_x) + in_w * input_ch + in_ch; in depthwise_conv_s8_mult_4()
91 int32_t in_val = input[in_idx + ker_w * input_ch] + input_offset; in depthwise_conv_s8_mult_4()
146 const uint16_t input_ch, in depthwise_conv_s8_generic() argument
179 for (int i_input_ch = 0; i_input_ch < input_ch; i_input_ch++) in depthwise_conv_s8_generic()
199 int32_t idx_0 = (idx_y * input_x + idx_x) * input_ch + i_input_ch; in depthwise_conv_s8_generic()
200 … int32_t ker_idx_0 = (i_ker_y * kernel_x + i_ker_x) * (input_ch * ch_mult) + idx_out_ch; in depthwise_conv_s8_generic()
218 input += (input_x * input_y * input_ch); in depthwise_conv_s8_generic()
/cmsis-3.4.0/CMSIS/NN/Include/
Darm_nnfunctions.h1822 const uint16_t input_ch,