Searched defs:diag_blocks (Results 1 – 1 of 1) sorted by relevance
50 XlaOp diag_blocks; in DiagonalBlocks() local239 XlaOp diag_blocks, bool lower_triangular, in InvertDiagonalBlocks()390 auto diag_blocks = DiagonalBlocks(a, block_size); in SolveByInvertingDiagonalBlocks() local