mirror of
https://github.com/zebrajr/pytorch.git
synced 2025-12-06 12:20:52 +01:00
d9aa6dfe88
Pull Request resolved: https://github.com/pytorch/pytorch/pull/85382 Approved by: https://github.com/ezyang
19 lines
376 B
Python
19 lines
376 B
Python
import torch
|
|
|
|
from torch._subclasses.fake_tensor import (
|
|
DynamicOutputShapeException,
|
|
FakeTensor,
|
|
FakeTensorMode,
|
|
UnsupportedFakeTensorException,
|
|
)
|
|
|
|
from torch._subclasses.fake_utils import CrossRefFakeMode
|
|
|
|
__all__ = [
|
|
"FakeTensor",
|
|
"FakeTensorMode",
|
|
"UnsupportedFakeTensorException",
|
|
"DynamicOutputShapeException",
|
|
"CrossRefFakeMode",
|
|
]
|