§
    ŠŠtj�  ã                   óœ  — U d dl mZ d dlmZ d dlZd dlmZ d dlmZ d dl	m
Z
 g d¢Z ed¬¦  «        d	ed
edefd„¦   «         Zej        j        j        j        dddgefej        j        j        j        dg d¢efej        j        j        j        dg d¢efej        j        j        j        dg efej        j        j        j        dg d¢efej        j        j        j        ddgefiZeeeeee         eeegef         f         f         e d<    ed¬¦  «        dej!        deeef         fd„¦   «         Z" ed¬¦  «        de
ddfd„¦   «         Z#dS )é    )ÚCallable)ÚAnyN)Úcompatibility)ÚGraphModule)Údefault_matchingÚextract_attrs_for_loweringÚlift_lowering_attrs_to_nodesF)Úis_backward_compatibleÚnameÚtarget_versionÚreturnc                 ó   — | S )zDefault matching method© )r   r   s     úY/var/www/html/CA-Chatbot/venv/lib/python3.11/site-packages/torch/fx/passes/param_fetch.pyr   r      s	   € ð €Kó    é   ÚweightÚbias)r   r   Úkernel_sizeÚstrideÚpaddingÚdilationÚgroupsÚpadding_modeé   )r   r   Úrunning_meanÚrunning_varÚeps)r   r   r   r   Úreturn_indicesÚ	ceil_modeÚinplaceÚmodule_fetch_bookÚmodc           	      ó®  — i }t          j        | ¦  «        |d<   t          | ¦  «        t          v r�t          t          | ¦  «                 \  }}}|| j        k     r0t          d|› dt          j        | ¦  «        › d| j        › d�¦  «        ‚|D ]$}t          |  ||| j        ¦  «        ¦  «        ||<   Œ%n$t          t          j        | ¦  «        › d�¦  «        ‚|S )z³If `mod` is in `module_fetch_book`, fetch the mod's attributes that in the `module_fetch_book`
    after checking module's version is compatible with the `module_fetch_book`.
    r   zFetcher version z try to fetch z	 version zl, please upgrade the module_fetch_book, open an issue and @842974287 or report a bug to AIACC team directly.z’ is not in the module_fetch_book yet, please add it to the module_fetch_book, open an issue and @842974287 or report a bug to AIACC team directly.)ÚtorchÚtypenameÚtyper"   Ú_versionÚRuntimeErrorÚgetattr)r#   Úattrs_for_loweringÚversionÚparam_to_fetchÚmatching_methodÚattrs         r   r   r   :   s  € ð
 *,ÐÝ!&¤°Ñ!4Ô!4Ð�vÑåˆC�y„yÕ%Ð%Ð%Ý3DÅTÈ#ÁYÄYÔ3OÑ0ˆ� Ø�S”\Ò!Ð!Ýð: 7ð :ð :½%¼.ÈÑ:MÔ:Mð :ð :ÐX[ÔXdð :ð :ð :ñô ð ð
 #ð 	Yð 	YˆDÝ'.¨s°O°OÀDÈ#Ì,Ñ4WÔ4WÑ'XÔ'XÐ˜tÑ$Ð$ð	Yõ ÝŒ~˜cÑ"Ô"ð 6ð 6ð 6ñ
ô 
ð 	
ð
 Ðr   Ú	fx_modulec                 ó0  — t          |                      ¦   «         ¦  «        }| j        j        D ]g}|j        dk    rZt          ||j                 t          ¦  «        rt          ||j                 ¦  «         ŒHt          ||j                 ¦  «        |_
        ŒhdS )zjRecursively traverse all `fx_module` nodes and fetch the module's attributes if the node is a leaf module.Úcall_moduleN)ÚdictÚnamed_modulesÚgraphÚnodesÚopÚ
isinstanceÚtargetr   r	   r   r+   )r0   Ú
submodulesÚnodes      r   r	   r	   U   s–   € õ �i×-Ò-Ñ/Ô/Ñ0Ô0€Jà”Ô%ð ð ˆØŒ7�mÒ#Ð#Ý˜* T¤[Ô1µ;Ñ?Ô?ð Ý,¨Z¸¼Ô-DÑEÔEÐEÐEå*DØ˜tœ{Ô+ñ+ô +�Ô'øðð r   )$Úcollections.abcr   Útypingr   r%   Útorch.nnÚnnÚtorch.fx._compatibilityr   Útorch.fx.graph_moduler   Ú__all__ÚstrÚintr   ÚmodulesÚlinearÚLinearÚconvÚConv2dÚ	batchnormÚBatchNorm2dÚpoolingÚAdaptiveAvgPool2dÚ	MaxPool2dÚ
activationÚReLUr"   r3   r'   ÚtupleÚlistÚ__annotations__ÚModuler   r	   r   r   r   ú<module>rU      sQ  ðØ $Ð $Ð $Ð $Ð $Ð $Ð $Ø Ð Ð Ð Ð Ð à €€€Ø Ð Ð Ð Ð Ð Ø 1Ð 1Ð 1Ð 1Ð 1Ð 1Ø -Ð -Ð -Ð -Ð -Ð -ðð ð €ð € eÐ,Ñ,Ô,ð˜3ð °ð ¸ð ð ð ñ -Ô,ðð 
„HÔÔÔ" Q¨°6Ð(:Ð<LÐ$MØ	„HÔÔÔ Ø	ð		
ð 		
ð 		
ð 	ð#ð 
„HÔÔÔ*Ø	Ø@Ð@Ð@Øð-ð
 
„HÔÔÔ.°°BÐ8HÐ0IØ	„HÔÔÔ&Ø	ØWÐWÐWØð)ð
 
„HÔÔÔ$ q¨9¨+Ð7GÐ&Hð7SÐ �4˜˜e C¨¨c¬°H¸cÀ3¸ZÈ¸_Ô4MÐ$MÔNÐNÔOð ð ñ ð> € eÐ,Ñ,Ô,ð B¤Ið °$°s¸C°x´.ð ð ð ñ -Ô,ðð4 € eÐ,Ñ,Ô,ð¨Kð ¸Dð ð ð ñ -Ô,ðð ð r   