Searched defs:bgemm_internal_cublas (Results 1 – 1 of 1) sorted by relevance
458 inline void bgemm_internal_cublas(CUDABLAS_BGEMM_ARGTYPES(Dtype)) { in bgemm_internal_cublas() function463 void bgemm_internal_cublas<double>(CUDABLAS_BGEMM_ARGTYPES(double)) { in bgemm_internal_cublas() function476 void bgemm_internal_cublas<float>(CUDABLAS_BGEMM_ARGTYPES(float)) { in bgemm_internal_cublas() function489 void bgemm_internal_cublas<c10::complex<double>>(CUDABLAS_BGEMM_ARGTYPES(c10::complex<double>)) { in bgemm_internal_cublas() function504 void bgemm_internal_cublas<c10::complex<float>>(CUDABLAS_BGEMM_ARGTYPES(c10::complex<float>)) { in bgemm_internal_cublas() function519 void bgemm_internal_cublas<at::Half>(CUDABLAS_BGEMM_ARGTYPES(at::Half)) { in bgemm_internal_cublas() function566 void bgemm_internal_cublas<at::BFloat16>(CUDABLAS_BGEMM_ARGTYPES(at::BFloat16)) { in bgemm_internal_cublas() function