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()
|