Home
last modified time | relevance | path

Searched defs:inv_diag_blocks (Results 1 – 1 of 1) sorted by relevance

/external/tensorflow/tensorflow/compiler/xla/service/
Dtriangular_solve_expander.cc134 XlaOp SolveWithInvertedDiagonalBlocks(XlaOp a, XlaOp b, XlaOp inv_diag_blocks, in SolveWithInvertedDiagonalBlocks()
362 auto inv_diag_blocks = GetTupleElement(invert_while, 1); in InvertDiagonalBlocks() local
400 auto inv_diag_blocks = InvertDiagonalBlocks(diag_blocks, lower, precision); in SolveByInvertingDiagonalBlocks() local