We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
1 parent 0b4536c commit e185d8bCopy full SHA for e185d8b
py/torch_tensorrt/__init__.py
@@ -85,6 +85,8 @@ def _find_lib(name: str, paths: List[str]) -> str:
85
from torch_tensorrt._Device import Device # noqa: F401
86
from torch_tensorrt._enums import * # noqa: F403
87
from torch_tensorrt._Input import Input # noqa: F401
88
+from torch_tensorrt.logging import *
89
+from torch_tensorrt.ptq import *
90
from torch_tensorrt._utils import * # noqa: F403
91
from torch_tensorrt._utils import sanitized_torch_version
92
0 commit comments