Searched refs:cusolver_dn_handle (Results 1 – 1 of 1) sorted by relevance
| /external/tensorflow/tensorflow/core/util/ |
| D | cuda_solvers.cc | 115 CHECK(cusolverDnCreate(&cusolver_dn_handle) == CUSOLVER_STATUS_SUCCESS) in CudaSolverHandles()
117 CHECK(cusolverDnSetStream(cusolver_dn_handle, stream) == in CudaSolverHandles()
129 CHECK(cusolverDnDestroy(cusolver_dn_handle) == CUSOLVER_STATUS_SUCCESS) in ~CudaSolverHandles()
133 cusolverDnHandle_t cusolver_dn_handle; member
189 cusolver_dn_handle_ = it->second->cusolver_dn_handle; in CudaSolver()
382 cusolverDnHandle_t cusolver_dn_handle, in PotrfImpl() argument
389 bufsize(cusolver_dn_handle, uplo, n, CUDAComplex(A), lda, &lwork)); in PotrfImpl()
395 cusolver_dn_handle, uplo, n, CUDAComplex(A), lda, in PotrfImpl()
415 cusolverDnHandle_t cusolver_dn_handle, cublasFillMode_t uplo, int n, in PotrfBatchedImpl() argument
428 solver(cusolver_dn_handle, uplo, n, in PotrfBatchedImpl()
[all …]
|