Home
last modified time | relevance | path

Searched refs:cublasHandle_t (Results 1 – 6 of 6) 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, int *version) {
20 using FuncPtr = cublasStatus_t (CUBLASWINAPI *)(cublasHandle_t, int *);
33 cublasStatus_t CUBLASWINAPI cublasSetStream_v2 (cublasHandle_t handle, cudaStream_t streamId) {
34 using FuncPtr = cublasStatus_t (CUBLASWINAPI *)(cublasHandle_t, cudaStream_t);
40 cublasStatus_t CUBLASWINAPI cublasGetStream_v2 (cublasHandle_t handle, cudaStream_t *streamId) {
41 using FuncPtr = 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, int *version) {
20 using FuncPtr = cublasStatus_t (CUBLASWINAPI *)(cublasHandle_t, int *);
33 cublasStatus_t CUBLASWINAPI cublasSetStream_v2 (cublasHandle_t handle, cudaStream_t streamId) {
34 using FuncPtr = cublasStatus_t (CUBLASWINAPI *)(cublasHandle_t, cudaStream_t);
40 cublasStatus_t CUBLASWINAPI cublasGetStream_v2 (cublasHandle_t handle, cudaStream_t *streamId) {
41 using FuncPtr = cublasStatus_t (CUBLASWINAPI *)(cublasHandle_t, cudaStream_t *);
[all …]
Dcuda_blas.h30 typedef struct cublasContext *cublasHandle_t; typedef
153 cublasHandle_t blas_ GUARDED_BY(mu_);
Dcuda_blas.cc135 explicit ScopedCublasPointerMode(cublasHandle_t handle) in ScopedCublasPointerMode()
171 cublasHandle_t handle_; // Handle to the cuBLAS instance of interest.
193 explicit ScopedCublasMathMode(cublasHandle_t handle) in ScopedCublasMathMode()
228 cublasHandle_t handle_; // Handle to the cuBLAS instance of interest.
/external/tensorflow/tensorflow/core/kernels/
Dcuda_solvers.cc111 cublasHandle_t cublas_handle;
325 static inline Status GeamImpl(SolverFnT solver, cublasHandle_t cublas_handle, in GeamImpl()
701 cublasHandle_t cublas_handle, int n, in GetrfBatchedImpl()
737 cublasHandle_t cublas_handle, cublasOperation_t trans, int n, int nrhs, in GetrsBatchedImpl()
784 cublasHandle_t cublas_handle, int n, const Scalar* const host_a_dev_ptrs[], in GetriBatchedImpl()
827 cublasHandle_t cublas_handle, int n, const Scalar* const host_a_dev_ptrs[], in MatInvBatchedImpl()
Dcuda_solvers.h325 cublasHandle_t cublas_handle_;