Searched refs:UnflattenedModule (Results 1 – 2 of 2) sorted by relevance
190 class UnflattenedModule(torch.nn.Module): class531 ) -> UnflattenedModule:552 return UnflattenedModule(module, flat_args_adapter)1105 def _outline_submodules(orig_graph: torch.fx.Graph, root_module: UnflattenedModule): argument
67 from .unflatten import FlatArgsAdapter, unflatten, UnflattenedModule