VhddlmZddlZddlmZmZdej jdej jfdZ y)) defaultdictN) _ModuleFrame_SubmoduleEntry orig_graphreturnc tjjtjj tjj }i}t t}t t}i}t|t|j||||ddgdi| j|jj|j|S)N)Nrr )module)torchfx GraphModulennModuleGraphrlistsetrtuplenodes run_outergraphlint recompile)r new_module seen_nodes seen_modules seen_attrscreated_moduless W/home/dcms/DCMS/lib/python3.12/site-packages/torch/distributed/pipelining/_unflatten.py_outline_submodulesrs%%ehhoo&79IJJ+-J5@5FL&1#&6J24O j    ik ) collectionsrr torch.export.unflattenrrr rr rr rr$s3# @EHHNNuxx7K7Kr