Searched defs:FlattenWithKeysFunc (Results 1 – 2 of 2) sorted by relevance
| /aosp_15_r20/external/pytorch/torch/utils/ |
| H A D | _pytree.py | 123 FlattenWithKeysFunc = Callable[[PyTree], Tuple[List[Tuple[KeyEntry, Any]], Any]] variable
179 flatten_with_keys_fn: Optional[FlattenWithKeysFunc] = None,
286 flatten_with_keys_fn: Optional[FlattenWithKeysFunc] = None,
339 flatten_with_keys_fn: Optional[FlattenWithKeysFunc] = None,
|
| H A D | _cxx_pytree.py | 90 FlattenWithKeysFunc = Callable[[PyTree], Tuple[List[Tuple[KeyEntry, Any]], Any]] variable
109 flatten_with_keys_fn: Optional[FlattenWithKeysFunc] = None,
|