__init__.py 376 B

123456789101112131415161718
  1. import torch
  2. from torch._subclasses.fake_tensor import (
  3. DynamicOutputShapeException,
  4. FakeTensor,
  5. FakeTensorMode,
  6. UnsupportedFakeTensorException,
  7. )
  8. from torch._subclasses.fake_utils import CrossRefFakeMode
  9. __all__ = [
  10. "FakeTensor",
  11. "FakeTensorMode",
  12. "UnsupportedFakeTensorException",
  13. "DynamicOutputShapeException",
  14. "CrossRefFakeMode",
  15. ]