Searched refs:DiagonalBlocks (Results 1 – 1 of 1) sorted by relevance
39 XlaOp DiagonalBlocks(XlaOp a, int64 block_size) { in DiagonalBlocks() function430 auto diag_blocks = DiagonalBlocks(a, block_size); in BuildTriangularSolve()