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,
|
|
)
|