Searched refs:upper_diagonal_coeffs (Results 1 – 1 of 1) sorted by relevance
155 XlaOp upper_diagonal_coeffs = ZerosLike(upper_diagonal); in ThomasSolver() local169 auto upper_diagonal_coeffs = values[0]; in ThomasSolver() local172 upper_diagonal_coeffs = in ThomasSolver()173 UpdateEq(upper_diagonal_coeffs, i, Coefficient(upper_diagonal, i)); in ThomasSolver()174 return std::vector<XlaOp>{upper_diagonal_coeffs, upper_diagonal}; in ThomasSolver()178 {upper_diagonal_coeffs, upper_diagonal}, in ThomasSolver()180 upper_diagonal_coeffs = values_after_preparation[0]; in ThomasSolver()190 auto upper_diagonal_coeffs = values[4]; in ThomasSolver() local206 main_diagonal_i - w_i * Coefficient(upper_diagonal_coeffs, i - one)); in ThomasSolver()217 upper_diagonal_coeffs, in ThomasSolver()[all …]