Searched refs:inv_diag_blocks (Results 1 – 1 of 1) sorted by relevance
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | triangular_solve_expander.cc | 134 XlaOp SolveWithInvertedDiagonalBlocks(XlaOp a, XlaOp b, XlaOp inv_diag_blocks, in SolveWithInvertedDiagonalBlocks() argument
140 TF_ASSIGN_OR_RETURN(Shape blocks_shape, builder->GetShape(inv_diag_blocks)); in SolveWithInvertedDiagonalBlocks()
175 MaybeConjugate(Collapse(SliceInMinorDims(inv_diag_blocks, {j, 0, 0}, in SolveWithInvertedDiagonalBlocks()
362 auto inv_diag_blocks = GetTupleElement(invert_while, 1); in InvertDiagonalBlocks() local
364 inv_diag_blocks = Div(inv_diag_blocks, diags, in InvertDiagonalBlocks()
368 return Reshape(inv_diag_blocks, AsInt64Slice(shape.dimensions())); in InvertDiagonalBlocks()
400 auto inv_diag_blocks = InvertDiagonalBlocks(diag_blocks, lower, precision); in SolveByInvertingDiagonalBlocks() local
403 return SolveWithInvertedDiagonalBlocks(a, b, inv_diag_blocks, left_side, in SolveByInvertingDiagonalBlocks()
|