Searched defs:nn_modules (Results 1 – 2 of 2) sorted by relevance
| /aosp_15_r20/external/pytorch/torch/ | ||
| H A D | _guards.py | 404 nn_modules: Dict[str, torch.nn.Module] = {} variable in ModuleContextCheckpointState |
| /aosp_15_r20/external/pytorch/torch/_dynamo/ | ||
| H A D | output_graph.py | 597 def nn_modules(self) -> Dict[str, Any]: member in OutputGraph |