Home
last modified time | relevance | path

Searched refs:svdj_info (Results 1 – 1 of 1) sorted by relevance

/external/tensorflow/tensorflow/core/util/
Dcuda_solvers.cc717 gesvdjInfo_t svdj_info; in GesvdjBatchedImpl() local
718 TF_RETURN_IF_CUSOLVER_ERROR(cusolverDnCreateGesvdjInfo(&svdj_info)); in GesvdjBatchedImpl()
721 ldu, CUDAComplex(V), ldv, &lwork, svdj_info, batch_size)); in GesvdjBatchedImpl()
728 lwork, dev_lapack_info, svdj_info, batch_size)); in GesvdjBatchedImpl()
729 TF_RETURN_IF_CUSOLVER_ERROR(cusolverDnDestroyGesvdjInfo(svdj_info)); in GesvdjBatchedImpl()