Home
last modified time | relevance | path

Searched refs:cublasHandle_t (Results 1 – 9 of 9) sorted by relevance

/external/tensorflow/tensorflow/stream_executor/cuda/
Dcublas_9_0.inc5 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 …]
Dcublas_11_0.inc5 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 …]
Dcublas_10_1.inc5 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 …]
Dcublas_10_2.inc5 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 …]
Dcublas_10_0.inc5 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 …]
Dcuda_blas.h33 typedef struct cublasContext *cublasHandle_t; typedef
168 cublasHandle_t blas_ TF_GUARDED_BY(mu_);
Dcuda_blas.cc123 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/
Dcuda_solvers.cc132 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()
Dcuda_solvers.h367 cublasHandle_t cublas_handle_;