pytorch

1
# This file has moved to under torch/_functorch. It is not public API.
2
# If you are not a PyTorch developer and you are relying on the following
3
# imports, please file an issue.
4
from torch._functorch.aot_autograd import (
5
    aot_autograd_decompositions,
6
    KNOWN_TYPES,
7
    PytreeThunk,
8
)
9