HomeSort by relevance Sort by last modified time
    Searched refs:square_mean (Results 1 - 3 of 3) sorted by null

  /external/tensorflow/tensorflow/compiler/xla/tests/
batch_normalization_test.cc 492 std::vector<float> square_mean(feature_bound);
494 square_mean[i] = sum_squared[i] / num_elements_per_feature;
499 var[i] = square_mean[i] - mean_square[i];
593 std::vector<float> square_mean(feature_bound);
595 square_mean[i] = sum_squared[i] / num_elements_per_feature;
600 var[i] = square_mean[i] - mean_square[i];
708 std::vector<float> square_mean(feature_bound);
711 square_mean[i] = sum_squared[i] / num_elements_per_feature;
713 square_mean[i] = 0;
719 var[i] = square_mean[i] - mean_square[i]
    [all...]
  /external/tensorflow/tensorflow/compiler/xla/service/
batchnorm_expander.cc 263 auto square_mean = add(Mean(elements_per_feature, squared_sum, add)); local
271 add_binary(feature_shape, HloOpcode::kSubtract, square_mean, mean_square);
  /external/tensorflow/tensorflow/contrib/factorization/python/ops/
gmm_ops.py 430 square_mean = math_ops.matmul(mean, mean, transpose_a=True)
431 new_cov = b[k, :, :] - square_mean + self._min_var

Completed in 181 milliseconds