Searched refs:kernel (Results 1 – 7 of 7) sorted by relevance
| /cmsis-nn-3.6.0-3.5.0/Source/FullyConnectedFunctions/ |
| D | arm_fully_connected_s16.c | 55 const int8_t *kernel, in arm_fully_connected_s16() argument
72 kernel, in arm_fully_connected_s16()
|
| D | arm_fully_connected_s8.c | 56 const int8_t *kernel, in arm_fully_connected_s8() argument
71 kernel, in arm_fully_connected_s8()
|
| /cmsis-nn-3.6.0-3.5.0/Source/ConvolutionFunctions/ |
| D | arm_depthwise_conv_s8.c | 51 const int8_t *kernel, in depthwise_conv_s8_mult_4() argument
74 const int8_t *kernel_base = kernel; in depthwise_conv_s8_mult_4()
100 kernel = kernel_base + mult_tile + ker_idx; in depthwise_conv_s8_mult_4()
106 ++ker_w, kernel += output_ch) in depthwise_conv_s8_mult_4()
109 out_buff[0] += in_val * kernel[0]; in depthwise_conv_s8_mult_4()
110 out_buff[1] += in_val * kernel[1]; in depthwise_conv_s8_mult_4()
111 out_buff[2] += in_val * kernel[2]; in depthwise_conv_s8_mult_4()
112 out_buff[3] += in_val * kernel[3]; in depthwise_conv_s8_mult_4()
159 const int8_t *kernel, in depthwise_conv_s8_generic() argument
246 acc_0 += (input[idx_0] + input_offset) * kernel[ker_idx_0]; in depthwise_conv_s8_generic()
[all …]
|
| D | arm_depthwise_conv_s16.c | 47 const int8_t *kernel, in depthwise_conv_s16_mult_4_s16() argument
102 out_buff[0] += in_val * kernel[ker_idx + 0 + mult_tile]; in depthwise_conv_s16_mult_4_s16()
103 out_buff[1] += in_val * kernel[ker_idx + 1 + mult_tile]; in depthwise_conv_s16_mult_4_s16()
104 out_buff[2] += in_val * kernel[ker_idx + 2 + mult_tile]; in depthwise_conv_s16_mult_4_s16()
105 out_buff[3] += in_val * kernel[ker_idx + 3 + mult_tile]; in depthwise_conv_s16_mult_4_s16()
138 const int8_t *kernel, in depthwise_conv_s16_generic_s16() argument
220 acc_0 += input[idx_0] * kernel[ker_idx_0]; in depthwise_conv_s16_generic_s16()
250 const int8_t *kernel, in arm_depthwise_conv_s16() argument
267 kernel, in arm_depthwise_conv_s16()
|
| D | arm_depthwise_conv_s8_opt.c | 57 const int8_t *kernel, in arm_depthwise_conv_s8_opt() argument
142 kernel + block_offset, in arm_depthwise_conv_s8_opt()
173 const int8_t *row_0 = kernel + offset; in arm_depthwise_conv_s8_opt()
290 const int8_t *row_pos = kernel + row_shift; in arm_depthwise_conv_s8_opt()
378 const int8_t *row_pos = kernel + row_shift; in arm_depthwise_conv_s8_opt()
413 kernel, in arm_depthwise_conv_s8_opt()
|
| D | arm_depthwise_conv_fast_s16.c | 57 const int8_t *kernel, in arm_depthwise_conv_fast_s16() argument
137 kernel, in arm_depthwise_conv_fast_s16()
162 const int8_t *row_0 = kernel + offset; in arm_depthwise_conv_fast_s16()
300 const int8_t *row_pos = kernel + row_shift; in arm_depthwise_conv_fast_s16()
399 const int8_t *row_pos = kernel + row_shift; in arm_depthwise_conv_fast_s16()
438 kernel, in arm_depthwise_conv_fast_s16()
|
| D | arm_depthwise_conv_3x3_s8.c | 58 const int8_t *kernel, in arm_depthwise_conv_3x3_s8() argument
118 const int8_t *kernel_ptr = kernel + ker_h_start * (input_ch * 3) + in_ch; in arm_depthwise_conv_3x3_s8()
248 const int8_t *kernel_ptr = kernel + ker_h_start * (input_ch * 3) + in_ch; in arm_depthwise_conv_3x3_s8()
|
