* optimize tensorrt usage * format code * fix input shape error for onnx model Co-authored-by: root <root@bjyz-sys-gpu-kongming3.bjyz.baidu.com>