Searched refs:GetMatrixDiagonal (Results 1 – 8 of 8) sorted by relevance
| /external/tensorflow/tensorflow/compiler/xla/client/lib/ |
| D | matrix_test.cc | 84 GetMatrixDiagonal(a, kv.first); in TestMatrixDiagonal()
102 GetMatrixDiagonal(SetMatrixDiagonal(a, b + ScalarLike(b, 1), kv.first), in TestSetMatrixDiagonal()
145 GetMatrixDiagonal(a, kv.first); in TestMatrixDiagonal4D()
|
| D | self_adjoint_eig.cc | 203 auto diag = GetMatrixDiagonal(w); in ComputeFrobeniusNorms()
356 result.w = GetMatrixDiagonal(GetTupleElement(sort_result, 0)); in SortByEigenvalues()
461 result.w = GetMatrixDiagonal(output[2]); in SelfAdjointEig()
|
| D | matrix.h | 47 XlaOp GetMatrixDiagonal(XlaOp x, int k = 0);
|
| D | svd.cc | 572 auto diag = GetMatrixDiagonal(w_sliced); in ComputeToleranceComparison()
731 auto d = GetMatrixDiagonal(result.d); in SortBySingularValuesAndPostProcessing()
757 result.d = GetMatrixDiagonal(GetTupleElement(sort_v_result, 0)); in SortBySingularValuesAndPostProcessing()
|
| D | matrix.cc | 75 XlaOp GetMatrixDiagonal(XlaOp x, int k) { in GetMatrixDiagonal() function
|
| /external/tensorflow/tensorflow/compiler/tf2xla/kernels/ |
| D | diag_op.cc | 153 : xla::GetMatrixDiagonal(reshape_input), in Compile()
|
| D | matrix_diag_ops.cc | 405 : xla::GetMatrixDiagonal(input, upper_diag_index)); in Compile()
412 : xla::GetMatrixDiagonal(input, diag_index); in Compile()
|
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | triangular_solve_expander.cc | 264 auto diags = GetMatrixDiagonal(diag_blocks); in InvertDiagonalBlocks()
|