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

AI & Data Science Deep Learning (Training & Inference) TensorRT
5

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.

1 Like