Searched refs:i_y (Results 1 – 4 of 4) sorted by relevance
| /cmsis-nn-latest/Source/PoolingFunctions/ |
| D | arm_avgpool_s16.c | 109 for (int i_y = 0; i_y < output_y; i_y++) in arm_avgpool_s16() local
113 const int32_t k_y_start = MAX(0, i_y * stride_y - pad_y); in arm_avgpool_s16()
114 const int32_t k_y_end = MIN(i_y * stride_y - pad_y + kernel_y, input_y); in arm_avgpool_s16()
120 int16_t *out = &dst[ch_src * (i_x + i_y * output_x)]; in arm_avgpool_s16()
213 for (int i_y = 0, idx_y = -pad_y; i_y < output_y; idx_y += stride_y, i_y++) in arm_avgpool_s16() local
276 for (int i_y = 0, base_idx_y = -pad_y; i_y < output_y; base_idx_y += stride_y, i_y++) in arm_avgpool_s16() local
312 dst[i_ch_in + ch_src * (i_x + i_y * output_x)] = sum; in arm_avgpool_s16()
|
| D | arm_avgpool_s8.c | 107 for (int i_y = 0; i_y < output_y; i_y++) in arm_avgpool_s8() local
111 const int32_t k_y_start = MAX(0, i_y * stride_y - pad_y); in arm_avgpool_s8()
112 const int32_t k_y_end = MIN(i_y * stride_y - pad_y + kernel_y, input_y); in arm_avgpool_s8()
118 int8_t *out = &dst[ch_src * (i_x + i_y * output_x)]; in arm_avgpool_s8()
277 for (int i_y = 0, idx_y = -pad_y; i_y < output_y; idx_y += stride_y, i_y++) in arm_avgpool_s8() local
341 for (int i_y = 0; i_y < output_y; i_y++) in arm_avgpool_s8() local
349 … for (int k_y = i_y * stride_y - pad_y; k_y < i_y * stride_y - pad_y + kernel_y; k_y++) in arm_avgpool_s8()
371 dst[i_ch_in + ch_src * (i_x + i_y * output_x)] = sum; in arm_avgpool_s8()
|
| D | arm_max_pool_s16.c | 180 for (int i_y = 0, base_idx_y = -pad_y; i_y < output_y; base_idx_y += stride_y, i_y++) in arm_max_pool_s16() local
|
| D | arm_max_pool_s8.c | 196 for (int i_y = 0, base_idx_y = -pad_y; i_y < output_y; base_idx_y += stride_y, i_y++) in arm_max_pool_s8() local
|
