Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/core/kernels/
Dquantized_instance_norm.cc119 const float inv_rows = 1.0f / static_cast<float>(rows); in ColMeanAndVariance() local
120 vst1q_f32(mean + col_offset, vmulq_n_f32(vcvtq_f32_u32(sum[3]), inv_rows)); in ColMeanAndVariance()
122 vmulq_n_f32(vcvtq_f32_u32(sum[2]), inv_rows)); in ColMeanAndVariance()
124 vmulq_n_f32(vcvtq_f32_u32(sum[1]), inv_rows)); in ColMeanAndVariance()
126 vmulq_n_f32(vcvtq_f32_u32(sum[0]), inv_rows)); in ColMeanAndVariance()
128 vst1q_f32(variance + col_offset, vmulq_n_f32(M2A[3], inv_rows)); in ColMeanAndVariance()
129 vst1q_f32(variance + col_offset + 4, vmulq_n_f32(M2A[2], inv_rows)); in ColMeanAndVariance()
130 vst1q_f32(variance + col_offset + 8, vmulq_n_f32(M2A[1], inv_rows)); in ColMeanAndVariance()
131 vst1q_f32(variance + col_offset + 12, vmulq_n_f32(M2A[0], inv_rows)); in ColMeanAndVariance()