File size: 205 Bytes
495526b
 
 
 
 
1
2
3
4
5
6
def tensorrt_optimize(onnx_path: str) -> None:
    print(
        "TensorRT optimization stub. Use trtexec or TensorRT Python APIs in GPU-enabled environment "
        f"for ONNX model: {onnx_path}"
    )