Searched refs:main_diagonal_xla (Results 1 – 1 of 1) sorted by relevance
60 XlaOp main_diagonal_xla; in XLA_TEST_P() local67 main_diagonal, 1, "main_diagonal", &builder, &main_diagonal_xla); in XLA_TEST_P()73 ThomasSolver(lower_diagonal_xla, main_diagonal_xla, in XLA_TEST_P()84 auto b_i = Coefficient(main_diagonal_xla, i); in XLA_TEST_P()