tkDNN
tkDNN copied to clipboard
Adding dilated convolution and support for reorg3d layer
I have added dilated convolution as discussed in #37. It works both for TensorRT and CUDNN.
The code also adds reorg3d
layer ~~as an alias for reorg
layer~~.
There is also a test_yolo3_tiny_dilation
application but I have not uploaded the cfg and weights files.
This pull request now contains the fix for #49.