Searched refs:squared_sum (Results 1 – 5 of 5) sorted by relevance
90 float squared_sum = GetSum(accum); in ComputeStdDevNeon() local94 squared_sum += Square(values[offset] - mean); in ComputeStdDevNeon()97 const float std_dev_neon = sqrt(squared_sum / static_cast<float>(num_vals)); in ComputeStdDevNeon()
269 float squared_sum = 0.0f; in ComputeStdDevCpu() local271 squared_sum += Square(values[i] - mean); in ComputeStdDevCpu()273 return sqrt(squared_sum / static_cast<float>(num_vals)); in ComputeStdDevCpu()
87 double squared_sum = 0.0; in MseInputOutput() local89 squared_sum += (input[n] - output[channels * n + delay]) * in MseInputOutput()92 return squared_sum / (num_samples - delay); in MseInputOutput()
65 HighPrecisionValueType squared_sum() const { return squared_sum_; } in squared_sum() function
252 auto squared_sum = add(HloInstruction::CreateReduce( in HandleBatchNormTraining() local263 auto square_mean = add(Mean(elements_per_feature, squared_sum, add)); in HandleBatchNormTraining()