Searched defs:lower_diagonal (Results 1 – 2 of 2) sorted by relevance
50 StatusOr<int64> CheckSystemAndReturnNumEquations(XlaOp lower_diagonal, in CheckSystemAndReturnNumEquations()145 StatusOr<XlaOp> ThomasSolver(XlaOp lower_diagonal, XlaOp main_diagonal, in ThomasSolver()186 auto lower_diagonal = values[0]; in ThomasSolver() local300 auto lower_diagonal = in ThomasSolver() local
47 Array3D<float> lower_diagonal(batch_size, 1, num_eqs); in XLA_TEST_P() local