#ifdef USE_CUDA #include // for the definition of AT_CUDNN_ENABLED #if AT_CUDNN_ENABLED() #include #include #include #include #include std::tuple> PackedLinearWeightCudnn::unpack() { return std::tuple>{orig_weight, bias_}; } #endif // AT_CUDNN_ENABLED #endif // USE_CUDA