Searched defs:FlattenFunc (Results 1 – 4 of 4) sorted by relevance
| /aosp_15_r20/external/pytorch/torch/utils/ |
| H A D | _cxx_pytree.py | 83 FlattenFunc = Callable[[PyTree], Tuple[List[Any], Context]] variable
103 flatten_fn: FlattenFunc,
173 flatten_fn: FlattenFunc,
224 flatten_fn: FlattenFunc,
|
| H A D | _pytree.py | 115 FlattenFunc = Callable[[PyTree], Tuple[List[Any], Context]] variable
173 flatten_fn: FlattenFunc,
278 flatten_fn: FlattenFunc,
333 flatten_fn: FlattenFunc,
|
| /aosp_15_r20/external/pytorch/torch/onnx/_internal/fx/ |
| H A D | dynamo_graph_extractor.py | 48 flatten_func: pytree.FlattenFunc,
|
| /aosp_15_r20/external/pytorch/torch/_export/ |
| H A D | utils.py | 352 flatten_fn: Optional[FlattenFunc] = None,
|