Searched refs:SolverAlgorithm (Results 1 – 2 of 2) sorted by relevance
25 enum SolverAlgorithm { kThomas }; enum27 StatusOr<XlaOp> TridiagonalSolver(SolverAlgorithm algo, XlaOp lower_diagonal,31 StatusOr<XlaOp> TridiagonalSolver(SolverAlgorithm algo, XlaOp diagonals,
132 template <SolverAlgorithm algo>284 StatusOr<XlaOp> TridiagonalSolver(SolverAlgorithm algo, XlaOp lower_diagonal, in TridiagonalSolver()310 StatusOr<XlaOp> TridiagonalSolver(SolverAlgorithm algo, XlaOp diagonals, in TridiagonalSolver()