What is the timeline for dilated convolution support in TensorRT (Python), or SpaceToBatchND / BatchToSpaceND?
My ResNet50-based TensorFlow model has SpaceToBatchND and BatchToSpaceND around its Conv2D nodes, like so: https://ibb.co/mdxrBU . But these are not supported by TensorRT (in Python, not sure about C++).
There is this related post https://devtalk.nvidia.com/default/topic/1027934/jetson-tx2/tensorflows-dilated-convolutions-with-uff-parser-not-supported but no solution. (Also this one https://devtalk.nvidia.com/default/topic/1037732/tensorrt/no-conversion-function-registered-for-layer-slice-yet- asking the same for the Slice operation (but I can solve that in a different way.)
I would like to know when SpaceToBatchND / BatchToSpaceND, and/or dilated convolution, will be supported in the Python API. If this is expected soon, I won’t have to spend time (a lot) to implement a workaround: write Plugin nodes in C++, use GraphSurgeon to insert it, or replacing SpaceToBatchND and BatchToSpaceND do with a combination of Slice, Reshape, Padding, etc. (which is problematic since Slice is not currently supported either).
Please let me know about the timeline for this, and provide advice on which path to take. Thank you very much in advance.