Searched defs:main_diagonal (Results 1 – 2 of 2) sorted by relevance
52 XlaOp main_diagonal, in CheckSystemAndReturnNumEquations()150 XlaOp main_diagonal, in TridiagonalSolverImpl()193 auto main_diagonal = values[1]; in TridiagonalSolverImpl() local285 XlaOp main_diagonal, XlaOp upper_diagonal, in TridiagonalSolver()319 auto main_diagonal = in TridiagonalSolver() local
48 Array3D<float> main_diagonal(batch_size, 1, num_eqs); in XLA_TEST_P() local