Χg^ :dZddlZddlmZddlmZddlmZm Z m Z m Z GddeZ dd ej jd ee ed ee e d ed ej jf dZd ej jfdZd ej jd e ed e efdZdS)zKThis module contains utility method for mobile model optimization and lint.N)Enum)_MobileOptimizerType)OptionalSetListAnyStrceZdZdZdZdZdZdS)LintCodeN)__name__ __module__ __qualname__ BUNDLED_INPUT REQUIRES_GRADDROPOUT BATCHNORMX/var/www/html/ai-engine/env/lib/python3.11/site-packages/torch/utils/mobile_optimizer.pyr r s"MMGIIIrr CPU script_moduleoptimization_blocklistpreserved_methodsbackendreturncttjjs t dt d|t }|g}d|D}t|}tfd|Drtt ||z}g}|D]'}t|s| |(|r%tdd ||}|dkr'tjj||}nh|d kr'tjj||}n;|d kr&tjj|}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)N)hasattr)r$r%rs r z&optimize_for_mobile..1s- R Rf7=& ) ) R R R R R RrzAThe 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_attributesalllistr(appendAttributeErrorjoinlower_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_methodsr%optimized_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%[%[ \ \'--}f-- -  $ $V , , ,p 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 r r!$_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.r/r0r1r2r(r7r rrInamed_parameters requires_gradrexport_opnamesrr)r lint_listrIparamop_namesop_names rgenerate_mobile_module_lintsrTOs 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(r7rX)rrrCall_info function_names rr4r4xs "}677C!(()ABBB!(()ABBB}EFFZ!(()PQQQ FFHH% Z ZM$555)00??? % , ,-J]-Z [ [ [ % , ,-H=-X Y Y Y Y $$r)NNr)__doc__r.enumrtorch._CrMobileOptimizerTypetypingrrrrr r/r0r#RecursiveScriptModulerFrTr4rrrrcsjQQ @@@@@@............tFJ48 =F=Fy-=F (-@)A B=F$DL1=F =F"'!@ =F=F=F=F@' 0F''''R%EI