Home
last modified time | relevance | path

Searched refs:linalg_ops (Results 1 – 25 of 69) sorted by relevance

123

/external/tensorflow/tensorflow/compiler/tests/
Dcholesky_op_test.py30 from tensorflow.python.ops import linalg_ops
62 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)
Dmatrix_solve_op_test.py25 from tensorflow.python.ops import linalg_ops
40 tf_ans = linalg_ops.matrix_solve(a, b, adjoint=adjoint)
72 s1 = linalg_ops.matrix_solve(lhs1, rhs1, adjoint=adjoint)
73 s2 = linalg_ops.matrix_solve(lhs2, rhs2, adjoint=adjoint)
Dmatrix_triangular_solve_op_test.py31 from tensorflow.python.ops import linalg_ops
65 x = linalg_ops.matrix_triangular_solve(
146 linalg_ops.matrix_triangular_solve(a, b)
156 linalg_ops.matrix_triangular_solve(lhs, rhs)
158 linalg_ops.matrix_triangular_solve(lhs, rhs)
168 linalg_ops.matrix_triangular_solve(lhs, rhs)
170 linalg_ops.matrix_triangular_solve(lhs, rhs)
Dself_adjoint_eig_op_test.py28 from tensorflow.python.ops import linalg_ops
45 e, v = linalg_ops.self_adjoint_eig(x_tf)
/external/tensorflow/tensorflow/python/kernel_tests/
Deig_op_test.py28 from tensorflow.python.ops import linalg_ops
50 linalg_ops.eig(scalar)
53 linalg_ops.eig(vector)
63 e1, v1 = linalg_ops.eig(matrix1)
64 e2, v2 = linalg_ops.eig(matrix2)
67 e1 = linalg_ops.eigvals(matrix1)
68 e2 = linalg_ops.eigvals(matrix2)
88 (e, v) = self.evaluate(linalg_ops.self_adjoint_eig(matrix_tensor))
171 tf_e, tf_v = linalg_ops.eig(constant_op.constant(a))
176 linalg_ops.matrix_inverse(tf_v))
[all …]
Dself_adjoint_eig_op_test.py28 from tensorflow.python.ops import linalg_ops
50 linalg_ops.self_adjoint_eig(scalar)
53 linalg_ops.self_adjoint_eig(vector)
63 e1, v1 = linalg_ops.self_adjoint_eig(matrix1)
64 e2, v2 = linalg_ops.self_adjoint_eig(matrix2)
67 e1 = linalg_ops.self_adjoint_eigvals(matrix1)
68 e2 = linalg_ops.self_adjoint_eigvals(matrix2)
88 (e, v) = self.evaluate(linalg_ops.self_adjoint_eig(matrix_tensor))
161 tf_e, tf_v = linalg_ops.self_adjoint_eig(constant_op.constant(a))
174 tf_e = linalg_ops.self_adjoint_eigvals(constant_op.constant(a))
[all …]
Dnorm_op_test.py26 from tensorflow.python.ops import linalg_ops
45 linalg_ops.norm(matrix, ord=ord_)
50 linalg_ops.norm(matrix, ord=ord_, axis=-1)
55 linalg_ops.norm(matrix, ord=ord_, axis=[-2, -1])
64 linalg_ops.norm(matrix, axis=axis_)
74 tf_norm = linalg_ops.norm(
79 tf_norm = linalg_ops.norm(
Dcholesky_op_test.py33 from tensorflow.python.ops import linalg_ops
64 l_inverse_middle = linalg_ops.matrix_triangular_solve(l, middle, adjoint=True)
70 grad_a = linalg_ops.matrix_triangular_solve(l, z_h, adjoint=True)
76 l_inverse = linalg_ops.matrix_inverse(l)
83 l_inverse = linalg_ops.matrix_triangular_solve(l,
84 linalg_ops.eye(
108 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)
[all …]
Dmatrix_solve_op_test.py32 from tensorflow.python.ops import linalg_ops
65 tf_ans = linalg_ops.matrix_solve(a_ph, b_ph, adjoint=adjoint)
69 tf_ans = linalg_ops.matrix_solve(a, b, adjoint=adjoint)
105 self.evaluate(linalg_ops.matrix_solve(matrix, matrix))
113 self.evaluate(linalg_ops.matrix_solve(matrix, rhs))
121 self.evaluate(linalg_ops.matrix_solve(matrix, matrix))
137 s1 = linalg_ops.matrix_solve(lhs1, rhs1, adjoint=adjoint_)
138 s2 = linalg_ops.matrix_solve(lhs2, rhs2, adjoint=adjoint_)
185 x = linalg_ops.matrix_solve(matrix, rhs, adjoint=adjoint)
203 x = linalg_ops.matrix_solve(matrix, rhs, adjoint=adjoint)
Dlinalg_grad_test.py28 from tensorflow.python.ops import linalg_ops
52 determinants = linalg_ops.matrix_determinant(batch_identity)
200 linalg_ops.matrix_solve, dtype, shape, adjoint=adjoint))
208 linalg_ops.matrix_triangular_solve,
222 linalg_ops.banded_triangular_solve,
239 _GetMatrixUnaryFunctorGradientTest(linalg_ops.matrix_inverse, dtype,
251 _GetMatrixUnaryFunctorGradientTest(linalg_ops.matrix_determinant,
257 lambda x: linalg_ops.log_matrix_determinant(x)[1], dtype,
268 _GetMatrixUnaryFunctorGradientTest(linalg_ops.matrix_square_root,
286 (lambda a, b, l=l2_regularization: linalg_ops.matrix_solve_ls(
Dmatrix_inverse_op_test.py28 from tensorflow.python.ops import linalg_ops
42 inv = linalg_ops.matrix_inverse(y, adjoint=adjoint)
107 linalg_ops.matrix_inverse(np.array([[1., 2., 3.], [3., 4., 5.]]))
114 linalg_ops.matrix_inverse(tensor3)
123 linalg_ops.matrix_inverse(tensor3).eval()
147 inv1 = linalg_ops.matrix_inverse(matrix1, adjoint=adjoint_)
148 inv2 = linalg_ops.matrix_inverse(matrix2, adjoint=adjoint_)
187 inv = linalg_ops.matrix_inverse(matrix, adjoint=adjoint)
201 inv = linalg_ops.matrix_inverse(matrix, adjoint=adjoint)
Dmatrix_solve_ls_op_test.py32 from tensorflow.python.ops import linalg_ops
119 tf_ans = linalg_ops.matrix_solve_ls(
122 tf_ans = linalg_ops.matrix_solve_ls(
138 tf_r_norm = linalg_ops.norm(tf_r, ord="fro", axis=[-2, -1])
154 linalg_ops.matrix_solve_ls(matrix, rhs)
163 linalg_ops.matrix_solve_ls(empty0, empty0, fast=fast))
166 linalg_ops.matrix_solve_ls(empty0, full, fast=fast))
169 linalg_ops.matrix_solve_ls(full, empty0, fast=fast))
172 linalg_ops.matrix_solve_ls(empty1, empty1, fast=fast))
180 answer = linalg_ops.matrix_solve(matrix, rhs)
[all …]
Dqr_op_test.py33 from tensorflow.python.ops import linalg_ops
55 linalg_ops.qr(scalar)
59 linalg_ops.qr(vector)
74 q1, r1 = linalg_ops.qr(matrix1, full_matrices=full_matrices_)
75 q2, r2 = linalg_ops.qr(matrix2, full_matrices=full_matrices_)
143 q_tf, r_tf = linalg_ops.qr(x_tf, full_matrices=full_matrices_)
184 ret = linalg_ops.qr(x, full_matrices=True)
216 lambda a: linalg_ops.qr(a, full_matrices=full_matrices_)[0],
217 lambda a: linalg_ops.qr(a, full_matrices=full_matrices_)[1]
257 q, r = linalg_ops.qr(matrix)
[all …]
Dlu_op_test.py30 from tensorflow.python.ops import linalg_ops
69 lu, perm = linalg_ops.lu(x, output_idx_type=output_idx_type)
80 eye = linalg_ops.eye(
151 _, p = linalg_ops.lu(data)
162 _, p = linalg_ops.lu(data)
175 linalg_ops.lu(
180 linalg_ops.lu(
231 lu1, p1 = linalg_ops.lu(matrix1)
232 lu2, p2 = linalg_ops.lu(matrix2)
270 lu, p = linalg_ops.lu(matrix)
[all …]
Dsvd_op_test.py34 from tensorflow.python.ops import linalg_ops
57 linalg_ops.svd(scalar)
61 linalg_ops.svd(vector)
74 s1, u1, v1 = linalg_ops.svd(
76 s2, u2, v2 = linalg_ops.svd(
80 s1 = linalg_ops.svd(
82 s2 = linalg_ops.svd(
161 s_tf, u_tf, v_tf = linalg_ops.svd(
170 s_tf = linalg_ops.svd(
206 tf_s, tf_u, tf_v = linalg_ops.svd(
[all …]
Ddeterminant_op_test.py29 from tensorflow.python.ops import linalg_ops
68 linalg_ops.matrix_determinant(matrix_x))
141 linalg_ops.matrix_determinant(
149 linalg_ops.matrix_determinant(tensor1)
160 det1 = linalg_ops.matrix_determinant(matrix1)
161 det2 = linalg_ops.matrix_determinant(matrix2)
196 d = linalg_ops.matrix_determinant(matrix)
209 d = linalg_ops.matrix_determinant(matrix)
Dlinalg_ops_test.py31 from tensorflow.python.ops import linalg_ops
67 chol = linalg_ops.cholesky(array)
71 x = linalg_ops.cholesky_solve(chol, rhs)
157 self.assertEqual((2, 2), linalg_ops.eye(num_rows=2).shape)
158 self.assertEqual((2, 3), linalg_ops.eye(num_rows=2, num_columns=3).shape)
164 linalg_ops.eye(num_rows=2, batch_shape=batch_shape).shape)
167 linalg_ops.eye(
187 identity_matrix = linalg_ops.eye(
221 eye_tf = self.evaluate(linalg_ops.eye(
257 eye = linalg_ops.eye(
/external/tensorflow/tensorflow/python/ops/linalg/
Dlinalg_impl.py31 from tensorflow.python.ops import linalg_ops
40 cholesky = linalg_ops.cholesky
41 cholesky_solve = linalg_ops.cholesky_solve
42 det = linalg_ops.matrix_determinant
47 eigh = linalg_ops.self_adjoint_eig
48 eigvalsh = linalg_ops.self_adjoint_eigvals
50 eye = linalg_ops.eye
51 inv = linalg_ops.matrix_inverse
55 lstsq = linalg_ops.matrix_solve_ls
56 norm = linalg_ops.norm
[all …]
Dlinear_operator_low_rank_update.py24 from tensorflow.python.ops import linalg_ops
392 det_c = linalg_ops.matrix_determinant(self._make_capacitance())
406 linalg_ops.cholesky(self._make_capacitance()))
410 det_c = linalg_ops.matrix_determinant(self._make_capacitance())
445 capinv_vh_linv_rhs = linalg_ops.cholesky_solve(
446 linalg_ops.cholesky(self._make_capacitance()), vh_linv_rhs)
Dlinear_operator.py34 from tensorflow.python.ops import linalg_ops
520 singular_values = linalg_ops.svd(self.to_dense(), compute_uv=False)
572 array_ops.matrix_diag_part(linalg_ops.cholesky(self.to_dense())),
735 return linalg_ops.matrix_determinant(self.to_dense())
761 diag = array_ops.matrix_diag_part(linalg_ops.cholesky(self.to_dense()))
792 return linalg_ops.cholesky_solve(
793 linalg_ops.cholesky(self.to_dense()), rhs)
1016 eye = linalg_ops.eye(num_rows=n, batch_shape=batch_shape, dtype=self.dtype)
1093 return linalg_ops.self_adjoint_eigvals(self.to_dense())
1118 vals = linalg_ops.svd(self.to_dense(), compute_uv=False)
DBUILD34 "//tensorflow/python:linalg_ops",
48 "//tensorflow/python:linalg_ops",
/external/tensorflow/tensorflow/compiler/mlir/hlo/lib/Dialect/mhlo/transforms/
Dlhlo_fuse_linalg.cc171 SmallVector<LinalgOp, 8> linalg_ops; in runOnFunction() local
172 func.walk([&](LinalgOp op) { linalg_ops.push_back(op); }); in runOnFunction()
173 for (LinalgOp op : llvm::reverse(linalg_ops)) { in runOnFunction()
176 linalg::LinalgDependenceGraph graph(aliases, linalg_ops); in runOnFunction()
181 linalg_ops.begin(), linalg_ops.end(), in runOnFunction()
/external/tensorflow/tensorflow/python/ops/
Dlinalg_grad.py45 from tensorflow.python.ops import linalg_ops
378 a_adj_inv = linalg_ops.matrix_inverse(a, adjoint=True)
424 eye = linalg_ops.eye(order, dtype=sqrtm.dtype) # m x m identity matrix
443 vec_dsqrtm = linalg_ops.matrix_solve(ksum, vec_da)
455 a_adj_inv = linalg_ops.matrix_inverse(a, adjoint=True)
469 l_inverse = linalg_ops.matrix_triangular_solve(l,
470 linalg_ops.eye(
506 linalg_ops.matrix_triangular_solve(
557 grad_b = linalg_ops.matrix_solve(a, grad, adjoint=not adjoint_a)
589 chol = linalg_ops._RegularizedGramianCholesky(
[all …]
Dimage_ops.py158 from tensorflow.python.ops import linalg_ops
265 inverse = linalg_ops.matrix_inverse(transforms)
299 inverse = linalg_ops.matrix_inverse(transforms)
/external/tensorflow/tensorflow/python/ops/parallel_for/
Dmath_test.py30 from tensorflow.python.ops import linalg_ops
672 + linalg_ops.eye(3)) # Ensure well-conditioned.
675 return linalg_ops.cholesky(array_ops.gather(x, i))
685 return linalg_ops.log_matrix_determinant(array_ops.gather(x, i))
693 10 * linalg_ops.eye(2)) # Ensure well-conditioned.
699 return linalg_ops.matrix_inverse(array_ops.gather(x, i),
712 10 * linalg_ops.eye(3)) # Ensure well-conditioned.
719 return linalg_ops.matrix_solve(a, b, adjoint=adjoint)
734 linalg_ops.eye(3), # Ensure well-conditioned.
742 return linalg_ops.matrix_triangular_solve(
[all …]

123