Searched refs:im2col (Results 1 – 2 of 2) sorted by relevance
| /cmsis-nn-latest/Source/ConvolutionFunctions/ |
| D | arm_convolve_1_x_n_s8.c | 121 int8_t *im2col = ctx->buf; in arm_convolve_1_x_n_s8() local
123 …arm_memset_s8(im2col, (int8_t)-conv_params->input_offset, sizeof(int8_t) * (uint32_t)pad_size_left… in arm_convolve_1_x_n_s8()
124 im2col += pad_size_left; in arm_convolve_1_x_n_s8()
125 arm_memcpy_s8(im2col, input_data, sizeof(int8_t) * num_elem_left); in arm_convolve_1_x_n_s8()
184 im2col = ctx->buf; in arm_convolve_1_x_n_s8()
187 arm_memcpy_s8(im2col, input_data + input_start, sizeof(int8_t) * num_elem_right); in arm_convolve_1_x_n_s8()
188 im2col += num_elem_right; in arm_convolve_1_x_n_s8()
189 …arm_memset_s8(im2col, (int8_t)-conv_params->input_offset, sizeof(int8_t) * (uint32_t)pad_size_righ… in arm_convolve_1_x_n_s8()
|
| D | arm_convolve_s16.c | 99 int16_t *im2col = buffer_a; in arm_convolve_s16() local
122 … arm_memset_s8((int8_t *)im2col, 0, sizeof(int16_t) * (uint32_t)input_ch); in arm_convolve_s16()
126 arm_memcpy_s8((int8_t *)im2col, in arm_convolve_s16()
130 im2col += input_ch; in arm_convolve_s16()
153 im2col = buffer_a; in arm_convolve_s16()
171 im2col = buffer_a; in arm_convolve_s16()
200 im2col = buffer_a; in arm_convolve_s16()
|
