Searched defs:UnflattenFunc (Results 1 – 4 of 4) sorted by relevance
| /aosp_15_r20/external/pytorch/torch/utils/ |
| H A D | _cxx_pytree.py | 84 UnflattenFunc = Callable[[Iterable[Any], Context], PyTree] variable
93 def _reverse_args(func: UnflattenFunc) -> OpTreeUnflattenFunc:
104 unflatten_fn: UnflattenFunc,
174 unflatten_fn: UnflattenFunc,
225 unflatten_fn: UnflattenFunc,
|
| H A D | _pytree.py | 116 UnflattenFunc = Callable[[Iterable[Any], Context], PyTree] variable
174 unflatten_fn: UnflattenFunc,
279 unflatten_fn: UnflattenFunc,
334 unflatten_fn: UnflattenFunc,
|
| /aosp_15_r20/external/pytorch/torch/onnx/_internal/fx/ |
| H A D | dynamo_graph_extractor.py | 49 unflatten_func: pytree.UnflattenFunc,
|
| /aosp_15_r20/external/pytorch/torch/_export/ |
| H A D | utils.py | 353 unflatten_fn: Optional[UnflattenFunc] = None,
|