D | cuda_solvers.cc | 631 cusolverEigMode_t jobz, cublasFillMode_t uplo, in HeevdImpl() argument 638 TF_RETURN_IF_CUSOLVER_ERROR(bufsize(cusolver_dn_handle, jobz, uplo, n, in HeevdImpl() 646 solver(cusolver_dn_handle, jobz, uplo, n, CUDAComplex(dev_A), lda, in HeevdImpl() 654 Status CudaSolver::Heevd(cusolverEigMode_t jobz, cublasFillMode_t uplo, \ 660 context_, cusolver_dn_handle_, jobz, uplo, n, dev_A, lda, \ 709 cusolverEigMode_t jobz, int m, int n, in GesvdjBatchedImpl() argument 720 cusolver_dn_handle, jobz, m, n, CUDAComplex(A), lda, S, CUDAComplex(U), in GesvdjBatchedImpl() 726 cusolver_dn_handle, jobz, m, n, CUDAComplex(A), lda, S, CUDAComplex(U), in GesvdjBatchedImpl() 736 cusolverEigMode_t jobz, int m, int n, Scalar* dev_A, int lda, \ 741 context_, cusolver_dn_handle_, jobz, m, n, dev_A, \
|