| /cmsis-3.4.0/CMSIS/NN/Source/ConcatenationFunctions/ |
| D | arm_concatenation_s8_z.c | 49 const uint16_t input_x, in arm_concatenation_s8_z() argument
57 const uint32_t input_copy_size = input_x * input_y * input_z; in arm_concatenation_s8_z()
58 const uint32_t output_stride = input_x * input_y * output_z; in arm_concatenation_s8_z()
60 output += offset_z * (input_x * input_y); in arm_concatenation_s8_z()
|
| D | arm_concatenation_s8_y.c | 49 const uint16_t input_x, in arm_concatenation_s8_y() argument
58 const uint32_t input_copy_size = input_x * input_y; in arm_concatenation_s8_y()
59 const uint32_t output_stride = input_x * output_y; in arm_concatenation_s8_y()
61 output += offset_y * input_x; in arm_concatenation_s8_y()
|
| D | arm_concatenation_s8_w.c | 49 const uint16_t input_x, in arm_concatenation_s8_w() argument
56 const uint32_t input_copy_size = input_x * input_y * input_z * input_w; in arm_concatenation_s8_w()
58 output += offset_w * (input_x * input_y * input_z); in arm_concatenation_s8_w()
|
| D | arm_concatenation_s8_x.c | 49 const uint16_t input_x, in arm_concatenation_s8_x() argument
66 memcpy(output, input, input_x); in arm_concatenation_s8_x()
67 input += input_x; in arm_concatenation_s8_x()
|
| /cmsis-3.4.0/CMSIS/NN/Source/ConvolutionFunctions/ |
| D | arm_depthwise_conv_3x3_s8.c | 67 const int32_t input_x = input_dims->w; in arm_depthwise_conv_3x3_s8() local
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()
136 if ((input_x - in_w) >= 3) in arm_depthwise_conv_3x3_s8()
147 input_ptr += (input_ch * input_x); in arm_depthwise_conv_3x3_s8()
177 …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()
189 if ((input_x - in_w) >= 3) in arm_depthwise_conv_3x3_s8()
194 input_ptr += (input_ch * input_x); in arm_depthwise_conv_3x3_s8()
|
| D | arm_convolve_s8.c | 67 const uint16_t input_x = input_dims->w; in arm_convolve_s8() local
110 if (i_ker_y < 0 || i_ker_y >= input_y || i_ker_x < 0 || i_ker_x >= input_x) 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()
217 if (i_ker_y < 0 || i_ker_y >= input_y || i_ker_x < 0 || i_ker_x >= input_x) 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()
328 const int32_t ker_x_end = MIN(kernel_x, input_x - base_idx_x); in arm_convolve_s8()
339 … (input_data[(in_row * input_x + in_col) * input_ch + i_input_ch] + input_offset) * in arm_convolve_s8()
359 input_data += (input_x * input_y * input_ch); in arm_convolve_s8()
|
| D | arm_depthwise_conv_s8.c | 44 const int32_t input_x, 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()
88 for (int32_t ker_w = ker_w_start; ker_w < MIN(kernel_x, input_x - in_w); in depthwise_conv_s8_mult_4()
144 const uint16_t input_x, in depthwise_conv_s8_generic() argument
190 const int ker_x_end = MIN(kernel_x, input_x - base_idx_x); 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()
218 input += (input_x * input_y * input_ch); in depthwise_conv_s8_generic()
|
| D | arm_depthwise_conv_s8_opt.c | 73 const int32_t input_x = input_dims->w; in arm_depthwise_conv_s8_opt() local
109 if (i_ker_y < 0 || i_ker_y >= input_y || i_ker_x < 0 || i_ker_x >= input_x) 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()
246 if (idx_x < 0 || idx_x >= input_x) in arm_depthwise_conv_s8_opt()
252 … arm_q7_to_q15_with_offset((q7_t *)input + (idx_y * input_x + idx_x) * input_ch, in arm_depthwise_conv_s8_opt()
|
| D | arm_convolve_1_x_n_s8.c | 73 const uint16_t input_x = input_dims->w; in arm_convolve_1_x_n_s8() local
99 ker_end_idx[i] = MIN(kernel_x, input_x - est_input_x_idx); in arm_convolve_1_x_n_s8()
|
| /cmsis-3.4.0/CMSIS/NN/Source/PoolingFunctions/ |
| D | arm_avgpool_s8.c | 84 const int32_t input_x = input_dims->w; in arm_avgpool_s8() local
115 k_x_end = MIN(i_x * stride_x - pad_x + kernel_x, input_x); in arm_avgpool_s8()
139 pTmpInner = pTmp + (ch_src * (k_x + k_y * input_x)); in arm_avgpool_s8()
224 sum += pTmp[ch_src * (k_x + k_y * input_x)]; in arm_avgpool_s8()
252 const int32_t input_x = input_dims->w; in arm_avgpool_s8() local
282 const int32_t kernel_x_end = MIN(kernel_x, input_x - idx_x); in arm_avgpool_s8()
290 const q7_t *start = src + ch_src * (k_x + idx_x + (k_y + idx_y) * input_x); in arm_avgpool_s8()
340 if (k_y >= 0 && k_x >= 0 && k_y < input_y && k_x < input_x) in arm_avgpool_s8()
342 sum += src[i_ch_in + ch_src * (k_x + k_y * input_x)]; in arm_avgpool_s8()
|
| D | arm_max_pool_s8.c | 171 const int32_t input_x = input_dims->w; in arm_max_pool_s8() local
196 const int32_t kernel_x_end = MIN(kernel_x, input_x - base_idx_x); in arm_max_pool_s8()
204 … const q7_t *start = src + channel_in * (k_x + base_idx_x + (k_y + base_idx_y) * input_x); in arm_max_pool_s8()
|
| /cmsis-3.4.0/CMSIS/NN/Include/ |
| D | arm_nnfunctions.h | 1820 const uint16_t input_x,
1903 const uint16_t input_x,
1948 const uint16_t input_x,
1993 const uint16_t input_x,
2034 const uint16_t input_x,
|
