Searched refs:cusparse_solver (Results 1 – 1 of 1) sorted by relevance
| /external/tensorflow/tensorflow/core/kernels/linalg/ |
| D | tridiagonal_solve_op_gpu.cu.cc | 159 std::unique_ptr<GpuSparse> cusparse_solver(new GpuSparse(context)); in ComputeMatrix() local
160 OP_REQUIRES_OK(context, cusparse_solver->Initialize()); in ComputeMatrix()
164 SolveWithGtsv(context, cusparse_solver, superdiag.data(), diag.data(), in ComputeMatrix()
178 SolveWithGtsv(context, cusparse_solver, superdiag.data(), diag.data(), in ComputeMatrix()
199 std::unique_ptr<GpuSparse>& cusparse_solver, in SolveWithGtsv() argument
207 OP_REQUIRES_OK(context, (cusparse_solver.get()->*buffer_function)( in SolveWithGtsv()
218 OP_REQUIRES_OK(context, (cusparse_solver.get()->*solver_function)( in SolveWithGtsv()
310 std::unique_ptr<GpuSparse> cusparse_solver(new GpuSparse(context)); in ComputeWithGtsvBatched() local
312 OP_REQUIRES_OK(context, cusparse_solver->Initialize()); in ComputeWithGtsvBatched()
315 OP_REQUIRES_OK(context, cusparse_solver->Gtsv2StridedBatchBufferSizeExt( in ComputeWithGtsvBatched()
[all …]
|