o #i@s6ddlmZddlZddlmZdejjfddZdS))DictN) _ModuleFrame orig_graphc Cs^tjtjtj}i}i}t|t|j||ddgdi|d |j | |S)N)module) torchfx GraphModulennModuleGraphrtuplenodes run_outergraphlint recompile)r new_module seen_nodes seen_modulesre/home/ubuntu/SoloSpeech/.venv/lib/python3.10/site-packages/torch/distributed/pipelining/_unflatten.py_outline_submodules s$ r)typingrrtorch.export.unflattenrrr rrrrrs