§
    Ÿ²ÅjÎ  ã                   óž   — d dl mZmZ d dlZd dlmZ d dlmZ d dlm	Z	 dgZ
 G d„ dej        j        j        ¦  «        Z G d	„ de¦  «        ZdS )
é    )ÚCallableÚListN)Ú_FusedModule)ÚTracer)ÚScopeÚQuantizationTracerc                   ó@   ‡ — e Zd Zdedej        j        defˆ fd„Zˆ xZ	S )ÚScopeContextManagerÚscopeÚcurrent_moduleÚcurrent_module_pathc           	      ó‚   •— t          ¦   «                              |t          |t          |¦  «        ¦  «        ¦  «         d S )N)ÚsuperÚ__init__r   Útype)Úselfr   r   r   Ú	__class__s       €úPC:\Projects\Historical\venv\Lib\site-packages\torch/ao/quantization/fx/tracer.pyr   zScopeContextManager.__init__   s8   ø€ õ 	‰Œ×Ò˜¥Ð&9½4ÀÑ;OÔ;OÑ PÔ PÑQÔQÐQÐQÐQó    )
Ú__name__Ú
__module__Ú__qualname__r   ÚtorchÚnnÚModuleÚstrr   Ú__classcell__©r   s   @r   r
   r
      sf   ø€ € € € € ðRØðRØ,1¬H¬OðRØRUðRð Rð Rð Rð Rð Rð Rð Rð Rð Rr   r
   c                   óh   ‡ — e Zd Zdee         dee         fˆ fd„Zdej        j	        dede
fd„Zˆ xZS )r   Úskipped_module_namesÚskipped_module_classesc                 óœ   •— t          ¦   «                              ¦   «          || _        || _        t	          dd ¦  «        | _        d| _        d S )NÚ T)r   r   r    r!   r   r   Úrecord_stack_traces)r   r    r!   r   s      €r   r   zQuantizationTracer.__init__   sJ   ø€ õ 	‰Œ×ÒÑÔÐØ$8ˆÔ!Ø&<ˆÔ#õ ˜2˜t‘_”_ˆŒ
Ø#'ˆÔ Ð Ð r   ÚmÚmodule_qualified_nameÚreturnc                 ó  — |j                              d¦  «        p|j                              d¦  «        ot          |t          j        j        ¦  «         p3|| j        v p*t          |¦  «        | j        v pt          |t          ¦  «        S )Nztorch.nnztorch.ao.nn)
r   Ú
startswithÚ
isinstancer   r   Ú
Sequentialr    r   r!   r   )r   r%   r&   s      r   Úis_leaf_modulez!QuantizationTracer.is_leaf_module$   sŽ   € ð ”L×+Ò+¨JÑ7Ô7ð >Ø”|×.Ò.¨}Ñ=Ô=ð;õ # 1¥e¤hÔ&9Ñ:Ô:Ð:ð	+ð %¨Ô(AÐAð	+õ �A‰wŒw˜$Ô5Ð5ð	+õ ˜!�\Ñ*Ô*ð	
r   )r   r   r   r   r   r   r   r   r   r   Úboolr,   r   r   s   @r   r   r      sƒ   ø€ € € € € ð(Ø$(¨¤Ið(ØGKÈHÄ~ð(ð (ð (ð (ð (ð (ð
 ¤¤ð 
Èð 
ÐPTð 
ð 
ð 
ð 
ð 
ð 
ð 
ð 
r   )Útypingr   r   r   Útorch.ao.nn.intrinsicr   Útorch.fx._symbolic_tracer   Útorch.fx.proxyr   Ú__all__ÚfxÚproxyr
   r   © r   r   ú<module>r6      sÕ   ðØ !Ð !Ð !Ð !Ð !Ð !Ð !Ð !à €€€Ø .Ð .Ð .Ð .Ð .Ð .Ø +Ð +Ð +Ð +Ð +Ð +Ø  Ð  Ð  Ð  Ð  Ð  ð ð€ð
Rð Rð Rð Rð R˜%œ(œ.Ô<ñ Rô Rð Rð
ð 
ð 
ð 
ð 
˜ñ 
ô 
ð 
ð 
ð 
r   