Searched defs:upper_diagonal (Results 1 – 4 of 4) sorted by relevance
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/client/lib/ |
| H A D | tridiagonal.cc | 55 XlaOp upper_diagonal, in CheckSystemAndReturnNumEquations()
164 XlaOp upper_diagonal, XlaOp main_diagonal, XlaOp lower_diagonal, in CheckMatMulSystemAndReturnShapeParams()
238 XlaOp upper_diagonal, in TridiagonalSolverImpl()
263 auto upper_diagonal = values[1]; in TridiagonalSolverImpl() local
372 XlaOp main_diagonal, XlaOp upper_diagonal, in TridiagonalSolver()
403 auto upper_diagonal = in TridiagonalSolver() local
443 StatusOr<XlaOp> TridiagonalMatMul(XlaOp upper_diagonal, XlaOp main_diagonal, in TridiagonalMatMul()
|
| H A D | tridiagonal_test.cc | 42 Array3D<float> upper_diagonal{{{34, 35, 999}}}; in XLA_TEST_P() local
83 Array<float> upper_diagonal = Array<float>({5, 3, 7}, 1); in XLA_TEST_P() local
119 Array3D<float> upper_diagonal(batch_size, 1, num_eqs); in XLA_TEST_P() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/tf2xla/kernels/ |
| H A D | tridiagonal_ops.cc | 56 auto upper_diagonal = ctx->Input(0); in Compile() local
|
| /aosp_15_r20/external/eigen/Eigen/src/SVD/ |
| H A D | UpperBidiagonalization.h | 154 typename MatrixType::RealScalar *upper_diagonal, in upperbidiagonalization_blocked_helper()
|