Searched refs:x_coeffs (Results 1 – 1 of 1) sorted by relevance
| /external/tensorflow/tensorflow/compiler/xla/client/lib/ |
| D | tridiagonal.cc | 156 XlaOp x_coeffs = ZerosLike(rhs); in ThomasSolver() local
237 x_coeffs = in ThomasSolver()
238 UpdateEq(x_coeffs, num_eqs - 1, in ThomasSolver()
244 auto x_coeffs = values[0]; in ThomasSolver() local
255 x_coeffs = UpdateEq(x_coeffs, i, in ThomasSolver()
258 Coefficient(x_coeffs, i + one)) / in ThomasSolver()
260 return std::vector<XlaOp>{x_coeffs, rhs_after_elimination, in ThomasSolver()
268 {x_coeffs, rhs_after_elimination, upper_diagonal_coeffs, in ThomasSolver()
271 x_coeffs = values_after_bwd_reduction[0]; in ThomasSolver()
273 return x_coeffs; in ThomasSolver()
|