File size: 233 Bytes
9c6594c
 
 
 
 
 
 
1
2
3
4
5
6
7
8
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,
)