Home
last modified time | relevance | path

Searched refs:cholesky (Results 1 – 25 of 86) sorted by relevance

1234

/external/tensorflow/tensorflow/compiler/xla/tests/
Dcholesky_test.cc241 auto cholesky = Triangle(Cholesky(matrix, lower), lower); in XLA_TEST_P() local
246 verification = BatchDot(cholesky, TransposeInMinorDims(cholesky), in XLA_TEST_P()
249 verification = BatchDot(TransposeInMinorDims(cholesky), cholesky, in XLA_TEST_P()
286 auto cholesky = Triangle(Cholesky(matrix, lower), lower); in XLA_TEST_P() local
291 verification = BatchDot(cholesky, TransposeInMinorDims(Conj(cholesky)), in XLA_TEST_P()
294 verification = BatchDot(TransposeInMinorDims(Conj(cholesky)), cholesky, in XLA_TEST_P()
/external/tensorflow/tensorflow/compiler/tests/
Dcholesky_op_test.py62 chol = linalg_ops.cholesky(placeholder)
91 linalg_ops.cholesky(np.array([[1., 2., 3.], [3., 4., 5.]], dtype=dtype))
93 linalg_ops.cholesky(
102 linalg_ops.cholesky(np.array([[1., 2., 3.], [3., 4., 5.]], dtype=dtype))
104 linalg_ops.cholesky(
114 linalg_ops.cholesky(tensor3)
116 linalg_ops.cholesky(tensor3)
123 linalg_ops.cholesky(tensor3)
125 linalg_ops.cholesky(tensor3)
/external/tensorflow/tensorflow/python/kernel_tests/
Dcholesky_op_test.py108 chol = linalg_ops.cholesky(x)
150 linalg_ops.cholesky(np.array([[1., 2., 3.], [3., 4., 5.]]))
152 linalg_ops.cholesky(
160 linalg_ops.cholesky(tensor3)
162 linalg_ops.cholesky(tensor3)
191 c1 = linalg_ops.cholesky(matrix1)
192 c2 = linalg_ops.cholesky(matrix2)
279 c = linalg_ops.cholesky(a)
332 l = linalg_ops.cholesky(matrix)
346 l = linalg_ops.cholesky(matrix)
[all …]
/external/tensorflow/tensorflow/core/api_def/python_api/
Dapi_def_Cholesky.pbtxt4 name: "linalg.cholesky"
7 name: "cholesky"
/external/tensorflow/tensorflow/python/ops/linalg/
Dcholesky_registrations.py37 linalg_ops.cholesky(linop.to_dense()),
85 operator.cholesky() for operator in block_diag_operator.operators],
98 operator.cholesky() for operator in kronecker_operator.operators],
Dlinear_operator.py572 array_ops.matrix_diag_part(linalg_ops.cholesky(self.to_dense())),
761 diag = array_ops.matrix_diag_part(linalg_ops.cholesky(self.to_dense()))
793 linalg_ops.cholesky(self.to_dense()), rhs)
978 def cholesky(self, name="cholesky"): member in LinearOperator
1001 return linear_operator_algebra.cholesky(self)
1171 @dispatch.dispatch_for_types(linalg.cholesky, LinearOperator)
1173 return input.cholesky(name)
/external/eigen/test/
Dcholesky.cpp58 template<typename MatrixType> void cholesky(const MatrixType& m) in cholesky() function
288 cholesky(m); in cholesky_cplx()
481 CALL_SUBTEST_1( cholesky(Matrix<double,1,1>()) ); in test_cholesky()
482 CALL_SUBTEST_3( cholesky(Matrix2d()) ); in test_cholesky()
485 CALL_SUBTEST_4( cholesky(Matrix3f()) ); in test_cholesky()
486 CALL_SUBTEST_5( cholesky(Matrix4d()) ); in test_cholesky()
489 CALL_SUBTEST_2( cholesky(MatrixXd(s,s)) ); in test_cholesky()
/external/tensorflow/tensorflow/python/kernel_tests/linalg/
Dlinear_operator_algebra_test.py106 CustomLinOp(dtype=None, is_self_adjoint=True).cholesky()
109 CustomLinOp(dtype=None, is_positive_definite=True).cholesky()
113 self.assertEqual("OK", custom_linop.cholesky())
/external/tensorflow/tensorflow/compiler/mlir/hlo/tests/
Dlhlo_gpu_ops.mlir219 // CHECK-LABEL: func @cholesky
220 func @cholesky(%arg : memref<10x10xf32>, %out: memref<10x10xf32>) {
223 "lmhlo_gpu.cholesky"(%arg, %out, %scratch, %info) { is_lower = true }
/external/tensorflow/tensorflow/core/api_def/base_api/
Dapi_def_CholeskyGrad.pbtxt6 Output of batch Cholesky algorithm l = cholesky(A). Shape is `[..., M, M]`.
/external/eigen/bench/btl/actions/
Daction_cholesky.hh98 Interface::cholesky(X,C,_size); in calculate()
/external/eigen/bench/btl/data/
Dgo_mean44 source mk_mean_script.sh cholesky $1 11 100 300 1000 $mode $prefix
Daction_settings.txt11 cholesky ; "{/*1.5 Cholesky decomposition}" ; "matrix size" ; 4:5000
/external/tensorflow/tensorflow/tools/api/golden/v1/
Dtensorflow.linalg.-linear-operator-block-diag.pbtxt118 name: "cholesky"
119 argspec: "args=[\'self\', \'name\'], varargs=None, keywords=None, defaults=[\'cholesky\'], "
Dtensorflow.linalg.-linear-operator-composition.pbtxt118 name: "cholesky"
119 argspec: "args=[\'self\', \'name\'], varargs=None, keywords=None, defaults=[\'cholesky\'], "
Dtensorflow.linalg.-linear-operator-permutation.pbtxt118 name: "cholesky"
119 argspec: "args=[\'self\', \'name\'], varargs=None, keywords=None, defaults=[\'cholesky\'], "
Dtensorflow.linalg.-linear-operator-inversion.pbtxt118 name: "cholesky"
119 argspec: "args=[\'self\', \'name\'], varargs=None, keywords=None, defaults=[\'cholesky\'], "
Dtensorflow.linalg.-linear-operator-toeplitz.pbtxt122 name: "cholesky"
123 argspec: "args=[\'self\', \'name\'], varargs=None, keywords=None, defaults=[\'cholesky\'], "
Dtensorflow.linalg.-linear-operator-identity.pbtxt115 name: "cholesky"
116 argspec: "args=[\'self\', \'name\'], varargs=None, keywords=None, defaults=[\'cholesky\'], "
Dtensorflow.linalg.-linear-operator-kronecker.pbtxt118 name: "cholesky"
119 argspec: "args=[\'self\', \'name\'], varargs=None, keywords=None, defaults=[\'cholesky\'], "
/external/tensorflow/tensorflow/tools/api/golden/v2/
Dtensorflow.linalg.-linear-operator-block-lower-triangular.pbtxt118 name: "cholesky"
119 argspec: "args=[\'self\', \'name\'], varargs=None, keywords=None, defaults=[\'cholesky\'], "
Dtensorflow.linalg.-linear-operator-block-diag.pbtxt118 name: "cholesky"
119 argspec: "args=[\'self\', \'name\'], varargs=None, keywords=None, defaults=[\'cholesky\'], "
Dtensorflow.linalg.-linear-operator-toeplitz.pbtxt122 name: "cholesky"
123 argspec: "args=[\'self\', \'name\'], varargs=None, keywords=None, defaults=[\'cholesky\'], "
Dtensorflow.linalg.-linear-operator-diag.pbtxt118 name: "cholesky"
119 argspec: "args=[\'self\', \'name\'], varargs=None, keywords=None, defaults=[\'cholesky\'], "
Dtensorflow.linalg.-linear-operator-permutation.pbtxt118 name: "cholesky"
119 argspec: "args=[\'self\', \'name\'], varargs=None, keywords=None, defaults=[\'cholesky\'], "

1234