Searched defs:diag_blocks (Results 1 – 1 of 1) sorted by relevance
47 XlaOp diag_blocks; in DiagonalBlocks() local124 XlaOp InvertDiagonalBlocks(XlaOp diag_blocks, bool lower, bool transpose_a, in InvertDiagonalBlocks()430 auto diag_blocks = DiagonalBlocks(a, block_size); in BuildTriangularSolve() local