U ʹj@sdZddlZddlmZddlmZddlmZm Z m Z m Z GdddeZ dej jee eee e eej jd d d Zej jd d dZej je ee edddZdS)zM This module contains utility method for mobile model optimization and lint. N)Enum)_MobileOptimizerType)OptionalSetListAnyStrc@seZdZdZdZdZdZdS)LintCodeN)__name__ __module__ __qualname__ BUNDLED_INPUT REQUIRES_GRADDROPOUT BATCHNORMrr[/var/www/html/bwave/python/venv/lib/python3.8/site-packages/torch/utils/mobile_optimizer.pyr srCPU) script_moduleoptimization_blocklistpreserved_methodsbackendreturnc s&ttjjs"tdtd|dkr0t}|dkr-sz'optimize_for_mobile..c3s|]}t|VqdS)N)hasattrrrrr 0sz&optimize_for_mobile..zAThe following methods to preserve do not exist in script_module: z, cpuZvulkanZmetalz:Unknown backend, must be one of 'CPU', 'Vulkan' or 'Metal') isinstancetorchjit ScriptModule TypeErrortypeset(_get_bundled_inputs_preserved_attributesalllistr#appendAttributeErrorjoinlowerZ_CZ_jit_pass_optimize_for_mobileZ_cZ$_jit_pass_vulkan_optimize_for_mobileZ#_jit_pass_metal_optimize_for_mobileZ _recursiveZwrap_cpp_module) rrrrZpreserved_methods_strbundled_inputs_attributesZnon_exist_methodsr!Zoptimized_cpp_modulerr$roptimize_for_mobilesJ    r6r$cCst|tjjs"tdt|dg}t|dsD|tj j dd| D]*\}}|j rL|tj j d|ddqLtj|}|D]H}d|kr|tjj d |dd |kr|tjj d |dq|S) z Args: script_module: An instance of torch script module with type of ScriptModule Returns: lint_map: A list of dictionary that contains modules lints rrZ$_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.ZdropoutzOperator {} exists, remember to call eval() before saving the module.and call torch.utils.mobile_optimizer.optimize_for_mobile to drop dropout operator.Z batch_normzOperator {} 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)r*r+r,r#r1rrr7Znamed_parametersZ requires_gradrZexport_opnamesrformatr)rZ lint_listr7paramZop_namesZop_namerrrgenerate_mobile_module_lintsNs*    r;)rrrcCszg}t|dr"|d|dt|drv|d|}|D]2}||krX|||d||d|qB|S)NZget_all_bundled_inputsZget_num_bundled_inputs%get_bundled_inputs_functions_and_infoZget_all_bundled_inputs_for_Z_bundled_inputs_deflated_)r#r1r<)rrr5Zall_infoZ function_namerrrr.ss      r.)NNr)__doc__r(enumrZtorch._CrZMobileOptimizerTypetypingrrrrrr)r*rZRecursiveScriptModuler6r;r.rrrrs"     >%