Searched refs:cholesky_solve (Results 1 - 7 of 7) sorted by relevance

/external/tensorflow/tensorflow/python/ops/linalg/
H A Dlinalg_impl.py32 cholesky_solve = linalg_ops.cholesky_solve variable
H A Dlinear_operator_low_rank_update.py437 capinv_vh_linv_rhs = linalg_ops.cholesky_solve(
H A Dlinear_operator.py729 return linalg_ops.cholesky_solve(self._get_cached_chol(), rhs)
/external/tensorflow/tensorflow/python/ops/
H A Dlinalg_ops.py81 @tf_export('cholesky_solve', 'linalg.cholesky_solve')
82 def cholesky_solve(chol, rhs, name=None): function
90 X = tf.cholesky_solve(chol, RHS) # shape 10 x 2 x 1
108 name: A name to give this `Op`. Defaults to `cholesky_solve`.
116 with ops.name_scope(name, 'cholesky_solve', [chol, rhs]):
290 return cholesky_solve(chol, math_ops.matmul(matrix, rhs, adjoint_a=True))
296 return math_ops.matmul(matrix, cholesky_solve(chol, rhs), adjoint_a=True)
H A Dlinalg_grad.py156 z = linalg_ops.cholesky_solve(chol, grad)
179 grad_b = linalg_ops.cholesky_solve(chol, math_ops.matmul(a, grad))
181 tmp = linalg_ops.cholesky_solve(chol, b)
/external/tensorflow/tensorflow/python/kernel_tests/
H A Dlinalg_ops_test.py63 x = linalg_ops.cholesky_solve(chol, rhs)
/external/tensorflow/tensorflow/contrib/kfac/python/ops/
H A Dutils.py192 return linalg_ops.cholesky_solve(chol, identity)

Completed in 315 milliseconds