Searched defs:GetActivationFromParams (Results 1 – 1 of 1) sorted by relevance
281 at::cuda::blas::GEMMAndBiasActivationEpilogue GetActivationFromParams(const GemmParams<T>* params) { in GetActivationFromParams() function286 at::cuda::blas::GEMMAndBiasActivationEpilogue GetActivationFromParams(const GemmAndBiasParams<T>* p… in GetActivationFromParams() function291 at::cuda::blas::GEMMAndBiasActivationEpilogue GetActivationFromParams(const GemmStridedBatchedParam… in GetActivationFromParams() function296 at::cuda::blas::GEMMAndBiasActivationEpilogue GetActivationFromParams(const ScaledGemmParams<T>* pa… in GetActivationFromParams() function