`ibddlmZddlZddlmZmZdejjdejjfdZ dS)) defaultdictN) _ModuleFrame_SubmoduleEntry orig_graphreturnc tjtjtj}i}t t}t t}i}t|t|j ||||ddgdi| |j ||S)N)Nrr )module)torchfx GraphModulennModuleGraphrlistsetrtuplenodes run_outergraphlint recompile)r new_module seen_nodes seen_modules seen_attrscreated_moduless {/home/jaya/work/projects/VOICE-AGENT/VIET/agent-env/lib/python3.11/site-packages/torch/distributed/pipelining/_unflatten.py_outline_submodulesrs%%ehoo&7&79I9IJJJ+-J5@5F5FL&1#&6&6J24O j      ikkk ) collectionsrr torch.export.unflattenrrr rr rr rr$sm###### @@@@@@@@EHNux7Kr