Searched refs:cublas_handle (Results 1 – 1 of 1) sorted by relevance
| /external/tensorflow/tensorflow/core/util/ |
| D | cuda_solvers.cc | 120 CHECK(cublasCreate(&cublas_handle) == CUBLAS_STATUS_SUCCESS) in CudaSolverHandles()
122 CHECK(cublasSetStream(cublas_handle, stream) == CUBLAS_STATUS_SUCCESS) in CudaSolverHandles()
127 CHECK(cublasDestroy(cublas_handle) == CUBLAS_STATUS_SUCCESS) in ~CudaSolverHandles()
132 cublasHandle_t cublas_handle; member
190 cublas_handle_ = it->second->cublas_handle; in CudaSolver()
346 static inline Status GeamImpl(SolverFnT solver, cublasHandle_t cublas_handle, in GeamImpl() argument
355 TF_RETURN_IF_CUBLAS_ERROR(solver(cublas_handle, transa, transb, m, n, in GeamImpl()
759 cublasHandle_t cublas_handle, int n, in GetrfBatchedImpl() argument
774 solver(cublas_handle, n, in GetrfBatchedImpl()
795 cublasHandle_t cublas_handle, cublasOperation_t trans, int n, int nrhs, in GetrsBatchedImpl() argument
[all …]
|