File size: 144 Bytes
a40a8f5 | 1 2 3 4 5 6 7 | from torch import cond # noqa: F401
from torch._higher_order_ops.map import ( # noqa: F401
_stack_pytree,
_unstack_pytree,
map,
)
|
a40a8f5 | 1 2 3 4 5 6 7 | from torch import cond # noqa: F401
from torch._higher_order_ops.map import ( # noqa: F401
_stack_pytree,
_unstack_pytree,
map,
)
|