Searched defs:upper_diagonal (Results 1 – 3 of 3) sorted by relevance
53 XlaOp upper_diagonal, in CheckSystemAndReturnNumEquations()151 XlaOp upper_diagonal, in TridiagonalSolverImpl()176 auto upper_diagonal = values[1]; in TridiagonalSolverImpl() local285 XlaOp main_diagonal, XlaOp upper_diagonal, in TridiagonalSolver()316 auto upper_diagonal = in TridiagonalSolver() local
49 Array3D<float> upper_diagonal(batch_size, 1, num_eqs); in XLA_TEST_P() local
154 typename MatrixType::RealScalar *upper_diagonal, in upperbidiagonalization_blocked_helper()