Jason 969531dcc8 Optimize TensorRT backend to support rebuild engine (#189) ...

* optimize tensorrt usage

* format code

* fix input shape error for onnx model

Co-authored-by: root <root@bjyz-sys-gpu-kongming3.bjyz.baidu.com>

2022-09-06 10:53:05 +08:00

..

trt_backend.cc

Optimize TensorRT backend to support rebuild engine (#189)

2022-09-06 10:53:05 +08:00

trt_backend.h

Optimize TensorRT backend to support rebuild engine (#189)

2022-09-06 10:53:05 +08:00

utils.cc

Optimize TensorRT backend to support rebuild engine (#189)

2022-09-06 10:53:05 +08:00

utils.h

Optimize TensorRT backend to support rebuild engine (#189)

2022-09-06 10:53:05 +08:00