sgXddlmZddlZddlmZdej j fdZy))DictN) _ModuleFrame orig_graphc ~tjjtjj tjj }i}i}t |t|j||ddgdi| j|jj|j|S)N)module) torchfx GraphModulennModuleGraphrtuplenodes run_outergraphlint recompile)r new_module seen_nodes seen_moduless Z/var/www/html/venv/lib/python3.12/site-packages/torch/distributed/pipelining/_unflatten.py_outline_submodulesr s%%ehhoo&79IJJ+-J/1L j    ik )typingrr torch.export.unflattenrr rrrrrs$ /EHHNNr