Searched defs:subdiag (Results 1 – 9 of 9) sorted by relevance
| /aosp_15_r20/external/tensorflow/tensorflow/core/kernels/linalg/ |
| H A D | tridiagonal_solve_op.cc | 119 const auto& subdiag = diagonals.row(2); in ComputeMatrix() local
230 const MatrixMapRow& diag, const MatrixMapRow& subdiag, in SolveWithGaussianEliminationWithPivotingAndPerturbSingular()
337 const MatrixMapRow& subdiag, in SolveWithGaussianEliminationWithPivoting()
399 const MatrixMapRow& subdiag, in SolveWithThomasAlgorithm()
|
| H A D | tridiagonal_solve_op_gpu.cu.cc | 149 const auto& subdiag = diagonals.row(2); in ComputeMatrix() local
207 const Scalar* subdiag, Scalar* rhs, const int num_eqs, in SolveWithGtsv()
301 const Scalar* subdiag = lhs_data + 2 * matrix_size * batch_size; in ComputeWithGtsvBatched() local
|
| H A D | tridiagonal_matmul_op_gpu.cu.cc | 40 const Scalar* __restrict__ subdiag, in TridiagonalMatMulKernel()
65 const Tensor& subdiag = context->input(2); in Compute() local
|
| H A D | tridiagonal_matmul_op.cc | 94 const auto& subdiag = inputs[2].row(0); in ComputeMatrix() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/tests/ |
| H A D | tridiagonal_matmul_ops_test.py | 47 subdiag, argument
102 def _makeTridiagonalMatrix(self, superdiag, maindiag, subdiag): argument
|
| /aosp_15_r20/external/tensorflow/tensorflow/python/kernel_tests/linalg/ |
| H A D | tridiagonal_matmul_op_test.py | 43 subdiag, argument
93 def _makeTridiagonalMatrix(self, superdiag, maindiag, subdiag): argument
|
| /aosp_15_r20/external/eigen/doc/snippets/ |
| H A D | Tridiagonalization_diagonal.cpp | 12 VectorXd subdiag = triOfA.subDiagonal(); variable
|
| H A D | Tridiagonalization_decomposeInPlace.cpp | 6 VectorXd subdiag(4); variable
|
| /aosp_15_r20/external/tensorflow/tensorflow/core/ops/ |
| H A D | linalg_ops.cc | 287 ShapeHandle subdiag; in TridiagonalMatMulShapeFn() local
|