Searched defs:cublas_handle (Results 1 – 1 of 1) sorted by relevance
| /aosp_15_r20/external/tensorflow/tensorflow/core/util/ |
| H A D | cuda_solvers.cc | 131 cublasHandle_t cublas_handle; member
345 static inline Status GeamImpl(SolverFnT solver, cublasHandle_t cublas_handle, in GeamImpl()
766 cublasHandle_t cublas_handle, int n, in GetrfBatchedImpl()
802 cublasHandle_t cublas_handle, cublasOperation_t trans, int n, int nrhs, in GetrsBatchedImpl()
849 cublasHandle_t cublas_handle, int n, const Scalar* const host_a_dev_ptrs[], in GetriBatchedImpl()
892 cublasHandle_t cublas_handle, int n, const Scalar* const host_a_dev_ptrs[], in MatInvBatchedImpl()
932 static inline Status TrsmImpl(SolverFnT solver, cublasHandle_t cublas_handle, in TrsmImpl()
961 static inline Status TrsvImpl(SolverFnT solver, cublasHandle_t cublas_handle, in TrsvImpl()
987 cublasHandle_t cublas_handle, cublasSideMode_t side, cublasFillMode_t uplo, in TrsmBatchedImpl()
|