Searched defs:linalg_ops (Results 1 - 25 of 59) sorted by relevance

123

/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)
H A Dtest_util.py26 from tensorflow.python.ops import linalg_ops namespace
287 return math_ops.cast(linalg_ops.norm(x - center, axis=-1) <= radius,
/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/contrib/timeseries/python/timeseries/state_space_models/
H A Dtest_utils.py28 from tensorflow.python.ops import linalg_ops namespace
46 starting_matrix = linalg_ops.eye(
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 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 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(
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_triangular_solve_op_test.py25 from tensorflow.python.ops import linalg_ops namespace
81 tf_ans = linalg_ops.matrix_triangular_solve(
88 tf_ans = linalg_ops.matrix_triangular_solve(
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 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 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...]
/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/distributions/python/ops/bijectors/
H A Dcholesky_outer_product.py28 from tensorflow.python.ops import linalg_ops namespace
118 else linalg_ops.cholesky(y))
/external/tensorflow/tensorflow/contrib/gan/python/eval/python/
H A Dsliced_wasserstein_impl.py32 from tensorflow.python.ops import linalg_ops namespace
190 sig, u = linalg_ops.svd(array_ops.concat([a, b], 0))[:2]
/external/tensorflow/tensorflow/python/kernel_tests/linalg/
H A Dlinear_operator_diag_test.py22 from tensorflow.python.ops import linalg_ops namespace
149 mat_solve = linalg_ops.matrix_solve(mat, x)
/external/tensorflow/tensorflow/python/ops/linalg/
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/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/image/python/ops/
H A Dimage_ops.py28 from tensorflow.python.ops import linalg_ops namespace
395 inverse = linalg_ops.matrix_inverse(transforms)
/external/tensorflow/tensorflow/contrib/kfac/python/kernel_tests/
H A Destimator_test.py32 from tensorflow.python.ops import linalg_ops namespace
215 cov_matrix.assign(2 * linalg_ops.eye(int(cov_matrix.shape[0])))
/external/tensorflow/tensorflow/contrib/kfac/python/ops/
H A Doptimizer.py29 from tensorflow.python.ops import linalg_ops namespace
493 return math_ops.matmul(linalg_ops.matrix_inverse(m), c)

Completed in 1390 milliseconds

123