Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/core/kernels/
Dquantized_instance_norm.cc60 uint32x4_t sub_sum[4] = {0}; in ColMeanAndVariance() local
79 sub_sum[0] = vaddw_u16(sub_sum[0], v_high_high); in ColMeanAndVariance()
80 sub_sum[1] = vaddw_u16(sub_sum[1], v_high_low); in ColMeanAndVariance()
81 sub_sum[2] = vaddw_u16(sub_sum[2], v_low_high); in ColMeanAndVariance()
82 sub_sum[3] = vaddw_u16(sub_sum[3], v_low_low); in ColMeanAndVariance()
92 sum[i] = vaddq_u32(sum[i], sub_sum[i]); in ColMeanAndVariance()
96 vmulq_n_f32(vcvtq_f32_u32(sub_sum[i]), 1.0f / nB); in ColMeanAndVariance()
104 const float32x4_t sub_sum_f32 = vcvtq_f32_u32(sub_sum[i]); in ColMeanAndVariance()