| /external/tensorflow/tensorflow/python/ops/linalg/ |
| linalg_impl.py | 28 from tensorflow.python.ops import linalg_ops
35 cholesky = linalg_ops.cholesky
36 cholesky_solve = linalg_ops.cholesky_solve
37 det = linalg_ops.matrix_determinant
42 eigh = linalg_ops.self_adjoint_eig
43 eigvalsh = linalg_ops.self_adjoint_eigvals
45 eye = linalg_ops.eye
46 inv = linalg_ops.matrix_inverse
50 lstsq = linalg_ops.matrix_solve_ls
51 norm = linalg_ops.nor
[all...] |
| cholesky_registrations.py | 21 from tensorflow.python.ops import linalg_ops
37 linalg_ops.cholesky(linop.to_dense()),
|
| /external/tensorflow/tensorflow/python/kernel_tests/ |
| linalg_grad_test.py | 28 from tensorflow.python.ops import linalg_ops
52 determinants = linalg_ops.matrix_determinant(batch_identity)
111 use_gpu = False if functor_ == linalg_ops.matrix_solve else True
160 linalg_ops.matrix_solve, dtype, shape, adjoint=adjoint))
167 linalg_ops.matrix_triangular_solve,
183 _GetMatrixUnaryFunctorGradientTest(linalg_ops.matrix_inverse,
191 _GetMatrixUnaryFunctorGradientTest(linalg_ops.matrix_determinant,
197 lambda x: linalg_ops.log_matrix_determinant(x)[1],
208 _GetMatrixUnaryFunctorGradientTest(linalg_ops.matrix_square_root,
227 linalg_ops.matrix_solve_ls(a, b, l))
[all...] |
| matrix_solve_op_test.py | 30 from tensorflow.python.ops import linalg_ops
63 tf_ans = linalg_ops.matrix_solve(a_ph, b_ph, adjoint=adjoint)
66 tf_ans = linalg_ops.matrix_solve(a, b, adjoint=adjoint)
103 linalg_ops.matrix_solve(matrix, matrix)
112 linalg_ops.matrix_solve(matrix, rhs)
121 linalg_ops.matrix_solve(matrix, matrix).eval()
132 s1 = linalg_ops.matrix_solve(lhs1, rhs1, adjoint=adjoint_)
133 s2 = linalg_ops.matrix_solve(lhs2, rhs2, adjoint=adjoint_)
180 x = linalg_ops.matrix_solve(matrix, rhs, adjoint=adjoint)
198 x = linalg_ops.matrix_solve(matrix, rhs, adjoint=adjoint
[all...] |
| norm_op_test.py | 26 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(
|
| matrix_inverse_op_test.py | 28 from tensorflow.python.ops import linalg_ops
43 inv = linalg_ops.matrix_inverse(y, adjoint=adjoint)
110 linalg_ops.matrix_inverse(np.array([[1., 2., 3.], [3., 4., 5.]]))
117 linalg_ops.matrix_inverse(tensor3)
126 linalg_ops.matrix_inverse(tensor3).eval()
150 inv1 = linalg_ops.matrix_inverse(matrix1, adjoint=adjoint_)
151 inv2 = linalg_ops.matrix_inverse(matrix2, adjoint=adjoint_)
190 inv = linalg_ops.matrix_inverse(matrix, adjoint=adjoint)
204 inv = linalg_ops.matrix_inverse(matrix, adjoint=adjoint)
|
| self_adjoint_eig_op_test.py | 28 from tensorflow.python.ops import linalg_ops
49 linalg_ops.self_adjoint_eig(scalar)
52 linalg_ops.self_adjoint_eig(vector)
62 e1, v1 = linalg_ops.self_adjoint_eig(matrix1)
63 e2, v2 = linalg_ops.self_adjoint_eig(matrix2)
66 e1 = linalg_ops.self_adjoint_eigvals(matrix1)
67 e2 = linalg_ops.self_adjoint_eigvals(matrix2)
87 (e, v) = self.evaluate(linalg_ops.self_adjoint_eig(matrix_tensor))
160 tf_e, tf_v = linalg_ops.self_adjoint_eig(constant_op.constant(a))
173 tf_e = linalg_ops.self_adjoint_eigvals(constant_op.constant(a)
[all...] |
| cholesky_op_test.py | 35 from tensorflow.python.ops import linalg_ops
71 l_inverse_middle = linalg_ops.matrix_triangular_solve(l, middle, adjoint=True)
77 grad_a = linalg_ops.matrix_triangular_solve(l, z_h, adjoint=True)
83 l_inverse = linalg_ops.matrix_inverse(l)
90 l_inverse = linalg_ops.matrix_triangular_solve(l,
91 linalg_ops.eye(
116 chol = linalg_ops.cholesky(x)
152 linalg_ops.cholesky(np.array([[1., 2., 3.], [3., 4., 5.]]))
154 linalg_ops.cholesky(
162 linalg_ops.cholesky(tensor3
[all...] |
| lu_op_test.py | 30 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(
148 _, p = linalg_ops.lu(data)
158 _, p = linalg_ops.lu(data)
170 linalg_ops.lu(
175 linalg_ops.lu(
221 lu1, p1 = linalg_ops.lu(matrix1)
222 lu2, p2 = linalg_ops.lu(matrix2)
260 lu, p = linalg_ops.lu(matrix
[all...] |
| determinant_op_test.py | 29 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)
|
| linalg_ops_test.py | 15 """Tests for tensorflow.python.ops.linalg_ops."""
30 from tensorflow.python.ops import linalg_ops
65 chol = linalg_ops.cholesky(array)
68 x = linalg_ops.cholesky_solve(chol, rhs)
150 self.assertEqual((2, 2), linalg_ops.eye(num_rows=2).shape)
151 self.assertEqual((2, 3), linalg_ops.eye(num_rows=2, num_columns=3).shape)
157 linalg_ops.eye(num_rows=2, batch_shape=batch_shape).shape)
160 linalg_ops.eye(
180 identity_matrix = linalg_ops.eye(
214 eye_tf = self.evaluate(linalg_ops.eye
[all...] |
| matrix_solve_ls_op_test.py | 31 from tensorflow.python.ops import linalg_ops
117 tf_ans = linalg_ops.matrix_solve_ls(
120 tf_ans = linalg_ops.matrix_solve_ls(
128 tf_r_norm = linalg_ops.norm(tf_r, ord="fro", axis=[-2, -1])
145 linalg_ops.matrix_solve_ls(matrix, rhs)
154 linalg_ops.matrix_solve_ls(empty0, empty0, fast=fast))
157 linalg_ops.matrix_solve_ls(empty0, full, fast=fast))
160 linalg_ops.matrix_solve_ls(full, empty0, fast=fast))
163 linalg_ops.matrix_solve_ls(empty1, empty1, fast=fast))
171 answer = linalg_ops.matrix_solve(matrix, rhs
[all...] |
| svd_op_test.py | 29 from tensorflow.python.ops import linalg_ops
50 linalg_ops.svd(scalar)
54 linalg_ops.svd(vector)
65 s1, u1, v1 = linalg_ops.svd(
67 s2, u2, v2 = linalg_ops.svd(
71 s1 = linalg_ops.svd(
73 s2 = linalg_ops.svd(
156 s_tf, u_tf, v_tf = linalg_ops.svd(
164 s_tf = linalg_ops.svd(
200 tf_s, tf_u, tf_v = linalg_ops.svd
[all...] |
| qr_op_test.py | 28 from tensorflow.python.ops import linalg_ops
49 linalg_ops.qr(scalar)
53 linalg_ops.qr(vector)
64 q1, r1 = linalg_ops.qr(matrix1, full_matrices=full_matrices_)
65 q2, r2 = linalg_ops.qr(matrix2, full_matrices=full_matrices_)
134 q_tf, r_tf = linalg_ops.qr(x_tf, full_matrices=full_matrices_)
184 tf_b = linalg_ops.qr(tf_a, full_matrices=full_matrices_)
|
| /external/tensorflow/tensorflow/contrib/timeseries/python/timeseries/state_space_models/ |
| varma.py | 58 from tensorflow.python.ops import linalg_ops
100 linalg_ops.eye(self.num_features, dtype=self.dtype)[None, :, :],
121 linalg_ops.eye(
144 linalg_ops.eye(self.num_features, dtype=self.dtype)],
160 linalg_ops.eye(self.num_features, dtype=self.dtype),
170 linalg_ops.eye(self.num_features, dtype=self.dtype),
173 linalg_ops.eye(self.num_features, dtype=self.dtype)
|
| kalman_filter.py | 27 from tensorflow.python.ops import linalg_ops
136 observation_covariance_cholesky = linalg_ops.cholesky(
249 kalman_gain_transposed = linalg_ops.matrix_solve(
259 identity_extradim = linalg_ops.eye(
|
| /external/tensorflow/tensorflow/contrib/distributions/python/ops/bijectors/ |
| matrix_inverse_tril.py | 25 from tensorflow.python.ops import linalg_ops
85 return linalg_ops.matrix_triangular_solve(
86 x, linalg_ops.eye(shape[-1], batch_shape=shape[:-2]), lower=True)
|
| /external/tensorflow/tensorflow/python/ops/ |
| linalg_grad.py | 15 """Gradients for operators defined in linalg_ops.py.
33 from tensorflow.python.ops import linalg_ops
51 a_adj_inv = linalg_ops.matrix_inverse(a, adjoint=True)
97 eye = linalg_ops.eye(order, dtype=sqrtm.dtype) # m x m identity matrix
116 vec_dsqrtm = linalg_ops.matrix_solve(ksum, vec_da)
128 a_adj_inv = linalg_ops.matrix_inverse(a, adjoint=True)
142 l_inverse = linalg_ops.matrix_triangular_solve(l,
143 linalg_ops.eye(
182 linalg_ops.matrix_triangular_solve(
196 grad_b = linalg_ops.matrix_solve(a, grad, adjoint=not adjoint_a
[all...] |
| /external/tensorflow/tensorflow/contrib/opt/python/training/ |
| matrix_functions.py | 22 from tensorflow.python.ops import linalg_ops
61 identity = linalg_ops.eye(math_ops.cast(mat_a_size, dtypes.int32))
104 identity = linalg_ops.eye(math_ops.cast(mat_g_size, dtypes.int32))
140 z = (1 - 1 / alpha) / (2 * linalg_ops.norm(damped_mat_g))
|
| /external/tensorflow/tensorflow/compiler/tests/ |
| self_adjoint_eig_op_test.py | 27 from tensorflow.python.ops import linalg_ops
44 e, v = linalg_ops.self_adjoint_eig(x_tf)
|
| cholesky_op_test.py | 28 from tensorflow.python.ops import linalg_ops
61 chol = linalg_ops.cholesky(placeholder)
89 linalg_ops.cholesky(np.array([[1., 2., 3.], [3., 4., 5.]], dtype=dtype))
91 linalg_ops.cholesky(
100 linalg_ops.cholesky(tensor3)
102 linalg_ops.cholesky(tensor3)
|
| matrix_triangular_solve_op_test.py | 29 from tensorflow.python.ops import linalg_ops
62 x = linalg_ops.matrix_triangular_solve(
122 linalg_ops.matrix_triangular_solve(a, b)
124 linalg_ops.matrix_triangular_solve(a, b)
132 linalg_ops.matrix_triangular_solve(lhs, rhs)
134 linalg_ops.matrix_triangular_solve(lhs, rhs)
|
| qr_op_test.py | 28 from tensorflow.python.ops import linalg_ops
77 q_tf, r_tf = linalg_ops.qr(x_tf, full_matrices=full_matrices)
|
| /external/tensorflow/tensorflow/contrib/distributions/python/ops/ |
| mvn_full_covariance.py | 26 from tensorflow.python.ops import linalg_ops
189 scale_tril = linalg_ops.cholesky(covariance_matrix)
|
| /external/tensorflow/tensorflow/contrib/solvers/python/ops/ |
| linear_equations.py | 29 from tensorflow.python.ops import linalg_ops
87 return math_ops.logical_and(i < max_iter, linalg_ops.norm(state.r) > tol)
120 tol *= linalg_ops.norm(r0)
|
