Searched refs:channel_in (Results 1 – 2 of 2) sorted by relevance
| /cmsis-nn-latest/Source/PoolingFunctions/ |
| D | arm_max_pool_s16.c | 166 const int32_t channel_in = input_dims->c; in arm_max_pool_s16() local
167 const int32_t batch_size = input_x * input_y * channel_in; in arm_max_pool_s16()
198 … const int16_t *start = src + channel_in * (k_x + base_idx_x + (k_y + base_idx_y) * input_x); in arm_max_pool_s16()
202 memcpy(dst, start, channel_in * sizeof(int16_t)); in arm_max_pool_s16()
207 compare_and_replace_if_larger(dst, start, channel_in); in arm_max_pool_s16()
212 dst += channel_in; in arm_max_pool_s16()
216 clamp_output(dst_base, output_x * output_y * channel_in, act_min, act_max); in arm_max_pool_s16()
|
| D | arm_max_pool_s8.c | 183 const int32_t channel_in = input_dims->c; in arm_max_pool_s8() local
184 const int32_t batch_size = input_x * input_y * channel_in; in arm_max_pool_s8()
214 … const int8_t *start = src + channel_in * (k_x + base_idx_x + (k_y + base_idx_y) * input_x); in arm_max_pool_s8()
218 arm_memcpy_s8(dst, start, channel_in); in arm_max_pool_s8()
223 compare_and_replace_if_larger_q7(dst, start, channel_in); in arm_max_pool_s8()
228 dst += channel_in; in arm_max_pool_s8()
232 clamp_output(dst_base, output_x * output_y * channel_in, act_min, act_max); in arm_max_pool_s8()
|
