§
    &Vji‚  ã                   ó”  — U 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	 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 )é    )ÚAnyÚCallableN)Ú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     úU/root/voice-cloning/.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 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   9   s  € ð
 *,ÐÝ!&¤°Ñ!4Ô!4ÐvÑåˆCy„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	   T   s–   € õ i×-Ò-Ñ/Ô/Ñ0Ô0€Jà”Ô%ð ð ˆØŒ7mÒ#Ð#Ý˜* T¤[Ô1µ;Ñ?Ô?ð Ý,¨Z¸¼Ô-DÑEÔEÐEÐEå*DØ˜tœ{Ô+ñ+ô +Ô'øðð r   )#Útypingr   r   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>rT      sE  ðØ  Ð  Ð  Ð  Ð  Ð  Ð  Ð  Ð  à €€€Ø Ð Ð Ð Ð Ð Ø 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   