Searched refs:inv_rows (Results 1 – 1 of 1) sorted by relevance
/external/tensorflow/tensorflow/core/kernels/ |
D | quantized_instance_norm.cc | 119 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()
|