
    קg                     F    d dl mZ d dlZd dlmZ dej        j        fdZdS )    )DictN)_ModuleFrame
orig_graphc                    t           j                            t           j                                        t           j                                                  }i }i }t          | t          | j                  ||d dgdi |	  	        	                                 |j
                                         |                                 |S )N )module)torchfxGraphModulennModuleGraphr   tuplenodes	run_outergraphlint	recompile)r   
new_module
seen_nodesseen_moduless       c/var/www/html/ai-engine/env/lib/python3.11/site-packages/torch/distributed/pipelining/_unflatten.py_outline_submodulesr   	   s    %%ehoo&7&79I9IJJJ+-J/1Lj	


 
 
 ikkk    )typingr   r	   torch.export.unflattenr   r
   r   r    r   r   <module>r      s]           / / / / / /EHN      r   