/external/tensorflow/tensorflow/python/ops/linalg/ |
H A D | linalg_impl.py | 24 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 D | linalg.py | 41 del linalg_ops
|
H A D | linear_operator_low_rank_update.py | 25 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 D | linear_operator_lower_triangular.py | 24 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 D | varma.py | 57 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 D | kalman_filter.py | 29 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 D | level_trend.py | 28 from tensorflow.python.ops import linalg_ops namespace 91 identity_matrices = linalg_ops.eye(
|
/external/tensorflow/tensorflow/python/kernel_tests/ |
H A D | norm_op_test.py | 25 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 D | matrix_inverse_op_test.py | 27 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 D | self_adjoint_eig_op_test.py | 27 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 D | determinant_op_test.py | 28 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 D | linalg_grad_test.py | 27 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 D | matrix_solve_op_test.py | 29 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 D | matrix_solve_ls_op_test.py | 29 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 D | cholesky_op_test.py | 34 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 D | svd_op_test.py | 26 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 D | qr_op_test.py | 26 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 D | linalg_ops_test.py | 15 """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 D | mvn_full_covariance.py | 26 from tensorflow.python.ops import linalg_ops namespace 180 scale_tril = linalg_ops.cholesky(covariance_matrix)
|
/external/tensorflow/tensorflow/python/ops/ |
H A D | linalg_grad.py | 15 """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 D | linear_equations.py | 29 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 D | cholesky_op_test.py | 30 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 D | matrix_triangular_solve_op_test.py | 29 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 D | utils.py | 30 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 D | cholesky_outer_product.py | 28 from tensorflow.python.ops import linalg_ops namespace 118 else linalg_ops.cholesky(y))
|