Searched refs:im2col_buf (Results 1 – 2 of 2) sorted by relevance
| /cmsis-nn-3.7.0/Source/ConvolutionFunctions/ |
| D | arm_convolve_s8.c | 113 int8_t *im2col_buf = (int8_t *)buffer_a; in arm_convolve_s8() local
116 int8_t *im2col_buf = (int8_t *)buffer_a + aligned_rhs_cols * 2; in arm_convolve_s8()
146 … arm_memset_s8(im2col_buf, (int8_t)-input_offset, sizeof(int8_t) * kernel_ch); in arm_convolve_s8()
150 arm_memcpy_s8(im2col_buf, in arm_convolve_s8()
154 im2col_buf += kernel_ch; in arm_convolve_s8()
160 im2col_buf += aligned_rhs_cols_offset; in arm_convolve_s8()
184 im2col_buf = (int8_t *)buffer_a; in arm_convolve_s8()
190 … im2col_buf - rhs_cols, im2col_buf_start_s16, rhs_cols, (int16_t)input_offset); in arm_convolve_s8()
194 … im2col_buf - rhs_cols, im2col_buf_start_s16, rhs_cols, (int16_t)input_offset); in arm_convolve_s8()
235 im2col_buf = (int8_t *)buffer_a + aligned_rhs_cols * 2; in arm_convolve_s8()
[all …]
|
| D | arm_convolve_s4.c | 100 int8_t *im2col_buf = (int8_t *)buffer_a; in arm_convolve_s4() local
122 … arm_memset_s8(im2col_buf, (int8_t)-input_offset, sizeof(int8_t) * input_ch); in arm_convolve_s4()
126 … arm_memcpy_s8(im2col_buf, input_data + (k_y * input_x + k_x) * input_ch, input_ch); in arm_convolve_s4()
128 im2col_buf += input_ch; in arm_convolve_s4()
153 im2col_buf = (int8_t *)buffer_a; in arm_convolve_s4()
182 im2col_buf = (int8_t *)buffer_a; in arm_convolve_s4()
|
