Why Gemfury? Push, build, and install  RubyGems npm packages Python packages Maven artifacts PHP packages Go Modules Debian packages RPM packages NuGet packages

Repository URL to install this package:

Details    
torch / _dynamo / backends / tensorrt.py
Size: Mime:
# import torch  # type: ignore[import]
# from .common import device_from_inputs, fake_tensor_unsupported  # type: ignore[import]
# from .registry import register_backend  # type: ignore[import]

"""
Placeholder for TensorRT backend for dynamo via torch-tensorrt
"""

# @register_backend
# def tensorrt(gm, example_inputs):
#    import torch_tensorrt # type: ignore[import]
#    pass