U yh@s6ddlmZddlZddlmZejjdddZdS))DictN) _ModuleFrame) orig_graphc Cs^tjtjtj}i}i}t|t|j||ddgdi|d |j | |S)N)module) torchfxZ GraphModulennModuleGraphrtupleZnodesZ run_outergraphZlintZ recompile)rZ new_moduleZ seen_nodesZ seen_modulesrY/var/www/html/venv/lib/python3.8/site-packages/torch/distributed/pipelining/_unflatten.py_outline_submodules s" r)typingrrZtorch.export.unflattenrrr rrrrrs