Home
last modified time | relevance | path

Searched refs:acc_reg (Results 1 – 2 of 2) sorted by relevance

/external/tensorflow/tensorflow/lite/kernels/internal/optimized/integer_ops/
Dpooling.h97 int8x16_t acc_reg = vld1q_s8(acc + channel); in MaxPool() local
100 acc_reg = vmaxq_s8(acc_reg, input_reg); in MaxPool()
101 vst1q_s8(acc + channel, acc_reg); in MaxPool()
105 int8x8_t acc_reg = vld1_s8(acc + channel); in MaxPool() local
108 acc_reg = vmax_s8(acc_reg, input_reg); in MaxPool()
109 vst1_s8(acc + channel, acc_reg); in MaxPool()
208 int16x4_t acc_reg[4]; in AveragePool() local
211 acc_reg[0] = vget_low_s16(vmovl_s8(vget_low_s8(input_reg))); in AveragePool()
212 acc_reg[1] = vget_high_s16(vmovl_s8(vget_low_s8(input_reg))); in AveragePool()
213 acc_reg[2] = vget_low_s16(vmovl_s8(vget_high_s8(input_reg))); in AveragePool()
[all …]
/external/tensorflow/tensorflow/lite/kernels/internal/optimized/
Doptimized_ops.h3297 uint16x4_t acc_reg[4]; in AveragePool() local
3300 acc_reg[0] = vget_low_u16(vmovl_u8(vget_low_u8(input_reg))); in AveragePool()
3301 acc_reg[1] = vget_high_u16(vmovl_u8(vget_low_u8(input_reg))); in AveragePool()
3302 acc_reg[2] = vget_low_u16(vmovl_u8(vget_high_u8(input_reg))); in AveragePool()
3303 acc_reg[3] = vget_high_u16(vmovl_u8(vget_high_u8(input_reg))); in AveragePool()
3307 vaddw_u16(vld1q_u32(acc + channel + 4 * i), acc_reg[i])); in AveragePool()
3311 uint16x4_t acc_reg[2]; in AveragePool() local
3314 acc_reg[0] = vget_low_u16(input_reg); in AveragePool()
3315 acc_reg[1] = vget_high_u16(input_reg); in AveragePool()
3319 vaddw_u16(vld1q_u32(acc + channel + 4 * i), acc_reg[i])); in AveragePool()
[all …]