Searched defs:lower_diagonal (Results 1 – 2 of 2) sorted by relevance
51 StatusOr<int64> CheckSystemAndReturnNumEquations(XlaOp lower_diagonal, in CheckSystemAndReturnNumEquations()149 StatusOr<XlaOp> TridiagonalSolverImpl<kThomas>(XlaOp lower_diagonal, in TridiagonalSolverImpl()192 auto lower_diagonal = values[0]; in TridiagonalSolverImpl() local284 StatusOr<XlaOp> TridiagonalSolver(SolverAlgorithm algo, XlaOp lower_diagonal, in TridiagonalSolver()322 auto lower_diagonal = in TridiagonalSolver() local
47 Array3D<float> lower_diagonal(batch_size, 1, num_eqs); in XLA_TEST_P() local