Centerpointpillar TRT deployment

Deer Tech team:
I am using TRT to do the centerpointpillar deployment, I met a problem as as below

I trained the Lidar model and then I export the .pth model to .onnx with model simplification work. Unfortunately, it has the shape problem. I refered the CUDApointpillar project, it should be fine.
Here is my simplified onnx model. if there’s a problem please tell me.
pfe_centerpoint.onnx (19.2 MB)

Hope you guys may provide me with some hint.
Thanks in advance.

Best regards,
Barry