Searched defs:_get_root_modules (Results 1 – 2 of 2) sorted by relevance
361 def _get_root_modules(modules: List[nn.Module]) -> List[nn.Module]: function
466 def _get_root_modules(modules: Set[nn.Module]) -> Set[nn.Module]: function