Searched refs:linalg_ops (Results 1 - 25 of 61) sorted by relevance

123

/external/tensorflow/tensorflow/python/ops/linalg/
H A Dlinalg_impl.py24 from tensorflow.python.ops import linalg_ops namespace
31 cholesky = linalg_ops.cholesky
32 cholesky_solve = linalg_ops.cholesky_solve
33 det = linalg_ops.matrix_determinant
39 eigh = linalg_ops.self_adjoint_eig
40 eigvalsh = linalg_ops.self_adjoint_eigvals
43 eye = linalg_ops.eye
44 inv = linalg_ops.matrix_inverse
46 lstsq = linalg_ops.matrix_solve_ls
47 norm = linalg_ops
[all...]
H A Dlinalg.py41 del linalg_ops
H A Dlinear_operator_low_rank_update.py25 from tensorflow.python.ops import linalg_ops namespace
281 self._chol_capacitance = linalg_ops.cholesky(self._capacitance)
387 det_c = linalg_ops.matrix_determinant(self._capacitance)
404 det_c = linalg_ops.matrix_determinant(self._capacitance)
437 capinv_vh_linv_rhs = linalg_ops.cholesky_solve(
440 capinv_vh_linv_rhs = linalg_ops.matrix_solve(
H A Dlinear_operator_lower_triangular.py24 from tensorflow.python.ops import linalg_ops namespace
210 return linalg_ops.matrix_triangular_solve(
/external/tensorflow/tensorflow/contrib/timeseries/python/timeseries/state_space_models/
H A Dvarma.py57 from tensorflow.python.ops import linalg_ops namespace
99 linalg_ops.eye(self.num_features, dtype=self.dtype)[None, :, :],
120 linalg_ops.eye(
143 linalg_ops.eye(self.num_features, dtype=self.dtype)],
159 linalg_ops.eye(self.num_features, dtype=self.dtype),
169 linalg_ops.eye(self.num_features, dtype=self.dtype),
172 linalg_ops.eye(self.num_features, dtype=self.dtype)
H A Dkalman_filter.py29 from tensorflow.python.ops import linalg_ops namespace
138 observation_covariance_cholesky = linalg_ops.cholesky(
250 kalman_gain_transposed = linalg_ops.matrix_solve(
260 identity_extradim = linalg_ops.eye(
H A Dlevel_trend.py28 from tensorflow.python.ops import linalg_ops namespace
91 identity_matrices = linalg_ops.eye(
/external/tensorflow/tensorflow/python/kernel_tests/
H A Dnorm_op_test.py25 from tensorflow.python.ops import linalg_ops namespace
43 linalg_ops.norm(matrix, ord="fro")
48 linalg_ops.norm(matrix, ord=ord_, axis=-1)
53 linalg_ops.norm(matrix, ord=ord_, axis=[-2, -1])
61 linalg_ops.norm(matrix, axis=axis_)
71 tf_norm = linalg_ops.norm(
76 tf_norm = linalg_ops.norm(
H A Dmatrix_inverse_op_test.py27 from tensorflow.python.ops import linalg_ops namespace
41 inv = linalg_ops.matrix_inverse(y, adjoint=adjoint)
107 linalg_ops.matrix_inverse(np.array([[1., 2., 3.], [3., 4., 5.]]))
113 linalg_ops.matrix_inverse(tensor3)
122 linalg_ops.matrix_inverse(tensor3).eval()
145 inv1 = linalg_ops.matrix_inverse(matrix1, adjoint=adjoint_)
146 inv2 = linalg_ops.matrix_inverse(matrix2, adjoint=adjoint_)
185 inv = linalg_ops.matrix_inverse(matrix, adjoint=adjoint)
199 inv = linalg_ops.matrix_inverse(matrix, adjoint=adjoint)
H A Dself_adjoint_eig_op_test.py27 from tensorflow.python.ops import linalg_ops namespace
47 linalg_ops.self_adjoint_eig(scalar)
50 linalg_ops.self_adjoint_eig(vector)
59 e1, v1 = linalg_ops.self_adjoint_eig(matrix1)
60 e2, v2 = linalg_ops.self_adjoint_eig(matrix2)
63 e1 = linalg_ops.self_adjoint_eigvals(matrix1)
64 e2 = linalg_ops.self_adjoint_eigvals(matrix2)
140 tf_e, tf_v = linalg_ops.self_adjoint_eig(constant_op.constant(a))
153 tf_e = linalg_ops.self_adjoint_eigvals(constant_op.constant(a))
190 tf_e, tf_v = linalg_ops
[all...]
H A Ddeterminant_op_test.py28 from tensorflow.python.ops import linalg_ops namespace
66 linalg_ops.matrix_determinant(matrix_x))
138 linalg_ops.matrix_determinant(
145 linalg_ops.matrix_determinant(tensor1)
155 det1 = linalg_ops.matrix_determinant(matrix1)
156 det2 = linalg_ops.matrix_determinant(matrix2)
191 d = linalg_ops.matrix_determinant(matrix)
204 d = linalg_ops.matrix_determinant(matrix)
H A Dlinalg_grad_test.py27 from tensorflow.python.ops import linalg_ops namespace
49 determinants = linalg_ops.matrix_determinant(batch_identity)
102 use_gpu = False if functor_ == linalg_ops.matrix_solve else True
151 linalg_ops.matrix_solve, dtype, shape, adjoint=adjoint))
158 linalg_ops.matrix_triangular_solve,
174 _GetMatrixUnaryFunctorGradientTest(linalg_ops.matrix_inverse,
178 _GetMatrixUnaryFunctorGradientTest(linalg_ops.matrix_determinant,
196 linalg_ops.matrix_solve_ls(a, b, l)),
H A Dmatrix_solve_op_test.py29 from tensorflow.python.ops import linalg_ops namespace
61 tf_ans = linalg_ops.matrix_solve(a_ph, b_ph, adjoint=adjoint)
64 tf_ans = linalg_ops.matrix_solve(a, b, adjoint=adjoint)
98 linalg_ops.matrix_solve(matrix, matrix)
106 linalg_ops.matrix_solve(matrix, rhs)
115 linalg_ops.matrix_solve(matrix, matrix).eval()
125 s1 = linalg_ops.matrix_solve(lhs1, rhs1, adjoint=adjoint_)
126 s2 = linalg_ops.matrix_solve(lhs2, rhs2, adjoint=adjoint_)
173 x = linalg_ops.matrix_solve(matrix, rhs, adjoint=adjoint)
191 x = linalg_ops
[all...]
H A Dmatrix_solve_ls_op_test.py29 from tensorflow.python.ops import linalg_ops namespace
114 tf_ans = linalg_ops.matrix_solve_ls(
117 tf_ans = linalg_ops.matrix_solve_ls(
125 tf_r_norm = linalg_ops.norm(tf_r, ord="fro", axis=[-2, -1])
141 linalg_ops.matrix_solve_ls(matrix, rhs)
149 tf_ans = linalg_ops.matrix_solve_ls(empty0, empty0, fast=fast).eval()
151 tf_ans = linalg_ops.matrix_solve_ls(empty0, full, fast=fast).eval()
153 tf_ans = linalg_ops.matrix_solve_ls(full, empty0, fast=fast).eval()
155 tf_ans = linalg_ops.matrix_solve_ls(empty1, empty1, fast=fast).eval()
162 answer = linalg_ops
[all...]
H A Dcholesky_op_test.py34 from tensorflow.python.ops import linalg_ops namespace
69 l_inverse_middle = linalg_ops.matrix_triangular_solve(l, middle, adjoint=True)
75 grad_a = linalg_ops.matrix_triangular_solve(l, z_h, adjoint=True)
81 l_inverse = linalg_ops.matrix_inverse(l)
88 l_inverse = linalg_ops.matrix_triangular_solve(l,
89 linalg_ops.eye(
114 chol = linalg_ops.cholesky(x)
149 linalg_ops.cholesky(np.array([[1., 2., 3.], [3., 4., 5.]]))
151 linalg_ops.cholesky(
158 linalg_ops
[all...]
H A Dsvd_op_test.py26 from tensorflow.python.ops import linalg_ops namespace
46 linalg_ops.svd(scalar)
50 linalg_ops.svd(vector)
60 s1, u1, v1 = linalg_ops.svd(
62 s2, u2, v2 = linalg_ops.svd(
66 s1 = linalg_ops.svd(
68 s2 = linalg_ops.svd(
150 s_tf, u_tf, v_tf = linalg_ops.svd(
158 s_tf = linalg_ops.svd(
196 tf_s, tf_u, tf_v = linalg_ops
[all...]
H A Dqr_op_test.py26 from tensorflow.python.ops import linalg_ops namespace
46 linalg_ops.qr(scalar)
50 linalg_ops.qr(vector)
60 q1, r1 = linalg_ops.qr(matrix1, full_matrices=full_matrices_)
61 q2, r2 = linalg_ops.qr(matrix2, full_matrices=full_matrices_)
129 q_tf, r_tf = linalg_ops.qr(x_tf, full_matrices=full_matrices_)
178 tf_b = linalg_ops.qr(tf_a, full_matrices=full_matrices_)
H A Dlinalg_ops_test.py15 """Tests for tensorflow.python.ops.linalg_ops."""
26 from tensorflow.python.ops import linalg_ops namespace
60 chol = linalg_ops.cholesky(array)
63 x = linalg_ops.cholesky_solve(chol, rhs)
159 eye = linalg_ops.eye(
172 eye_tf = linalg_ops.eye(
/external/tensorflow/tensorflow/contrib/distributions/python/ops/
H A Dmvn_full_covariance.py26 from tensorflow.python.ops import linalg_ops namespace
180 scale_tril = linalg_ops.cholesky(covariance_matrix)
/external/tensorflow/tensorflow/python/ops/
H A Dlinalg_grad.py15 """Gradients for operators defined in linalg_ops.py.
33 from tensorflow.python.ops import linalg_ops namespace
51 a_adj_inv = linalg_ops.matrix_inverse(a, adjoint=True)
66 l_inverse = linalg_ops.matrix_triangular_solve(l,
67 linalg_ops.eye(
106 linalg_ops.matrix_triangular_solve(
120 grad_b = linalg_ops.matrix_solve(a, grad, adjoint=not adjoint_a)
152 chol = linalg_ops._RegularizedGramianCholesky(
156 z = linalg_ops.cholesky_solve(chol, grad)
176 chol = linalg_ops
[all...]
/external/tensorflow/tensorflow/contrib/solvers/python/ops/
H A Dlinear_equations.py29 from tensorflow.python.ops import linalg_ops namespace
31 from tensorflow.python.ops import linalg_ops namespace
88 return math_ops.logical_and(i < max_iter, linalg_ops.norm(state.r) > tol)
121 tol *= linalg_ops.norm(r0)
/external/tensorflow/tensorflow/compiler/tests/
H A Dcholesky_op_test.py30 from tensorflow.python.ops import linalg_ops namespace
56 chol = linalg_ops.cholesky(placeholder)
84 linalg_ops.cholesky(np.array([[1., 2., 3.], [3., 4., 5.]], dtype=dtype))
86 linalg_ops.cholesky(
95 linalg_ops.cholesky(tensor3)
97 linalg_ops.cholesky(tensor3)
H A Dmatrix_triangular_solve_op_test.py29 from tensorflow.python.ops import linalg_ops namespace
54 x = linalg_ops.matrix_triangular_solve(
114 linalg_ops.matrix_triangular_solve(a, b)
116 linalg_ops.matrix_triangular_solve(a, b)
124 linalg_ops.matrix_triangular_solve(lhs, rhs)
126 linalg_ops.matrix_triangular_solve(lhs, rhs)
/external/tensorflow/tensorflow/contrib/kfac/python/ops/
H A Dutils.py30 from tensorflow.python.ops import linalg_ops namespace
179 identity = linalg_ops.eye(tensor.shape.as_list()[0], dtype=tensor.dtype)
186 return linalg_ops.matrix_inverse(tensor + damping * identity)
191 chol = linalg_ops.cholesky(tensor + damping * identity)
192 return linalg_ops.cholesky_solve(chol, identity)
197 eigenvalues, eigenvectors = linalg_ops.self_adjoint_eig(
217 evals, evecs, _ = linalg_ops.svd(mat)
224 evals, evecs = linalg_ops.self_adjoint_eig(mat)
/external/tensorflow/tensorflow/contrib/distributions/python/ops/bijectors/
H A Dcholesky_outer_product.py28 from tensorflow.python.ops import linalg_ops namespace
118 else linalg_ops.cholesky(y))

Completed in 453 milliseconds

123