Home
last modified time | relevance | path

Searched refs:rhs_after_elimination (Results 1 – 1 of 1) sorted by relevance

/external/tensorflow/tensorflow/compiler/xla/client/lib/
Dtridiagonal.cc154 XlaOp rhs_after_elimination = ZerosLike(rhs); in ThomasSolver() local
163 rhs_after_elimination = in ThomasSolver()
164 UpdateEq(rhs_after_elimination, 0, Coefficient(rhs, 0)); in ThomasSolver()
191 auto rhs_after_elimination = values[5]; in ThomasSolver() local
209 rhs_after_elimination = in ThomasSolver()
210 UpdateEq(rhs_after_elimination, i, in ThomasSolver()
211 rhs_i - w_i * Coefficient(rhs_after_elimination, i - one)); in ThomasSolver()
218 rhs_after_elimination}; in ThomasSolver()
225 upper_diagonal_coeffs, rhs_after_elimination}, in ThomasSolver()
232 rhs_after_elimination = values_after_fwd_transformation[5]; in ThomasSolver()
[all …]