Searched defs:main_diagonal (Results 1 – 2 of 2) sorted by relevance
51 XlaOp main_diagonal, in CheckSystemAndReturnNumEquations()145 StatusOr<XlaOp> ThomasSolver(XlaOp lower_diagonal, XlaOp main_diagonal, in ThomasSolver()187 auto main_diagonal = values[1]; in ThomasSolver() local297 auto main_diagonal = in ThomasSolver() local
48 Array3D<float> main_diagonal(batch_size, 1, num_eqs); in XLA_TEST_P() local