onnx-simplifier
onnx-simplifier copied to clipboard
Weird Question
Is it possible to have the opposite behaviour of onnx-simplifier? I have a torch.nn.functional.grid_sample
operation that I can export to onnx using opset=16
as a GridSample
node. The problem is that when I feed the onnx to tensorrt
, quite unsurprisingly, the node ain't recongnized.
I'd like to split GridSample
down to simpler onnx operators but I don't know if it's even possible. Does anyone have an idea on how to do that?
If possible, I would try to split the grid_sample into multiple objects within the python code if I were in your shoes.