Share implementation between a PyTorch and TensorFlow

cgtuebingen / Flex-Convolution

Source code for: Flex-Convolution (Million-Scale Point-Cloud Learning Beyond Grid-Worlds), accepted at ACCV 2018

Apache License 2.0

115 stars 11 forks source link

Open PatWie opened 5 years ago

PatWie commented 5 years ago

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