sgWBUddlZddlmZddlmZmZddlmZddlm Z ddl m Z gdZ eZ eZee e fed<d Zd ee e ffd ZeZee e fed <iZee e fed <iZee e fed<ddZd ee e ffdZd ee e ffdZdee e fd ee e ffdZy)N) OrderedDict)AnyDict)FixedQParamsFakeQuantize) ObserverBase)Pattern)get_default_fusion_patternsget_default_quant_patterns.get_default_output_activation_post_process_map_DEFAULT_FUSION_PATTERNScfd}|S)Nc|t<|SN)r )fnpatterns Y/var/www/html/venv/lib/python3.12/site-packages/torch/ao/quantization/fx/pattern_utils.pyinsertz(_register_fusion_pattern..inserts,. ) )rrs` r_register_fusion_patternrs Mrreturnc4tjtSr)copyr rrrr r s 99- ..r_DEFAULT_QUANTIZATION_PATTERNS!_DEFAULT_OUTPUT_FAKE_QUANTIZE_MAP_DEFAULT_OUTPUT_OBSERVER_MAPcfd}|S)Ncj|t<&tjt<t<|S)N)observer)rr with_argsrr)rfixed_qparams_observerrs rrz'_register_quant_pattern..insert/sE24&w/ ! -)22.get_len]sH  gu % %wt}$ %  1HC rcFt|dtr |d SdS)Nrr)r*)kvr/s rz'_sorted_patterns_dict..ns"jA.FGBqEN?Ar)key)rsorteditems)r&r/s @r_sorted_patterns_dictr6Ss.      !M  rr)r collectionsrtypingrr#torch.ao.quantization.fake_quantizertorch.ao.quantization.observerrtorch.ao.quantization.utilsr__all__QuantizeHandlerr __annotations__rr rrrr"r r r6rrrr?s #H7/ N EG!4(@#AF?Ad7O#;<A 5D/)A$B5 7 '<  7"01 '? "#r