Home
last modified time | relevance | path

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

/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/client/lib/
H A Dtridiagonal.h25 enum SolverAlgorithm { kThomas }; enum
27 StatusOr<XlaOp> TridiagonalSolver(SolverAlgorithm algo, XlaOp lower_diagonal,
31 StatusOr<XlaOp> TridiagonalSolver(SolverAlgorithm algo, XlaOp diagonals,
H A Dtridiagonal.cc219 template <SolverAlgorithm algo>
371 StatusOr<XlaOp> TridiagonalSolver(SolverAlgorithm algo, XlaOp lower_diagonal, in TridiagonalSolver()
397 StatusOr<XlaOp> TridiagonalSolver(SolverAlgorithm algo, XlaOp diagonals, in TridiagonalSolver()