| D | tridiagonal.cc | 111 XlaOp Coefficient(XlaOp operand, int32 i) { in Coefficient() function
117 XlaOp Coefficient(XlaOp operand, XlaOp i) { in Coefficient() function
160 UpdateEq(main_diag_after_elimination, 0, Coefficient(main_diagonal, 0)); in ThomasSolver()
164 UpdateEq(rhs_after_elimination, 0, Coefficient(rhs, 0)); in ThomasSolver()
173 UpdateEq(upper_diagonal_coeffs, i, Coefficient(upper_diagonal, i)); in ThomasSolver()
195 auto lower_diagonal_i = Coefficient(lower_diagonal, i); in ThomasSolver()
196 auto main_diagonal_i = Coefficient(main_diagonal, i); in ThomasSolver()
197 auto rhs_i = Coefficient(rhs, i); in ThomasSolver()
200 lower_diagonal_i / Coefficient(main_diag_after_elimination, i - one); in ThomasSolver()
206 main_diagonal_i - w_i * Coefficient(upper_diagonal_coeffs, i - one)); in ThomasSolver()
[all …]
|