Searched refs:ThomasSolver (Results 1 – 4 of 4) sorted by relevance
| /external/tensorflow/tensorflow/compiler/xla/client/lib/ |
| D | tridiagonal.h | 25 StatusOr<XlaOp> ThomasSolver(XlaOp lower_diagonal, XlaOp main_diagonal,
28 StatusOr<XlaOp> ThomasSolver(XlaOp diagonals, XlaOp rhs);
|
| D | tridiagonal.cc | 145 StatusOr<XlaOp> ThomasSolver(XlaOp lower_diagonal, XlaOp main_diagonal, in ThomasSolver() function
289 StatusOr<XlaOp> ThomasSolver(XlaOp diagonals, XlaOp rhs) { in ThomasSolver() function
312 TF_ASSIGN_OR_RETURN(XlaOp x, ThomasSolver(lower_diagonal, main_diagonal, in ThomasSolver()
|
| D | tridiagonal_test.cc | 73 ThomasSolver(lower_diagonal_xla, main_diagonal_xla, in XLA_TEST_P()
|
| /external/tensorflow/tensorflow/compiler/tf2xla/kernels/ |
| D | tridiagonal_ops.cc | 42 auto result = xla::tridiagonal::ThomasSolver(diagonals, rhs); in Compile()
|