ai-content-maker/.venv/Lib/site-packages/functorch/experimental/control_flow.py

9 lines
234 B
Python
Raw Normal View History

2024-05-03 04:18:51 +03:00
from torch import cond # noqa: F401
from torch._higher_order_ops.cond import UnsupportedAliasMutationException # noqa: F401
from torch._higher_order_ops.map import ( # noqa: F401
_stack_pytree,
_unstack_pytree,
map,
)