Searched refs:UpdateEq (Results 1 – 1 of 1) sorted by relevance
| /external/tensorflow/tensorflow/compiler/xla/client/lib/ |
| D | tridiagonal.cc | 122 XlaOp UpdateEq(XlaOp updated, int32 i, XlaOp update) { in UpdateEq() function
127 XlaOp UpdateEq(XlaOp updated, XlaOp i, XlaOp update) { in UpdateEq() 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()
204 main_diag_after_elimination = UpdateEq( in ThomasSolver()
210 UpdateEq(rhs_after_elimination, i, in ThomasSolver()
238 UpdateEq(x_coeffs, num_eqs - 1, in ThomasSolver()
255 x_coeffs = UpdateEq(x_coeffs, i, in ThomasSolver()
|