You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
|
|
|
|
from .engine import EngineBuilder, TRTModule, TRTProfilerV0, TRTProfilerV1 # isort:skip # noqa: E501
|
|
|
|
|
import warnings
|
|
|
|
|
|
|
|
|
|
import torch
|
|
|
|
|
|
|
|
|
|
warnings.filterwarnings(action='ignore', category=torch.jit.TracerWarning)
|
|
|
|
|
warnings.filterwarnings(action='ignore', category=torch.jit.ScriptWarning)
|
|
|
|
|
warnings.filterwarnings(action='ignore', category=UserWarning)
|
|
|
|
|
warnings.filterwarnings(action='ignore', category=FutureWarning)
|
|
|
|
|
warnings.filterwarnings(action='ignore', category=DeprecationWarning)
|
|
|
|
|
__all__ = ['EngineBuilder', 'TRTModule', 'TRTProfilerV0', 'TRTProfilerV1']
|
