Searched refs:sum_1 (Results 1 – 5 of 5) sorted by relevance
/cmsis-nn-latest/Source/PoolingFunctions/ |
D | arm_avgpool_s16.c | 129 int32x4_t sum_1 = vdupq_n_s32(0); in arm_avgpool_s16() local 145 sum_1 = vaddq_s32(sum_1, temp_lo); in arm_avgpool_s16() 162 mve_pred16_t p = vcmpgtq_n_s32(sum_1, 0); in arm_avgpool_s16() 163 sum_1 = vaddq_m_n_s32(sum_1, sum_1, half_count, p); in arm_avgpool_s16() 164 sum_1 = vsubq_m_n_s32(sum_1, sum_1, half_count, ~p); in arm_avgpool_s16() 172 sum_1[i] = sum_1[i] / count; in arm_avgpool_s16() 176 sum_1 = vmaxq_s32(sum_1, vdupq_n_s32(act_min)); in arm_avgpool_s16() 177 sum_1 = vminq_s32(sum_1, vdupq_n_s32(act_max)); in arm_avgpool_s16() 183 temp = vmovnbq_s32(temp, sum_1); in arm_avgpool_s16()
|
D | arm_avgpool_s8.c | 130 int32x4_t sum_1 = vdupq_n_s32(0); in arm_avgpool_s8() local 154 sum_1 = vaddq_s32(sum_1, temp_lo_lo); in arm_avgpool_s8() 173 mve_pred16_t p = vcmpgtq_n_s32(sum_1, 0); in arm_avgpool_s8() 174 sum_1 = vaddq_m_n_s32(sum_1, sum_1, half_count, p); in arm_avgpool_s8() 175 sum_1 = vsubq_m_n_s32(sum_1, sum_1, half_count, ~p); in arm_avgpool_s8() 191 sum_1[i] = sum_1[i] / count; in arm_avgpool_s8() 197 sum_1 = vmaxq_s32(sum_1, vdupq_n_s32(act_min)); in arm_avgpool_s8() 198 sum_1 = vminq_s32(sum_1, vdupq_n_s32(act_max)); in arm_avgpool_s8() 209 temp_lo = vmovnbq_s32(temp_lo, sum_1); in arm_avgpool_s8()
|
/cmsis-nn-latest/Source/BasicMathFunctions/ |
D | arm_elementwise_add_s16.c | 111 int16_t sum_1, sum_2; in arm_elementwise_add_s16() local 126 sum_1 = (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()
|
/cmsis-nn-latest/Source/ConvolutionFunctions/ |
D | arm_nn_depthwise_conv_s8_core.c | 170 int32_t sum_1 = bias[i]; in arm_nn_depthwise_conv_s8_core() local 176 sum_1 += row_val * col_pos_1[j * num_ch]; in arm_nn_depthwise_conv_s8_core() 179 col_1_sum[i] = sum_1; in arm_nn_depthwise_conv_s8_core()
|
D | arm_depthwise_conv_fast_s16.c | 287 int32_t sum_1 = 0; in arm_depthwise_conv_fast_s16() local 324 sum_1 = SMLAD(col_c, row_c, sum_1); in arm_depthwise_conv_fast_s16() 348 sum_1 += row_pos[0] * col_pos[0]; in arm_depthwise_conv_fast_s16() 359 int64_t acc_1 = sum_1; in arm_depthwise_conv_fast_s16()
|