def _cholesky_diag(diag_operator): return linear_operator_diag.LinearOperatorDiag(math_ops.sqrt( diag_operator.diag), is_non_singular=True, is_self_adjoint=True, is_positive_definite=True, is_square=True)
def _cholesky_scaled_identity(identity_operator): return linear_operator_identity.LinearOperatorScaledIdentity( num_rows=identity_operator._num_rows, # pylint: disable=protected-access multiplier=math_ops.sqrt(identity_operator.multiplier), is_non_singular=True, is_self_adjoint=True, is_positive_definite=True, is_square=True)