a h@s@ddlmZddlZddlmZmZejjejjdddZ dS)) defaultdictN) _ModuleFrame_SubmoduleEntry) orig_graphreturnc Csrtjtjtj}i}tt}tt}i}t |t |j ||||ddgdi|d |j ||S)N)Nrr)module)torchfx GraphModulennModuleGraphrlistsetrtuplenodesZ run_outergraphZlintZ recompile)rZ new_moduleZ seen_nodesZ seen_modulesZ seen_attrsZcreated_modulesrU/var/www/auris/lib/python3.9/site-packages/torch/distributed/pipelining/_unflatten.py_outline_submoduless* r) collectionsrr Ztorch.export.unflattenrrr rr rrrrrs