Searched defs:diagonals (Results 1 – 8 of 8) sorted by relevance
| /external/tensorflow/tensorflow/python/ops/linalg/ |
| D | linear_operator_tridiag.py | 130 diagonals, argument
226 def _shape_tensor(self, diagonals=None): argument
279 def _construct_adjoint_diagonals(self, diagonals): argument
381 def diagonals(self): member in LinearOperatorTridiag
|
| D | linalg_impl.py | 449 def tridiagonal_solve(diagonals, argument
621 def _tridiagonal_solve_compact_format(diagonals, rhs, transpose_rhs, argument
673 def tridiagonal_matmul(diagonals, rhs, diagonals_format='compact', name=None): argument
|
| /external/tensorflow/tensorflow/compiler/tf2xla/kernels/ |
| D | tridiagonal_ops.cc | 31 auto diagonals = ctx->Input(0); in Compile() local
|
| /external/tensorflow/tensorflow/core/kernels/linalg/ |
| D | tridiagonal_solve_op_gpu.cu.cc | 143 const auto diagonals = inputs[0]; in ComputeMatrix() local
229 void SolveForSizeOneOrTwo(OpKernelContext* context, const Scalar* diagonals, in SolveForSizeOneOrTwo()
|
| D | tridiagonal_solve_op.cc | 110 const auto diagonals = inputs[0]; in ComputeMatrix() local
|
| /external/tensorflow/tensorflow/compiler/xla/client/lib/ |
| D | tridiagonal.cc | 310 StatusOr<XlaOp> TridiagonalSolver(SolverAlgorithm algo, XlaOp diagonals, in TridiagonalSolver()
|
| /external/mesa3d/src/gallium/auxiliary/tessellator/ |
| D | tessellator.cpp | 1853 void CHWTessellator::StitchRegular(bool bTrapezoid,DIAGONALS diagonals, in StitchRegular()
|
| /external/mesa3d/src/gallium/drivers/swr/rasterizer/core/ |
| D | tessellator.cpp | 1839 void CHWTessellator::StitchRegular(bool bTrapezoid,DIAGONALS diagonals, in StitchRegular()
|