Searched defs:subdiag (Results 1 – 8 of 8) sorted by relevance
| /external/tensorflow/tensorflow/core/kernels/linalg/ |
| D | tridiagonal_matmul_op_gpu.cu.cc | 40 const Scalar* __restrict__ subdiag, in TridiagonalMatMulKernel()
65 const Tensor& subdiag = context->input(2); in Compute() local
|
| D | tridiagonal_solve_op.cc | 109 const auto& subdiag = diagonals.row(2); in ComputeMatrix() local
141 const MatrixMapRow& subdiag, in SolveWithGaussianEliminationWithPivoting()
197 const MatrixMapRow& subdiag, in SolveWithThomasAlgorithm()
|
| D | tridiagonal_solve_op_gpu.cu.cc | 143 const auto& subdiag = diagonals.row(2); in ComputeMatrix() local
201 const Scalar* subdiag, Scalar* rhs, const int num_eqs, in SolveWithGtsv()
300 const Scalar* subdiag = lhs_data + 2 * matrix_size * batch_size; in ComputeWithGtsvBatched() local
|
| D | tridiagonal_matmul_op.cc | 94 const auto& subdiag = inputs[2].row(0); in ComputeMatrix() local
|
| /external/eigen/doc/snippets/ |
| D | Tridiagonalization_decomposeInPlace.cpp | 6 VectorXd subdiag(4); variable
|
| D | Tridiagonalization_diagonal.cpp | 12 VectorXd subdiag = triOfA.subDiagonal(); variable
|
| /external/tensorflow/tensorflow/python/kernel_tests/ |
| D | tridiagonal_matmul_op_test.py | 44 subdiag, argument
94 def _makeTridiagonalMatrix(self, superdiag, maindiag, subdiag): argument
|
| /external/tensorflow/tensorflow/core/ops/ |
| D | linalg_ops.cc | 287 ShapeHandle subdiag; in TridiagonalMatMulShapeFn() local
|