Flex-Convolution icon indicating copy to clipboard operation
Flex-Convolution copied to clipboard

Share implementation between a PyTorch and TensorFlow

Open PatWie opened this issue 5 years ago • 0 comments

Just an attempt to refactor the code and reuse the CUDA kernels for a PyTorch version.

PatWie avatar Feb 23 '19 19:02 PatWie