Searched defs:upper_diagonal (Results 1 – 3 of 3) sorted by relevance
52 XlaOp upper_diagonal, in CheckSystemAndReturnNumEquations()146 XlaOp upper_diagonal, XlaOp rhs) { in ThomasSolver()170 auto upper_diagonal = values[1]; in ThomasSolver() local294 auto upper_diagonal = in ThomasSolver() 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()