Home
last modified time | relevance | path

Searched refs:inv_block (Results 1 – 1 of 1) sorted by relevance

/external/tensorflow/tensorflow/compiler/xla/service/
Dtriangular_solve_expander.cc287 auto inv_block = in SolveWithInvertedDiagonalBlocks() local
335 x_update = BatchDot(MaybeTransposeInMinorDims(inv_block, transpose_a), in SolveWithInvertedDiagonalBlocks()
339 MaybeTransposeInMinorDims(inv_block, transpose_a), in SolveWithInvertedDiagonalBlocks()