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

  /external/tensorflow/tensorflow/contrib/distributions/python/kernel_tests/
mvn_full_covariance_test.py 137 mu_a, sigma_a = self._random_mu_and_sigma(batch_shape, event_shape)
140 loc=mu_a,
152 expected_kl_0 = _compute_non_batch_kl(mu_a[0, :], sigma_a[0, :, :],
154 expected_kl_1 = _compute_non_batch_kl(mu_a[1, :], sigma_a[1, :, :],
160 def _compute_non_batch_kl(mu_a, sigma_a, mu_b, sigma_b):
161 """Non-batch KL for N(mu_a, sigma_a), N(mu_b, sigma_b)."""
168 q = (mu_b - mu_a).dot(sigma_b_inv).dot(mu_b - mu_a)
169 k = mu_a.shape[0]
mvn_tril_test.py 241 mu_a, sigma_a = self._random_mu_and_sigma(batch_shape, event_shape)
244 loc=mu_a,
256 expected_kl = _compute_non_batch_kl(mu_a, sigma_a, mu_b, sigma_b)
263 mu_a, sigma_a = self._random_mu_and_sigma(batch_shape, event_shape)
266 loc=mu_a,
278 expected_kl_0 = _compute_non_batch_kl(mu_a[0, :], sigma_a[0, :, :],
280 expected_kl_1 = _compute_non_batch_kl(mu_a[1, :], sigma_a[1, :, :],
289 mu_a, sigma_a = self._random_mu_and_sigma(batch_shape, event_shape)
291 loc=mu_a,
409 def _compute_non_batch_kl(mu_a, sigma_a, mu_b, sigma_b)
    [all...]
  /external/tensorflow/tensorflow/python/kernel_tests/distributions/
normal_test.py 477 mu_a = np.array([3.0] * batch_size)
482 n_a = normal_lib.Normal(loc=mu_a, scale=sigma_a)
488 kl_expected = ((mu_a - mu_b)**2 / (2 * sigma_b**2) + 0.5 * (
  /external/clang/test/SemaCXX/
warn-thread-safety-analysis.cpp     [all...]

Completed in 389 milliseconds