Searched defs:superdiag (Results 1 – 7 of 7) sorted by relevance
| /external/tensorflow/tensorflow/core/kernels/linalg/ |
| D | tridiagonal_solve_op.cc | 115 const auto& superdiag = diagonals.row(0); in ComputeMatrix() local
229 OpKernelContext* context, const MatrixMapRow& superdiag, in SolveWithGaussianEliminationWithPivotingAndPerturbSingular()
335 const MatrixMapRow& superdiag, in SolveWithGaussianEliminationWithPivoting()
397 const MatrixMapRow& superdiag, in SolveWithThomasAlgorithm()
|
| D | tridiagonal_matmul_op_gpu.cu.cc | 38 const Scalar* __restrict__ superdiag, in TridiagonalMatMulKernel()
63 const Tensor& superdiag = context->input(0); in Compute() local
|
| D | tridiagonal_solve_op_gpu.cu.cc | 145 const auto& superdiag = diagonals.row(0); in ComputeMatrix() local
206 const Scalar* superdiag, const Scalar* diag, in SolveWithGtsv()
299 const Scalar* superdiag = lhs_data; in ComputeWithGtsvBatched() local
|
| D | tridiagonal_matmul_op.cc | 87 const auto& superdiag = inputs[0].row(0); in ComputeMatrix() local
|
| /external/tensorflow/tensorflow/compiler/tests/ |
| D | tridiagonal_matmul_ops_test.py | 45 superdiag, argument
102 def _makeTridiagonalMatrix(self, superdiag, maindiag, subdiag): argument
|
| /external/tensorflow/tensorflow/python/kernel_tests/linalg/ |
| D | tridiagonal_matmul_op_test.py | 41 superdiag, argument
93 def _makeTridiagonalMatrix(self, superdiag, maindiag, subdiag): argument
|
| /external/tensorflow/tensorflow/core/ops/ |
| D | linalg_ops.cc | 285 ShapeHandle superdiag; in TridiagonalMatMulShapeFn() local
|