Home
last modified time | relevance | path

Searched defs:bgemm (Results 1 – 2 of 2) sorted by relevance

/external/pytorch/aten/src/ATen/cuda/
DCUDABlas.cpp698 …static tunable::GemmStridedBatchedTunableOp<DType, tunable::BlasOp::T, tunable::BlasOp::T> bgemm{}; in bgemm_tunable() local
702 …static tunable::GemmStridedBatchedTunableOp<DType, tunable::BlasOp::T, tunable::BlasOp::N> bgemm{}; in bgemm_tunable() local
706 …static tunable::GemmStridedBatchedTunableOp<DType, tunable::BlasOp::N, tunable::BlasOp::T> bgemm{}; in bgemm_tunable() local
710 …static tunable::GemmStridedBatchedTunableOp<DType, tunable::BlasOp::N, tunable::BlasOp::N> bgemm{}; in bgemm_tunable() local
719 void bgemm<double>(CUDABLAS_BGEMM_ARGTYPES(double)) { in bgemm() function
730 void bgemm<float>(CUDABLAS_BGEMM_ARGTYPES(float)) { in bgemm() function
741 void bgemm<c10::complex<double>>(CUDABLAS_BGEMM_ARGTYPES(c10::complex<double>)) { in bgemm() function
752 void bgemm<c10::complex<float>>(CUDABLAS_BGEMM_ARGTYPES(c10::complex<float>)) { in bgemm() function
763 void bgemm<at::Half>(CUDABLAS_BGEMM_ARGTYPES(at::Half)) { in bgemm() function
774 void bgemm<at::BFloat16>(CUDABLAS_BGEMM_ARGTYPES(at::BFloat16)) { in bgemm() function
DCUDABlas.h156 inline void bgemm(CUDABLAS_BGEMM_ARGTYPES(Dtype)) { in bgemm() function