§
    ×§gÇ  ã                   ó¤  — U d dl mZ d dlmZmZmZmZmZmZ d dl	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 )é    )ÚGraphModule)ÚAnyÚCallableÚDictÚListÚTupleÚTypeN)Úcompatibility)Ú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     úW/var/www/html/ai-engine/env/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 rt          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ÑåˆCy„yÕ%Ð%Ð%Ý3DÅTÈ#ÁYÄYÔ3OÑ0ˆ ØS”\Ò!Ð!Ýð  I°'ð  Ið  IÍÌÐX[ÑI\ÔI\ð  Ið  IÐgjÔgsð  Ið  Ið  Iñ Jô Jð Jð #ð 	Yð 	YˆDÝ'.¨s°O°OÀDÈ#Ì,Ñ4WÔ4WÑ'XÔ'XÐ˜tÑ$Ð$ð	Yõ eœn¨SÑ1Ô1ð Eð Eð Eñ Fô Fð 	Fð Ðr   Ú	fx_modulec                 ó0  — t          |                      ¦   «         ¦  «        }| j        j        D ]g}|j        dk    rZt          ||j                 t          ¦  «        rt          ||j                 ¦  «         ŒHt          ||j                 ¦  «        |_
        ŒhdS )zoRecursively 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/   )r4   Ú
submodulesÚnodes      r   r   r   7   s˜   € õ i×-Ò-Ñ/Ô/Ñ0Ô0€Jà”Ô%ð ^ð ^ˆØŒ7mÒ#Ð#Ý˜* T¤[Ô1µ;Ñ?Ô?ð ^Ý,¨Z¸¼Ô-DÑEÔEÐEÐEå*DÀZÐPTÔP[ÔE\Ñ*]Ô*]Ô'øð^ð ^r   )#Útorch.fx.graph_moduler   Útypingr   r   r   r   r   r	   r)   Útorch.nnÚnnÚtorch.fx._compatibilityr
   Ú__all__ÚstrÚintr   ÚmodulesÚlinearÚLinearÚconvÚConv2dÚ	batchnormÚBatchNorm2dÚpoolingÚAdaptiveAvgPool2dÚ	MaxPool2dÚ
activationÚReLUr&   Ú__annotations__ÚModuler   r   r   r   r   ú<module>rV      sR  ðØ -Ð -Ð -Ð -Ð -Ð -Ð -Ø 9Ð 9Ð 9Ð 9Ð 9Ð 9Ð 9Ð 9Ð 9Ð 9Ð 9Ð 9Ð 9Ð 9Ð 9Ð 9Ø €€€Ø Ð Ð Ð Ð Ð à 1Ð 1Ð 1Ð 1Ð 1Ð 1à
\Ð
\Ð
\€ð € eÐ,Ñ,Ô,ð˜3ð °ð ¸ð ð ð ñ -Ô,ðð 
„HÔÔÔ" Q¨°6Ð(:Ð<LÐ$MØ	„HÔÔÔ Ø	ÐgÐgÐgÐiyð#ð 
„HÔÔÔ*¨QÐ0hÐ0hÐ0hÐjzÐ,{Ø	„HÔÔÔ.°°BÐ8HÐ0IØ	„HÔÔÔ&Ø	ÐZÐZÐZÐ\lð)ð 
„HÔÔÔ$ q¨9¨+Ð7GÐ&HðSÐ 4˜˜e C¨¨c¬°H¸cÀ3¸ZÈ¸_Ô4MÐ$MÔNÐNÔOð ð ñ ð € eÐ,Ñ,Ô,ð B¤Ið °$°s¸C°x´.ð ð ð ñ -Ô,ðð* € eÐ,Ñ,Ô,ð
^¨Kð 
^¸Dð 
^ð 
^ð 
^ñ -Ô,ð
^ð 
^ð 
^r   