Searched refs:square_mean (Results 1 – 3 of 3) sorted by relevance
492 std::vector<float> square_mean(feature_bound); in XLA_TEST_P() local494 square_mean[i] = sum_squared[i] / num_elements_per_feature; in XLA_TEST_P()499 var[i] = square_mean[i] - mean_square[i]; in XLA_TEST_P()593 std::vector<float> square_mean(feature_bound); in XLA_TEST_P() local595 square_mean[i] = sum_squared[i] / num_elements_per_feature; in XLA_TEST_P()600 var[i] = square_mean[i] - mean_square[i]; in XLA_TEST_P()708 std::vector<float> square_mean(feature_bound); in XLA_TEST_P() local711 square_mean[i] = sum_squared[i] / num_elements_per_feature; in XLA_TEST_P()713 square_mean[i] = 0; in XLA_TEST_P()719 var[i] = square_mean[i] - mean_square[i]; in XLA_TEST_P()
430 square_mean = math_ops.matmul(mean, mean, transpose_a=True)431 new_cov = b[k, :, :] - square_mean + self._min_var
263 auto square_mean = add(Mean(elements_per_feature, squared_sum, add)); in HandleBatchNormTraining() local271 add_binary(feature_shape, HloOpcode::kSubtract, square_mean, mean_square); in HandleBatchNormTraining()