Inference in TensorRT error due to F.pad : IShuffleLayer applied to shape tensor must have 0 or 1 reshape dimensions: dimensions were [-1,2]

[spolisetty]

Hi,

Following similar post may help you.

IShuffleLayer applied to shape tensor must have 0 or 1 reshape dimensions: dimensions were [-1,2])

Hi, I think padding related node is causing error, we don’t support 2D shape tensors yet. We can try workaround constant-fold with polygraphy. After this we are able to successfully generate engine. Please try polygraphy surgeon sanitize --fold-constants grid_sample.onnx -o 2001833/folded.onnx For more details, Thank you.

Thank you.