Searched refs:x_update (Results 1 – 1 of 1) sorted by relevance
330 XlaOp x_update; in SolveWithInvertedDiagonalBlocks() local335 x_update = BatchDot(MaybeTransposeInMinorDims(inv_block, transpose_a), in SolveWithInvertedDiagonalBlocks()338 x_update = BatchDot(remainder, in SolveWithInvertedDiagonalBlocks()343 x = DynamicUpdateSliceInMinorDims(x, x_update, /*starts=*/update_starts); in SolveWithInvertedDiagonalBlocks()