Searched defs:gemm_internal (Results 1 – 2 of 2) sorted by relevance
993 void gemm_internal<double>(CUDABLAS_GEMM_ARGTYPES(double)) in gemm_internal() function1009 void gemm_internal<float>(CUDABLAS_GEMM_ARGTYPES(float)) in gemm_internal() function1020 void gemm_internal<c10::complex<double>>(CUDABLAS_GEMM_ARGTYPES(c10::complex<double>)) in gemm_internal() function1036 void gemm_internal<c10::complex<float>>(CUDABLAS_GEMM_ARGTYPES(c10::complex<float>)) in gemm_internal() function1052 void gemm_internal<at::Half>(CUDABLAS_GEMM_ARGTYPES(at::Half)) in gemm_internal() function1063 void gemm_internal<at::BFloat16>(CUDABLAS_GEMM_ARGTYPES(at::BFloat16)) in gemm_internal() function
68 inline void gemm_internal(CUDABLAS_GEMM_ARGTYPES(Dtype)) { in gemm_internal() function