Home
last modified time | relevance | path

Searched refs:sum_2 (Results 1 – 6 of 6) sorted by relevance

/cmsis-nn-3.7.0/Source/PoolingFunctions/
Darm_avgpool_s16.c130 int32x4_t sum_2 = vdupq_n_s32(0); in arm_avgpool_s16() local
146 sum_2 = vaddq_s32(sum_2, temp_hi); in arm_avgpool_s16()
166 p = vcmpgtq_n_s32(sum_2, 0); in arm_avgpool_s16()
167 sum_2 = vaddq_m_n_s32(sum_2, sum_2, half_count, p); in arm_avgpool_s16()
168 sum_2 = vsubq_m_n_s32(sum_2, sum_2, half_count, ~p); in arm_avgpool_s16()
173 sum_2[i] = sum_2[i] / count; in arm_avgpool_s16()
179 sum_2 = vmaxq_s32(sum_2, vdupq_n_s32(act_min)); in arm_avgpool_s16()
180 sum_2 = vminq_s32(sum_2, vdupq_n_s32(act_max)); in arm_avgpool_s16()
184 temp = vmovntq_s32(temp, sum_2); in arm_avgpool_s16()
Darm_avgpool_s8.c131 int32x4_t sum_2 = vdupq_n_s32(0); in arm_avgpool_s8() local
155 sum_2 = vaddq_s32(sum_2, temp_lo_hi); in arm_avgpool_s8()
177 p = vcmpgtq_n_s32(sum_2, 0); in arm_avgpool_s8()
178 sum_2 = vaddq_m_n_s32(sum_2, sum_2, half_count, p); in arm_avgpool_s8()
179 sum_2 = vsubq_m_n_s32(sum_2, sum_2, half_count, ~p); in arm_avgpool_s8()
192 sum_2[i] = sum_2[i] / count; in arm_avgpool_s8()
200 sum_2 = vmaxq_s32(sum_2, vdupq_n_s32(act_min)); in arm_avgpool_s8()
201 sum_2 = vminq_s32(sum_2, vdupq_n_s32(act_max)); in arm_avgpool_s8()
210 temp_lo = vmovntq_s32(temp_lo, sum_2); in arm_avgpool_s8()
/cmsis-nn-3.7.0/Source/ConvolutionFunctions/
Darm_depthwise_conv_s4_opt.c324 int32_t sum_2 = 0; in arm_depthwise_conv_s4_opt() local
330 sum_2 = *bias++; in arm_depthwise_conv_s4_opt()
364 sum_2 = SMLAD(op_a, op_b, sum_2); in arm_depthwise_conv_s4_opt()
388 sum_2 += ker1 * col_pos[1]; in arm_depthwise_conv_s4_opt()
399 sum_2 += ker1 * col_pos[1 + input_ch]; in arm_depthwise_conv_s4_opt()
426 sum_2 += ker1 * col_pos[1]; in arm_depthwise_conv_s4_opt()
442 sum_2 = arm_nn_requantize(sum_2, *output_mult++, *output_shift++); in arm_depthwise_conv_s4_opt()
443 sum_2 += output_offset; in arm_depthwise_conv_s4_opt()
444 sum_2 = MAX(sum_2, output_activation_min); in arm_depthwise_conv_s4_opt()
445 sum_2 = MIN(sum_2, output_activation_max); in arm_depthwise_conv_s4_opt()
[all …]
Darm_depthwise_conv_s8_opt.c277 int32_t sum_2 = 0; in arm_depthwise_conv_s8_opt() local
283 sum_2 = *bias++; in arm_depthwise_conv_s8_opt()
317 sum_2 = SMLAD(op_a, op_b, sum_2); in arm_depthwise_conv_s8_opt()
339 sum_2 += row_pos[1] * col_pos[1]; in arm_depthwise_conv_s8_opt()
354 sum_2 = arm_nn_requantize(sum_2, *output_mult++, *output_shift++); in arm_depthwise_conv_s8_opt()
355 sum_2 += output_offset; in arm_depthwise_conv_s8_opt()
356 sum_2 = MAX(sum_2, output_activation_min); in arm_depthwise_conv_s8_opt()
357 sum_2 = MIN(sum_2, output_activation_max); in arm_depthwise_conv_s8_opt()
358 *output++ = (int8_t)sum_2; in arm_depthwise_conv_s8_opt()
Darm_depthwise_conv_fast_s16.c288 int32_t sum_2 = 0; in arm_depthwise_conv_fast_s16() local
327 sum_2 = SMLAD(col_a, row_c, sum_2); in arm_depthwise_conv_fast_s16()
349 sum_2 += row_pos[1] * col_pos[1]; in arm_depthwise_conv_fast_s16()
360 int64_t acc_2 = sum_2; in arm_depthwise_conv_fast_s16()
/cmsis-nn-3.7.0/Source/BasicMathFunctions/
Darm_elementwise_add_s16.c111 int16_t sum_1, sum_2; in arm_elementwise_add_s16() local
136 sum_2 = (int16_t)sum; in arm_elementwise_add_s16()
138 arm_nn_write_q15x2_ia(&output, PACK_Q15x2_32x1(sum_1, sum_2)); in arm_elementwise_add_s16()