I’m trying to use ONNX as an intermediate bewteeen Pytorch and TensorRT. When exporting, onnx complains about grid_sample not being supported. Is there a way to get it to throw in a placeholder node with the same args etc for a PyTorch op?
I have previously added ONNX support for a correlation function which somewhat exports correctly. ONNX complains about exporting the autograd function, but I am able to just comment out the autograd call and use the custom op directly just when exporting for it to work correctly. I tried to be cheeky about it and use the raw function directly during self.eval() during exporting instead of the autograd function but it still tries to go through autograd.
I have already written both grid_sample and correlation IPluginV2ExtIO, so this intermediate step is currently the bottleneck (until I start debugging them haha).