Searched refs:main_diag_after_elimination (Results 1 – 1 of 1) sorted by relevance
| /external/tensorflow/tensorflow/compiler/xla/client/lib/ |
| D | tridiagonal.cc | 153 XlaOp main_diag_after_elimination = ZerosLike(main_diagonal); in ThomasSolver() local
159 main_diag_after_elimination = in ThomasSolver()
160 UpdateEq(main_diag_after_elimination, 0, Coefficient(main_diagonal, 0)); in ThomasSolver()
189 auto main_diag_after_elimination = values[3]; in ThomasSolver() local
200 lower_diagonal_i / Coefficient(main_diag_after_elimination, i - one); in ThomasSolver()
204 main_diag_after_elimination = UpdateEq( in ThomasSolver()
205 main_diag_after_elimination, i, in ThomasSolver()
216 main_diag_after_elimination, in ThomasSolver()
224 {lower_diagonal, main_diagonal, rhs, main_diag_after_elimination, in ThomasSolver()
230 main_diag_after_elimination = values_after_fwd_transformation[3]; in ThomasSolver()
[all …]
|