Searched defs:inv_diag_blocks (Results 1 – 1 of 1) sorted by relevance
134 XlaOp SolveWithInvertedDiagonalBlocks(XlaOp a, XlaOp b, XlaOp inv_diag_blocks, in SolveWithInvertedDiagonalBlocks()362 auto inv_diag_blocks = GetTupleElement(invert_while, 1); in InvertDiagonalBlocks() local400 auto inv_diag_blocks = InvertDiagonalBlocks(diag_blocks, lower, precision); in SolveByInvertingDiagonalBlocks() local