Searched refs:cublas_handle (Results 1 – 1 of 1) sorted by relevance
| /external/tensorflow/tensorflow/core/kernels/ |
| D | cuda_solvers.cc | 99 CHECK(cublasCreate(&cublas_handle) == CUBLAS_STATUS_SUCCESS) in CudaSolverHandles()
101 CHECK(cublasSetStream(cublas_handle, stream) == CUBLAS_STATUS_SUCCESS) in CudaSolverHandles()
106 CHECK(cublasDestroy(cublas_handle) == CUBLAS_STATUS_SUCCESS) in ~CudaSolverHandles()
111 cublasHandle_t cublas_handle; member
169 cublas_handle_ = it->second->cublas_handle; in CudaSolver()
325 static inline Status GeamImpl(SolverFnT solver, cublasHandle_t cublas_handle, in GeamImpl() argument
334 TF_RETURN_IF_CUBLAS_ERROR(solver(cublas_handle, transa, transb, m, n, in GeamImpl()
701 cublasHandle_t cublas_handle, int n, in GetrfBatchedImpl() argument
716 solver(cublas_handle, n, in GetrfBatchedImpl()
737 cublasHandle_t cublas_handle, cublasOperation_t trans, int n, int nrhs, in GetrsBatchedImpl() argument
[all …]
|