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

  /external/tensorflow/tensorflow/compiler/xla/tests/
batch_normalization_test.cc 483 std::vector<float> square_mean(feature_bound);
485 square_mean[i] = sum_squared[i] / num_elements_per_feature;
490 var[i] = square_mean[i] - mean_square[i];
583 std::vector<float> square_mean(feature_bound);
585 square_mean[i] = sum_squared[i] / num_elements_per_feature;
590 var[i] = square_mean[i] - mean_square[i];
698 std::vector<float> square_mean(feature_bound);
701 square_mean[i] = sum_squared[i] / num_elements_per_feature;
703 square_mean[i] = 0;
709 var[i] = square_mean[i] - mean_square[i]
    [all...]
  /external/tensorflow/tensorflow/compiler/xla/service/
batchnorm_expander.cc 239 auto square_mean = add(HloInstruction::CreateBinary( local
248 feature_shape, HloOpcode::kSubtract, square_mean, mean_square));
  /external/tensorflow/tensorflow/contrib/factorization/python/ops/
gmm_ops.py 428 square_mean = math_ops.matmul(mean, mean, transpose_a=True)
429 new_cov = b[k, :, :] - square_mean + self._min_var

Completed in 504 milliseconds