Searched defs:gemm_internal_cublas (Results 1 – 1 of 1) sorted by relevance
791 inline void gemm_internal_cublas(CUDABLAS_GEMM_ARGTYPES(Dtype)) { in gemm_internal_cublas() function796 void gemm_internal_cublas<double>(CUDABLAS_GEMM_ARGTYPES(double)) { in gemm_internal_cublas() function809 void gemm_internal_cublas<float>(CUDABLAS_GEMM_ARGTYPES(float)) { in gemm_internal_cublas() function822 void gemm_internal_cublas<c10::complex<double>>(CUDABLAS_GEMM_ARGTYPES(c10::complex<double>)) { in gemm_internal_cublas() function837 void gemm_internal_cublas<c10::complex<float>>(CUDABLAS_GEMM_ARGTYPES(c10::complex<float>)) { in gemm_internal_cublas() function852 void gemm_internal_cublas<at::Half>(CUDABLAS_GEMM_ARGTYPES(at::Half)) { in gemm_internal_cublas() function948 void gemm_internal_cublas<at::BFloat16>(CUDABLAS_GEMM_ARGTYPES(at::BFloat16)) { in gemm_internal_cublas() function