Searched refs:cublasHandle_t (Results 1 – 9 of 9) sorted by relevance
/external/tensorflow/tensorflow/stream_executor/cuda/ |
D | cublas_9_0.inc | 5 cublasStatus_t CUBLASWINAPI cublasCreate_v2(cublasHandle_t *handle) { 6 using FuncPtr = cublasStatus_t(CUBLASWINAPI *)(cublasHandle_t *); 12 cublasStatus_t CUBLASWINAPI cublasDestroy_v2(cublasHandle_t handle) { 13 using FuncPtr = cublasStatus_t(CUBLASWINAPI *)(cublasHandle_t); 19 cublasStatus_t CUBLASWINAPI cublasGetVersion_v2(cublasHandle_t handle, 21 using FuncPtr = cublasStatus_t(CUBLASWINAPI *)(cublasHandle_t, int *); 35 cublasStatus_t CUBLASWINAPI cublasSetStream_v2(cublasHandle_t handle, 37 using FuncPtr = cublasStatus_t(CUBLASWINAPI *)(cublasHandle_t, cudaStream_t); 43 cublasStatus_t CUBLASWINAPI cublasGetStream_v2(cublasHandle_t handle, 46 cublasStatus_t(CUBLASWINAPI *)(cublasHandle_t, cudaStream_t *); [all …]
|
D | cublas_11_0.inc | 5 cublasStatus_t CUBLASWINAPI cublasCreate_v2(cublasHandle_t *handle) { 6 using FuncPtr = cublasStatus_t(CUBLASWINAPI *)(cublasHandle_t *); 12 cublasStatus_t CUBLASWINAPI cublasDestroy_v2(cublasHandle_t handle) { 13 using FuncPtr = cublasStatus_t(CUBLASWINAPI *)(cublasHandle_t); 19 cublasStatus_t CUBLASWINAPI cublasGetVersion_v2(cublasHandle_t handle, 21 using FuncPtr = cublasStatus_t(CUBLASWINAPI *)(cublasHandle_t, int *); 42 cublasStatus_t CUBLASWINAPI cublasSetStream_v2(cublasHandle_t handle, 44 using FuncPtr = cublasStatus_t(CUBLASWINAPI *)(cublasHandle_t, cudaStream_t); 50 cublasStatus_t CUBLASWINAPI cublasGetStream_v2(cublasHandle_t handle, 53 cublasStatus_t(CUBLASWINAPI *)(cublasHandle_t, cudaStream_t *); [all …]
|
D | cublas_10_1.inc | 5 cublasStatus_t CUBLASWINAPI cublasCreate_v2(cublasHandle_t *handle) { 6 using FuncPtr = cublasStatus_t(CUBLASWINAPI *)(cublasHandle_t *); 12 cublasStatus_t CUBLASWINAPI cublasDestroy_v2(cublasHandle_t handle) { 13 using FuncPtr = cublasStatus_t(CUBLASWINAPI *)(cublasHandle_t); 19 cublasStatus_t CUBLASWINAPI cublasGetVersion_v2(cublasHandle_t handle, 21 using FuncPtr = cublasStatus_t(CUBLASWINAPI *)(cublasHandle_t, int *); 42 cublasStatus_t CUBLASWINAPI cublasSetStream_v2(cublasHandle_t handle, 44 using FuncPtr = cublasStatus_t(CUBLASWINAPI *)(cublasHandle_t, cudaStream_t); 50 cublasStatus_t CUBLASWINAPI cublasGetStream_v2(cublasHandle_t handle, 53 cublasStatus_t(CUBLASWINAPI *)(cublasHandle_t, cudaStream_t *); [all …]
|
D | cublas_10_2.inc | 5 cublasStatus_t CUBLASWINAPI cublasCreate_v2(cublasHandle_t *handle) { 6 using FuncPtr = cublasStatus_t(CUBLASWINAPI *)(cublasHandle_t *); 12 cublasStatus_t CUBLASWINAPI cublasDestroy_v2(cublasHandle_t handle) { 13 using FuncPtr = cublasStatus_t(CUBLASWINAPI *)(cublasHandle_t); 19 cublasStatus_t CUBLASWINAPI cublasGetVersion_v2(cublasHandle_t handle, 21 using FuncPtr = cublasStatus_t(CUBLASWINAPI *)(cublasHandle_t, int *); 42 cublasStatus_t CUBLASWINAPI cublasSetStream_v2(cublasHandle_t handle, 44 using FuncPtr = cublasStatus_t(CUBLASWINAPI *)(cublasHandle_t, cudaStream_t); 50 cublasStatus_t CUBLASWINAPI cublasGetStream_v2(cublasHandle_t handle, 53 cublasStatus_t(CUBLASWINAPI *)(cublasHandle_t, cudaStream_t *); [all …]
|
D | cublas_10_0.inc | 5 cublasStatus_t CUBLASWINAPI cublasCreate_v2(cublasHandle_t *handle) { 6 using FuncPtr = cublasStatus_t(CUBLASWINAPI *)(cublasHandle_t *); 12 cublasStatus_t CUBLASWINAPI cublasDestroy_v2(cublasHandle_t handle) { 13 using FuncPtr = cublasStatus_t(CUBLASWINAPI *)(cublasHandle_t); 19 cublasStatus_t CUBLASWINAPI cublasGetVersion_v2(cublasHandle_t handle, 21 using FuncPtr = cublasStatus_t(CUBLASWINAPI *)(cublasHandle_t, int *); 35 cublasStatus_t CUBLASWINAPI cublasSetStream_v2(cublasHandle_t handle, 37 using FuncPtr = cublasStatus_t(CUBLASWINAPI *)(cublasHandle_t, cudaStream_t); 43 cublasStatus_t CUBLASWINAPI cublasGetStream_v2(cublasHandle_t handle, 46 cublasStatus_t(CUBLASWINAPI *)(cublasHandle_t, cudaStream_t *); [all …]
|
D | cuda_blas.h | 33 typedef struct cublasContext *cublasHandle_t; typedef 168 cublasHandle_t blas_ TF_GUARDED_BY(mu_);
|
D | cuda_blas.cc | 123 explicit ScopedCublasPointerMode(cublasHandle_t handle) in ScopedCublasPointerMode() 159 cublasHandle_t handle_; // Handle to the cuBLAS instance of interest. 181 explicit ScopedCublasMathMode(cublasHandle_t handle) in ScopedCublasMathMode() 216 cublasHandle_t handle_; // Handle to the cuBLAS instance of interest.
|
/external/tensorflow/tensorflow/core/util/ |
D | cuda_solvers.cc | 132 cublasHandle_t cublas_handle; 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()
|
D | cuda_solvers.h | 367 cublasHandle_t cublas_handle_;
|