GPU version

[ccj5351]

Thanks a lot for sharing this code. Is there any work done or being done to convert the CPU code to GPU implementation? Because I am considering incorporating this layer to some SOTA image segmentation CNNs. It is not trivial to do this conversion, right?

Thanks.