Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/compiler/xla/client/lib/
Dmatrix.h51 XlaOp SetMatrixDiagonal(XlaOp matrix, XlaOp diag, int k = 0);
Dmatrix_test.cc102 GetMatrixDiagonal(SetMatrixDiagonal(a, b + ScalarLike(b, 1), kv.first), in TestSetMatrixDiagonal()
Dmatrix.cc172 XlaOp SetMatrixDiagonal(XlaOp matrix, XlaOp diag, int k) { in SetMatrixDiagonal() function