Searched refs:upper_diagonal (Results 1 – 4 of 4) sorted by relevance
| /external/tensorflow/tensorflow/compiler/xla/client/lib/ |
| D | tridiagonal.cc | 52 XlaOp upper_diagonal, in CheckSystemAndReturnNumEquations() argument
61 builder->GetShape(upper_diagonal)); in CheckSystemAndReturnNumEquations()
146 XlaOp upper_diagonal, XlaOp rhs) { in ThomasSolver() argument
151 lower_diagonal, main_diagonal, upper_diagonal, rhs)); in ThomasSolver()
155 XlaOp upper_diagonal_coeffs = ZerosLike(upper_diagonal); in ThomasSolver()
170 auto upper_diagonal = values[1]; in ThomasSolver() local
173 UpdateEq(upper_diagonal_coeffs, i, Coefficient(upper_diagonal, i)); in ThomasSolver()
174 return std::vector<XlaOp>{upper_diagonal_coeffs, upper_diagonal}; in ThomasSolver()
178 {upper_diagonal_coeffs, upper_diagonal}, in ThomasSolver()
294 auto upper_diagonal = in ThomasSolver() local
[all …]
|
| D | tridiagonal_test.cc | 49 Array3D<float> upper_diagonal(batch_size, 1, num_eqs); in XLA_TEST_P() local
55 upper_diagonal.FillRandom(1.0, /*mean=*/0.0, in XLA_TEST_P()
69 upper_diagonal, 2, "upper_diagonal", &builder, &upper_diagonal_xla); in XLA_TEST_P()
|
| D | tridiagonal.h | 26 XlaOp upper_diagonal, XlaOp rhs);
|
| /external/eigen/Eigen/src/SVD/ |
| D | UpperBidiagonalization.h | 95 typename MatrixType::RealScalar *upper_diagonal,
127 .makeHouseholderInPlace(mat.coeffRef(k,k+1), upper_diagonal[k]);
154 typename MatrixType::RealScalar *upper_diagonal, in upperbidiagonalization_blocked_helper() argument
224 u_k.makeHouseholderInPlace(tau_u, upper_diagonal[k]); in upperbidiagonalization_blocked_helper()
|