Yh 2dZddlZddlmZddlmZddlmZm Z GddeZ dd ej j d ee ed eee d ed ej jf dZd ej j fdZd ej j d eed eefdZdS)zKThis module contains utility method for mobile model optimization and lint.N)Enum)_MobileOptimizerType)OptionalAnyStrceZdZdZdZdZdZdS)LintCodeN)__name__ __module__ __qualname__ BUNDLED_INPUT REQUIRES_GRADDROPOUT BATCHNORMn/var/www/tools.fuzzalab.pt/emblema-extractor/venv/lib/python3.11/site-packages/torch/utils/mobile_optimizer.pyrr s"MMGIIIrrCPU script_moduleoptimization_blocklistpreserved_methodsbackendreturncZttjjs t dt d|t }|g}d|D}t|}tfd|Drtt ||z}fd|D}|r%tdd || }|d kr'tj j||}nh|d kr'tj j||}n;|d kr&tj j|}nt d tjj|S) a Optimize a torch script module for mobile deployment. Args: script_module: An instance of torch script module with type of ScriptModule. optimization_blocklist: A set with type of MobileOptimizerType. When set is not passed, optimization method will run all the optimizer pass; otherwise, optimizer method will run the optimization pass that is not included inside optimization_blocklist. preserved_methods: A list of methods that needed to be preserved when freeze_module pass is invoked backend: Device type to use for running the result model ('CPU'(default), 'Vulkan' or 'Metal'). Returns: A new optimized torch script module Got , but ScriptModule is expected.Nc,g|]}t|Sr)str).0methods r z'optimize_for_mobile...s'T'T'TF 'T'T'Trc38K|]}t|VdS)Nhasattrr"r#rs r z&optimize_for_mobile..1s- R Rf7=& ) ) R R R R R Rrc4g|]}t||Srr&r(s rr$z'optimize_for_mobile..4s*jjjF7S`bhKiKijjjjrzAThe following methods to preserve do not exist in script_module: z, cpuvulkanmetalz:Unknown backend, must be one of 'CPU', 'Vulkan' or 'Metal') isinstancetorchjit ScriptModule TypeErrortypeset(_get_bundled_inputs_preserved_attributesalllistAttributeErrorjoinlower_C_jit_pass_optimize_for_mobile_c$_jit_pass_vulkan_optimize_for_mobile#_jit_pass_metal_optimize_for_mobile _recursivewrap_cpp_module)rrrrpreserved_methods_strbundled_inputs_attributesnon_exist_methodsoptimized_cpp_modules` roptimize_for_mobilerFs$ mUY%; < <I G4 && G G GII I%!$  (U'TBS'T'T'T HXm n n R R R R8Q R R RRR] $S)>AZ)Z%[%[ \ \jjjj.Cjjjp nPTPYPYZkPlPl n npp pmmooG%$xEE   " ! # # H  $xLL   " ! # # G  $xKKML\^sttTUUU 9  / /0D E EErct|tjjs t dt |dg}t |ds'|tj j dd| D]7\}}|j r+|tj j d|dd8tj|}|D]`}d|vr+|tjj d |d dd |vr+|tjj d |d da|S) z Generate a list of lints for a given torch script module. Args: script_module: An instance of torch script module with type of ScriptModule. Returns: lint_map: A list of dictionary that contains modules lints rr$_generate_bundled_inputs_for_forwardzNo bundled input for forward, please add bundled inputs before saving the module using torch.utils.bundled_inputs.augment_model_with_bundled_inputs.)namemessagezParam zw requires grad, please set torch.no_grad() to reduce memory usage and improve computation speed during inference phase.dropoutz Operator z exists, remember to call eval() before saving the module.and call torch.utils.mobile_optimizer.optimize_for_mobile to drop dropout operator. batch_normz exists, remember to call eval() before saving the module and call torch.utils.mobile_optimizer.optimize_for_mobile to drop batch_norm operator.)r.r/r0r1r2r3r'appendrrrInamed_parameters requires_gradrexport_opnamesrr)r lint_listrIparamop_namesop_names rgenerate_mobile_module_lintsrULs mUY%; < <I G4 && G G GII II ="H I I{("8"=Kyzz { { {%557733 e   3   h&<&AO1W[O1O1O122 3 3 3y'' 66H , ,      h&6&;**W******++ , , , 7 " "   h&8&=**W******++ , , , rcg}t|dr*|d|dt|drw|d|}|D]K}||vr|||d|z|d|zL|S)Nget_all_bundled_inputsget_num_bundled_inputs%get_bundled_inputs_functions_and_infoget_all_bundled_inputs_for__bundled_inputs_deflated_)r'rMrY)rrrCall_info function_names rr5r5us "}677C!(()ABBB!(()ABBB}EFFZ!(()PQQQ FFHH% Z ZM$555)00??? % , ,-J]-Z [ [ [ % , ,-H=-X Y Y Y Y $$r)NNr)__doc__r/enumrtorch._CrMobileOptimizerTypetypingrrrr0r1r4r7r!RecursiveScriptModulerFrUr5rrrrds^QQ @@@@@@########tFJ48 :F:Fy-:F (-@)A B:F$DL1:F :F"'!@ :F:F:F:Fz' 0F''''R%EI