Searched defs:bgemm (Results 1 – 2 of 2) sorted by relevance
698 …static tunable::GemmStridedBatchedTunableOp<DType, tunable::BlasOp::T, tunable::BlasOp::T> bgemm{}; in bgemm_tunable() local702 …static tunable::GemmStridedBatchedTunableOp<DType, tunable::BlasOp::T, tunable::BlasOp::N> bgemm{}; in bgemm_tunable() local706 …static tunable::GemmStridedBatchedTunableOp<DType, tunable::BlasOp::N, tunable::BlasOp::T> bgemm{}; in bgemm_tunable() local710 …static tunable::GemmStridedBatchedTunableOp<DType, tunable::BlasOp::N, tunable::BlasOp::N> bgemm{}; in bgemm_tunable() local719 void bgemm<double>(CUDABLAS_BGEMM_ARGTYPES(double)) { in bgemm() function730 void bgemm<float>(CUDABLAS_BGEMM_ARGTYPES(float)) { in bgemm() function741 void bgemm<c10::complex<double>>(CUDABLAS_BGEMM_ARGTYPES(c10::complex<double>)) { in bgemm() function752 void bgemm<c10::complex<float>>(CUDABLAS_BGEMM_ARGTYPES(c10::complex<float>)) { in bgemm() function763 void bgemm<at::Half>(CUDABLAS_BGEMM_ARGTYPES(at::Half)) { in bgemm() function774 void bgemm<at::BFloat16>(CUDABLAS_BGEMM_ARGTYPES(at::BFloat16)) { in bgemm() function
156 inline void bgemm(CUDABLAS_BGEMM_ARGTYPES(Dtype)) { in bgemm() function