PyTorch extension enabling direct access to cuDNN-accelerated C++ convolution functions.
Primary LanguageC++