o H&i @shUddlmZddlmZmZmZmZmZmZddl Z ddl m Z ddl m Z gdZe ddded ed efd d Ze j jjjd ddgefe j jjjd gdefe j jjjdgdefe j jjjd gefe j jjjd gdefe j jjjd dgefiZeeeeeeeeegefffed<e ddde j d eeeffddZ!e ddded dfddZ"dS)) GraphModule)AnyCallableDictListTupleTypeN) compatibility)default_matchingextract_attrs_for_loweringlift_lowering_attrs_to_nodesF)Zis_backward_compatiblenametarget_versionreturncCs|S)zDefault matching method )r rrrFC:\wamp64\www\opt\env\Lib\site-packages\torch/fx/passes/param_fetch.pyr sr weightbias)rr kernel_sizestridepaddingdilationgroupsZ padding_mode)rrZ running_meanZ running_varZeps)rrrrZreturn_indicesZ ceil_modeZinplacemodule_fetch_bookmodcCsi}t||d<t|tvrAtt|\}}}||jkr/td|dt|d|jd|D] }t||||j||<q1|Stt|d)zIf `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 zFetcher 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.)torchtypenametyper_version RuntimeErrorgetattr)rattrs_for_loweringversionZparam_to_fetchZmatching_methodattrrrrr !s  $r fx_modulecCsXt|}|jjD]}|jdkr)t||jtr!t||jq t ||j|_ q dS)zoRecursively traverse all `fx_module` nodes and fetch the module's attributes if the node is a leaf module. Z call_moduleN) dictZ named_modulesgraphnodesop isinstancetargetrr r r#)r&Z submodulesnoderrrr 7s   r )#Ztorch.fx.graph_modulertypingrrrrrrrZtorch.nnnnZtorch.fx._compatibilityr __all__strintr modulesZlinearZLinearconvZConv2dZ batchnormZ BatchNorm2dZpoolingZAdaptiveAvgPool2dZ MaxPool2dZ activationZReLUr__annotations__Moduler r rrrrs.       .