+ &jtRt^RIt^RIHt^RIHt^RIHt!RR]4t R RRllt R R lt R R lt R#)zKThis module contains utility method for mobile model optimization and lint.N)Enum)_MobileOptimizerType)AnyStrc&]tRt^ t^t^t^t^tRtR#)LintCodeN) __name__ __module__ __qualname____firstlineno__ BUNDLED_INPUT REQUIRES_GRADDROPOUT BATCHNORM__static_attributes__rt/Users/jameslopez/projects/CWCArchive/cwc-podcast/.venv/lib/python3.14/site-packages/torch/utils/mobile_optimizer.pyrr sMMGIrrc V^8dQhR\PPR\\,R,R\ \ ,R,R\R\PP/#) script_moduleoptimization_blocklistNpreserved_methodsbackendreturn) torchjit ScriptModulesetMobileOptimizerTypelistrstrRecursiveScriptModule)formats"r __annotate__r#si:F:Fyy--:F #$7 84 ?:F <$.:F :F"'!@!@ :Frca\S\PP4'g\ R\ S4 R24hVf \ 4pVf.pVUu.uFp\V4NK pp\SV4p\;QJdV3RlV4F 'dK RM RM!V3RlV44'd\\ WV,44pVUu.uFp\SV4'dKVNK ppV'd\RRPV4 24hVP4pVR8Xd-\PP!SP"VV4pMpVR 8Xd-\PP%SP"VV4pM=VR 8Xd,\PP'SP"V4pM \ R 4h\PP(P+V4#uupiuupi) 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.c3<<"TFp\SV4xK R#5i)N)hasattr).0methodrs& r &optimize_for_mobile..1s R8Qf7=& ) )8QsFTzAThe following methods to preserve do not exist in script_module: z, cpuvulkanmetalz:Unknown backend, must be one of 'CPU', 'Vulkan' or 'Metal') isinstancerrr TypeErrortyperr (_get_bundled_inputs_preserved_attributesallrr(AttributeErrorjoinlower_C_jit_pass_optimize_for_mobile_c$_jit_pass_vulkan_optimize_for_mobile#_jit_pass_metal_optimize_for_mobile _recursivewrap_cpp_module) rrrrr*preserved_methods_strbundled_inputs_attributesnon_exist_methodsoptimized_cpp_modules f&&& roptimize_for_mobilerCs$ mUYY%;%; < <4 &''F GI I%!$  CT'TBSF BS'T HXm n s R8Q Rsss R8Q RRR $S)>)Z%[ \.Cj.CF7S`bhKi.CjOPTPYPYZkPlOm np pmmoG%$xxEE    " ! # H $xxLL    " ! # G $xxKKML\L\^stTUU 99   / /0D EE7(U ksG>H5HcLV^8dQhR\PP/#)rr)rrr)r"s"rr#r#Ls'' 0F0F'rc \V\PP4'g\ R\ V4 R24h.p\ VR4'g.VPR\PPRR/4VP4FJwr#VP'gKVPR\PPRRV R2/4KL \PPV4pVFupR V9d2VPR\PPRR V R 2/4R V9gKDVPR\P PRR V R 2/4Kw V#)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_forwardnamemessagezNo bundled input for forward, please add bundled inputs before saving the module using torch.utils.bundled_inputs.augment_model_with_bundled_inputs.zParam 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.)r0rrrr1r2r(appendrr rGnamed_parameters requires_gradr export_opnamesrr)r lint_listrGparamop_namesop_names& rgenerate_mobile_module_lintsrSLs mUYY%;%; < <4 &''F GI II ="H I I&("8"8"="=yKyz {%557        fh&<&<&A&A9PVW[V\]1O12 38 yy'' 6H     fh&6&6&;&;'9WI>***+ , 7 "   fh&8&8&=&='9WI>***+ , rcV^8dQhR\PPR\\,R\\,/#)rrrr)rrrrr )r"s"rr#r#us9%%EIIrbs2Q @t :Fz'R%r