Searched refs:act_max (Results 1 – 4 of 4) sorted by relevance
| /cmsis-nn-latest/Source/PoolingFunctions/ |
| D | arm_max_pool_s8.c | 97 …tic void clamp_output(int8_t *source, int32_t length, const int32_t act_min, const int32_t act_max) in clamp_output() argument
102 const int8x16_t vmax = vdupq_n_s8((int8_t)act_max); in clamp_output()
123 in.bytes[0] = MIN(in.bytes[0], act_max); in clamp_output()
125 in.bytes[1] = MIN(in.bytes[1], act_max); in clamp_output()
127 in.bytes[2] = MIN(in.bytes[2], act_max); in clamp_output()
129 in.bytes[3] = MIN(in.bytes[3], act_max); in clamp_output()
140 comp = MIN(comp, act_max); in clamp_output()
182 const int32_t act_max = pool_params->activation.max; in arm_max_pool_s8() local
232 clamp_output(dst_base, output_x * output_y * channel_in, act_min, act_max); in arm_max_pool_s8()
|
| D | arm_max_pool_s16.c | 85 …ic void clamp_output(int16_t *source, int32_t length, const int16_t act_min, const int16_t act_max) in clamp_output() argument
89 const int16x8_t max = vdupq_n_s16((int16_t)act_max); in clamp_output()
111 in.half_words[0] = MIN(in.half_words[0], act_max); in clamp_output()
113 in.half_words[1] = MIN(in.half_words[1], act_max); in clamp_output()
123 comp = MIN(comp, act_max); in clamp_output()
165 const int16_t act_max = pool_params->activation.max; in arm_max_pool_s16() local
216 clamp_output(dst_base, output_x * output_y * channel_in, act_min, act_max); in arm_max_pool_s16()
|
| D | arm_avgpool_s8.c | 40 const int act_max) in scale_q31_to_q7_and_clamp() argument
49 sum = MIN(sum, act_max); in scale_q31_to_q7_and_clamp()
94 const int32_t act_max = pool_params->activation.max; in arm_avgpool_s8() local
198 sum_1 = vminq_s32(sum_1, vdupq_n_s32(act_max)); in arm_avgpool_s8()
201 sum_2 = vminq_s32(sum_2, vdupq_n_s32(act_max)); in arm_avgpool_s8()
204 sum_3 = vminq_s32(sum_3, vdupq_n_s32(act_max)); in arm_avgpool_s8()
207 sum_4 = vminq_s32(sum_4, vdupq_n_s32(act_max)); in arm_avgpool_s8()
255 const int32_t act_max = pool_params->activation.max; in arm_avgpool_s8() local
323 scale_q31_to_q7_and_clamp(buffer, dst, ch_src, count, act_min, act_max); in arm_avgpool_s8()
369 sum = MIN(sum, act_max); in arm_avgpool_s8()
|
| D | arm_avgpool_s16.c | 41 const int act_max) in scale_q31_to_q15_and_clamp() argument
50 sum = MIN(sum, act_max); in scale_q31_to_q15_and_clamp()
92 const int32_t act_max = pool_params->activation.max; in arm_avgpool_s16() local
177 sum_1 = vminq_s32(sum_1, vdupq_n_s32(act_max)); in arm_avgpool_s16()
180 sum_2 = vminq_s32(sum_2, vdupq_n_s32(act_max)); in arm_avgpool_s16()
259 scale_q31_to_q15_and_clamp(buffer, dst, ch_src, count, act_min, act_max); in arm_avgpool_s16()
310 sum = MIN(sum, act_max); in arm_avgpool_s16()
|
