Searched refs:GetMatrixDiagonal (Results 1 – 7 of 7) sorted by relevance
| /external/tensorflow/tensorflow/compiler/tf2xla/kernels/ |
| D | diag_op.cc | 154 xla::GetMatrixDiagonal(xla::Reshape(input, {new_size, new_size})), in Compile()
205 ctx->SetOutput(0, xla::GetMatrixDiagonal(input)); in Compile()
|
| /external/tensorflow/tensorflow/compiler/xla/client/lib/ |
| 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 | 41 XlaOp GetMatrixDiagonal(XlaOp x, int k = 0);
|
| D | matrix_test.cc | 70 GetMatrixDiagonal(a, kv.first); in TestMatrixDiagonal()
|
| D | svd.cc | 582 auto diag = GetMatrixDiagonal(w); in ComputeFrobeniusNorms()
741 auto d = GetMatrixDiagonal(result.d); in SortBySingularValuesAndPostProcessing()
769 result.d = Max(-GetMatrixDiagonal(GetTupleElement(sort_v_result, 0)), in SortBySingularValuesAndPostProcessing()
|
| D | matrix.cc | 49 XlaOp GetMatrixDiagonal(XlaOp x, int k) { in GetMatrixDiagonal() function
|
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | triangular_solve_expander.cc | 144 auto diags = GetMatrixDiagonal(diag_blocks); in InvertDiagonalBlocks()
|