Searched defs:cublas_handle (Results 1 – 1 of 1) sorted by relevance
/external/tensorflow/tensorflow/core/util/ |
D | cuda_solvers.cc | 132 cublasHandle_t cublas_handle; member 346 static inline Status GeamImpl(SolverFnT solver, cublasHandle_t cublas_handle, in GeamImpl() 759 cublasHandle_t cublas_handle, int n, in GetrfBatchedImpl() 795 cublasHandle_t cublas_handle, cublasOperation_t trans, int n, int nrhs, in GetrsBatchedImpl() 842 cublasHandle_t cublas_handle, int n, const Scalar* const host_a_dev_ptrs[], in GetriBatchedImpl() 885 cublasHandle_t cublas_handle, int n, const Scalar* const host_a_dev_ptrs[], in MatInvBatchedImpl() 925 static inline Status TrsmImpl(SolverFnT solver, cublasHandle_t cublas_handle, in TrsmImpl() 954 static inline Status TrsvImpl(SolverFnT solver, cublasHandle_t cublas_handle, in TrsvImpl() 980 cublasHandle_t cublas_handle, cublasSideMode_t side, cublasFillMode_t uplo, in TrsmBatchedImpl()
|