Searched defs:bgemm_internal (Results 1 – 2 of 2) sorted by relevance
593 void bgemm_internal<double>(CUDABLAS_BGEMM_ARGTYPES(double)) in bgemm_internal() function609 void bgemm_internal<float>(CUDABLAS_BGEMM_ARGTYPES(float)) in bgemm_internal() function620 void bgemm_internal<c10::complex<double>>(CUDABLAS_BGEMM_ARGTYPES(c10::complex<double>)) in bgemm_internal() function636 void bgemm_internal<c10::complex<float>>(CUDABLAS_BGEMM_ARGTYPES(c10::complex<float>)) in bgemm_internal() function652 void bgemm_internal<at::Half>(CUDABLAS_BGEMM_ARGTYPES(at::Half)) in bgemm_internal() function663 void bgemm_internal<at::BFloat16>(CUDABLAS_BGEMM_ARGTYPES(at::BFloat16)) in bgemm_internal() function
174 inline void bgemm_internal(CUDABLAS_BGEMM_ARGTYPES(Dtype)) { in bgemm_internal() function