Searched refs:GetMatrixDiagonal (Results 1 – 7 of 7) sorted by relevance
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | eigh_expander.cc | 105 w_tl = GetMatrixDiagonal(Real(w_tl)); in HermitianEigenDecomposition2x2()
106 w_tr = GetMatrixDiagonal(w_tr); in HermitianEigenDecomposition2x2()
107 w_br = GetMatrixDiagonal(Real(w_br)); in HermitianEigenDecomposition2x2()
371 w = GetMatrixDiagonal(GetTupleElement(sort_result, 0)); in SortByEigenvalues()
466 return Tuple(builder, {FullLike(a, 1), GetMatrixDiagonal(Real(a))}); in BuildEigh()
519 builder, {GetMatrixDiagonal(Real(tl)), GetMatrixDiagonal(Real(br))}, in BuildEigh()
|
| D | triangular_solve_expander.cc | 257 auto diags = GetMatrixDiagonal(diag_blocks); in InvertDiagonalBlocks()
|
| /external/tensorflow/tensorflow/compiler/xla/client/lib/ |
| D | matrix_test.cc | 180 GetMatrixDiagonal(a, kv.first); in TestMatrixDiagonal()
198 GetMatrixDiagonal(SetMatrixDiagonal(a, b + ScalarLike(b, 1), kv.first), in TestSetMatrixDiagonal()
241 GetMatrixDiagonal(a, kv.first); in TestMatrixDiagonal4D()
|
| D | matrix.h | 49 XlaOp GetMatrixDiagonal(XlaOp x, int k = 0);
|
| D | svd.cc | 573 auto diag = GetMatrixDiagonal(w_sliced); in ComputeToleranceComparison()
746 auto d = GetMatrixDiagonal(result.d); in SortBySingularValuesAndPostProcessing()
772 result.d = GetMatrixDiagonal(GetTupleElement(sort_v_result, 0)); in SortBySingularValuesAndPostProcessing()
|
| D | matrix.cc | 76 XlaOp GetMatrixDiagonal(XlaOp x, int k) { in GetMatrixDiagonal() function
|
| /external/tensorflow/tensorflow/compiler/tf2xla/kernels/ |
| D | matrix_diag_ops.cc | 408 : xla::GetMatrixDiagonal(input, upper_diag_index)); in Compile()
415 : xla::GetMatrixDiagonal(input, diag_index); in Compile()
|