| /external/tensorflow/tensorflow/contrib/distributions/python/kernel_tests/ |
| D | vector_student_t_test.py | 45 def __init__(self, df, loc, scale_tril): argument
48 self._scale_tril = np.asarray(scale_tril)
51 def _compute(df, loc, scale_tril, x): argument
52 k = scale_tril.shape[-1]
53 ildj = np.sum(np.log(np.abs(np.diag(scale_tril))), axis=-1)
58 y = linalg.solve_triangular(scale_tril, np.matrix(x - loc).T,
84 scale_tril = np.diag(scale_diag)
87 df=df, loc=loc, scale_tril=scale_tril)
112 scale_tril = np.concatenate([[np.diag(scale_diag[i])]
117 df=df, loc=loc, scale_tril=scale_tril)
[all …]
|
| D | mvn_tril_test.py | 245 scale_tril=np.linalg.cholesky(sigma_a),
249 scale_tril=np.linalg.cholesky(sigma_b),
267 scale_tril=np.linalg.cholesky(sigma_a),
271 scale_tril=np.linalg.cholesky(sigma_b),
294 scale_tril=np.linalg.cholesky(sigma_a),
298 scale_tril=np.linalg.cholesky(sigma_b),
319 scale_tril=np.linalg.cholesky(sigma_a),
331 scale_tril = np.array([[3., 0], [1, -2]], dtype=np.float32) / 3.
334 true_scale = scale_tril
342 scale_tril=scale_tril,
[all …]
|
| D | distribution_util_test.py | 60 scale_tril=None, argument
64 if scale_tril is not None:
65 scale_tril = np.tril(scale_tril)
67 scale_tril += _matrix_diag(np.array(scale_diag, dtype=np.float32))
69 scale_tril += (
71 [scale_tril.shape[-1]], dtype=np.float32)))
72 return scale_tril
147 scale = distribution_util.make_tril_scale(scale_tril=[[1., 1], [1., 1.]])
154 scale_tril=[[0., 1], [1., 1.]], validate_args=True)
161 scale_tril=[[-1., 1], [1., 1.]],
|
| D | autoregressive_test.py | 59 affine = Affine(scale_tril=self._random_scale_tril(event_size))
73 affine = Affine(scale_tril=self._random_scale_tril(event_size))
|
| D | transformed_distribution_test.py | 327 bijector=bs.Affine(shift=self._shift, scale_tril=self._tril),
335 bijector=bs.Affine(shift=self._shift, scale_tril=self._tril),
436 bijector=bs.Affine(shift=self._shift, scale_tril=self._tril),
453 bijector=bs.Affine(shift=self._shift, scale_tril=self._tril),
466 bijector=bs.Affine(shift=self._shift, scale_tril=self._tril),
|
| D | mvn_diag_plus_low_rank_test.py | 171 scale_tril=np.array([[6., 0, 0],
|
| D | mixture_test.py | 906 loc=mu, scale_tril=linalg_ops.cholesky(sigma))
|
| /external/tensorflow/tensorflow/contrib/distributions/python/ops/ |
| D | mvn_tril.py | 149 scale_tril=None, argument
195 if loc is None and scale_tril is None:
198 with ops.name_scope("init", values=[loc, scale_tril]):
200 scale_tril = _convert_to_tensor(scale_tril, name="scale_tril")
201 if scale_tril is None:
213 scale_tril,
|
| D | distribution_util.py | 92 scale_tril=None, argument
154 scale_tril = _convert_to_tensor(scale_tril, name="scale_tril")
160 if scale_tril is not None:
161 scale_tril = array_ops.matrix_band_part(scale_tril, -1, 0) # Zero out TriU.
162 tril_diag = array_ops.matrix_diag_part(scale_tril)
168 scale_tril = array_ops.matrix_set_diag(scale_tril, tril_diag)
171 tril=_maybe_attach_assertion(scale_tril),
|
| D | mvn_full_covariance.py | 174 scale_tril = None
189 scale_tril = linalg_ops.cholesky(covariance_matrix)
192 scale_tril=scale_tril,
|
| D | vector_student_t.py | 139 scale_tril=None, argument
190 scale_tril, scale_perturb_factor, scale_perturb_diag]
213 scale_tril=scale_tril,
|
| /external/tensorflow/tensorflow/contrib/distributions/python/ops/bijectors/ |
| D | affine.py | 121 scale_tril=None, argument
197 self._is_only_identity_multiplier = (scale_tril is None and
202 shift, scale_identity_multiplier, scale_diag, scale_tril,
225 tril=scale_tril,
318 scale_tril=tril,
|
| D | __init__.py | 84 from tensorflow.contrib.distributions.python.ops.bijectors.scale_tril import *
|
| /external/tensorflow/tensorflow/contrib/distributions/python/kernel_tests/bijectors/ |
| D | affine_test.py | 255 scale_tril=[[1., 0], [2., 1]])
279 shift=mu, scale_diag=[1., 2.], scale_tril=[[1., 0], [2., 1]])
306 scale_tril=[[1., 0], [2., 1]])
404 scale_tril=[[2., 0, 0], [1, 3, 0], [2, 3, 4]],
408 shift=mu, scale_tril=[[10., 0, 0], [1, 3, 0], [2, 3, 5]])
441 scale_tril=[[2., 0, 0], [1, 3, 0], [2, 3, 4]],
445 shift=mu, scale_tril=[[6., 0, 0], [1, 3, 0], [2, 3, 5]])
476 scale_tril=None, argument
482 tril = scale_tril
|
| /external/tensorflow/tensorflow/contrib/timeseries/python/timeseries/ |
| D | math_utils.py | 62 def mvn_tril_log_prob(loc, scale_tril, x): argument
66 scale_tril, x0[..., array_ops.newaxis])[..., 0]
68 array_ops.matrix_diag_part(scale_tril)), axis=-1)
69 d = math_ops.cast(array_ops.shape(scale_tril)[-1], log_det_cov.dtype)
|
| /external/tensorflow/tensorflow/contrib/timeseries/python/timeseries/state_space_models/ |
| D | kalman_filter.py | 140 scale_tril=observation_covariance_cholesky,
|