Searched defs:_unstack_pytree (Results 1 – 2 of 2) sorted by relevance
| /external/pytorch/torch/_higher_order_ops/ | ||
| D | utils.py | 342 def _unstack_pytree(xs): function |
| /external/executorch/exir/ | ||
| D | pass_base.py | 75 def _unstack_pytree(xs) -> List[PyTree]: # pyre-ignore function |