HomeSort by relevance Sort by last modified time
    Searched refs:cublasHandle_t (Results 1 - 4 of 4) sorted by null

  /external/tensorflow/tensorflow/stream_executor/cuda/
cuda_blas.h 30 typedef struct cublasContext *cublasHandle_t;
159 cublasHandle_t blas_ GUARDED_BY(mu_);
cuda_blas.cc 337 explicit ScopedCublasPointerMode(CUDAExecutor *parent, cublasHandle_t handle)
376 cublasHandle_t handle_; // Handle to the cuBLAS instance of interest.
398 explicit ScopedCublasMathMode(CUDAExecutor *parent, cublasHandle_t handle)
434 cublasHandle_t handle_; // Handle to the cuBLAS instance of interest.
    [all...]
  /external/tensorflow/tensorflow/core/kernels/
cuda_solvers.cc 111 cublasHandle_t cublas_handle;
325 static inline Status GeamImpl(SolverFnT solver, cublasHandle_t cublas_handle,
657 cublasHandle_t cublas_handle, int n,
    [all...]
cuda_solvers.h 316 cublasHandle_t cublas_handle_;

Completed in 216 milliseconds