Searched refs:TriangleMask (Results 1 – 3 of 3) sorted by relevance
| /external/tensorflow/tensorflow/compiler/xla/client/lib/ |
| D | matrix.cc | 91 XlaOp TriangleMask(XlaOp x, int diagonal) { in TriangleMask() function
110 return lower ? Select(TriangleMask(x, 0), x, ZerosLike(x)) in Triangle()
111 : Select(TriangleMask(x, -1), ZerosLike(x), x); in Triangle()
|
| D | matrix.h | 45 XlaOp TriangleMask(XlaOp x, int diagonal);
|
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | triangular_solve_expander.cc | 419 a = lower ? Select(TriangleMask(a, -1), a, ZerosLike(a)) in BuildTriangularSolve()
420 : Select(TriangleMask(a, 0), ZerosLike(a), a); in BuildTriangularSolve()
|