Searched refs:TriangleMask (Results 1 – 4 of 4) sorted by relevance
| /external/tensorflow/tensorflow/compiler/xla/client/lib/ |
| D | matrix.h | 55 XlaOp TriangleMask(XlaOp x, int diagonal);
|
| D | matrix.cc | 211 XlaOp TriangleMask(XlaOp x, int diagonal) { in TriangleMask() function
230 return lower ? Select(TriangleMask(x, 0), x, ZerosLike(x)) in Triangle()
231 : Select(TriangleMask(x, -1), ZerosLike(x), x); in Triangle()
|
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | qr_expander.cc | 328 vtv = Select(TriangleMask(vtv, 0), ZerosLike(vtv), vtv); in CompactWYRepresentation()
383 Select(TriangleMask(qr_block.a, -1), qr_block.a, ZerosLike(qr_block.a)), in BuildQrDecomposition()
|
| D | triangular_solve_expander.cc | 386 a = lower ? Select(TriangleMask(a, -1), a, ZerosLike(a)) in SolveByInvertingDiagonalBlocks()
387 : Select(TriangleMask(a, 0), ZerosLike(a), a); in SolveByInvertingDiagonalBlocks()
|