Searched refs:SetMatrixDiagonal (Results 1 – 4 of 4) sorted by relevance
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | eigh_expander.cc | 246 w_tl = SetMatrixDiagonal(w_tl, rotation.rt1); in ApplyRotations()
247 w_tr = SetMatrixDiagonal(w_tr, ZerosLike(rotation.rt1)); in ApplyRotations()
248 w_bl = SetMatrixDiagonal(w_bl, ZerosLike(rotation.rt1)); in ApplyRotations()
249 w_br = SetMatrixDiagonal(w_br, rotation.rt2); in ApplyRotations()
|
| /external/tensorflow/tensorflow/compiler/xla/client/lib/ |
| D | matrix.h | 53 XlaOp SetMatrixDiagonal(XlaOp matrix, XlaOp diag, int k = 0);
|
| D | matrix_test.cc | 198 GetMatrixDiagonal(SetMatrixDiagonal(a, b + ScalarLike(b, 1), kv.first), in TestSetMatrixDiagonal()
|
| D | matrix.cc | 173 XlaOp SetMatrixDiagonal(XlaOp matrix, XlaOp diag, int k) { in SetMatrixDiagonal() function
|