mirror of
https://github.com/pytorch/pytorch.git
synced 2025-10-20 21:14:14 +08:00
68034198e5
This PR reworks the way the input mutations and various aliases are checked Pull Request resolved: https://github.com/pytorch/pytorch/pull/146658 Approved by: https://github.com/ydwu4
7 lines
144 B
Python
7 lines
144 B
Python
from torch import cond # noqa: F401
|
|
from torch._higher_order_ops.map import ( # noqa: F401
|
|
_stack_pytree,
|
|
_unstack_pytree,
|
|
map,
|
|
)
|