Searched refs:gramian (Results 1 – 4 of 4) sorted by relevance
| /external/tensorflow/tensorflow/core/kernels/ |
| D | matrix_solve_ls_op_impl.h | 109 Matrix gramian(cols, cols); in ComputeMatrix()
110 gramian.template triangularView<Eigen::Lower>() = in ComputeMatrix()
113 gramian += in ComputeMatrix()
116 const Eigen::LLT<Eigen::Ref<Matrix>, Eigen::Lower> llt(gramian); in ComputeMatrix()
130 Matrix gramian(rows, rows); in ComputeMatrix()
131 gramian.template triangularView<Eigen::Lower>() = in ComputeMatrix()
134 gramian += in ComputeMatrix()
137 const Eigen::LLT<Eigen::Ref<Matrix>, Eigen::Lower> llt(gramian); in ComputeMatrix()
|
| /external/tensorflow/tensorflow/contrib/factorization/python/ops/ |
| D | factorization_ops.py | 454 def _prepare_gramian(self, factors, gramian): argument
470 with ops.colocate_with(gramian):
471 prep_gramian = state_ops.assign(gramian,
855 gramian = self._col_gramian_cache
865 gramian = self._row_gramian_cache
907 total_lhs = (self._unobserved_weight * gramian)
995 math_ops.trace(math_ops.matmul(partial_row_gramian, gramian))) +
1002 math_ops.trace(gramian))
|
| /external/tensorflow/tensorflow/python/kernel_tests/ |
| D | matrix_solve_ls_op_test.py | 69 gramian = np.dot(np.conj(matrix.T), matrix) + preconditioner
71 return np.linalg.solve(gramian, rhs)
74 gramian = np.dot(matrix, np.conj(matrix.T)) + preconditioner
75 z = np.linalg.solve(gramian, rhs)
|
| /external/tensorflow/tensorflow/python/ops/ |
| D | linalg_ops.py | 65 gramian = math_ops.matmul(
78 gramian += l2_regularizer * identity
79 return gen_linalg_ops.cholesky(gramian)
|