[Th6%SSKrSSKJrJrJrJr SSKrSSKJs J r SSK Js J s J r SSKJs J s J s Jr SSKJs J s J s Jr SSKJs J s J s Js Jr SSKJs J s Jr SSKJs J s Js Jr SSKJs J s Jr SSKJs J s Js Jr SSKJs J s Js J r! SSK"rSSK#J s J$r% SSKJ r SSK&J'r'J(r( SSK)J*r*J+r+ SSK,J-r- SSK.J/r/ /SQr0\+\Rb\*\Rd\ Rf\!Rf\ Rh\!Rh\ Rj\!Rj\ Rl\!Rl\ Rn\!Rn\ Rp\!Rp\ Rr\!Rr\ Rt\!Rt\ Rv\!Rv\ Rx\!Rx\ Rz\!Rz\ R|\!R|\ R~\!R~0r@\A\\4\BS '0\+\Rb_\*\Rd_\ R\R_\ R\R_\ R\R_\ Rh\Rh_\ Rj\Rj_\ Rl\Rl_\ Rn\Rn_\ Rp\Rp_\ Rr\Rr_\ R\R_\ Rt\Rt_\ Rv\Rv_\ R\R_\ R\R_\ R\R_0\ R\R_\ R\R_\ R\R_\ R\R_\ RRR\Rf_\ Rf\Rf_\ R\R_\ R\R_\ R\R_\R\R_\ R\R_\ R\R_\ R\R_\ R\R_\ R\R_\ R\R_\ R\R_E0\ R\R_\ R\R_\ R\R_\R\Rh_\R\Rj_\R\Rl_\R\R_\R\R_\R\R_\R\R_\R\R_\R\R_\R\Rf_\Rf\Rf_\Rj\Rj_\Rl\Rl_Erf\A\\4\BS '\ Rj\Rj\ Rl\Rl\ Rf\Rf\ RRR\Rf\ R\R\ R\R\ R\R\ R\R\ R\R\ R\R\ R\R\ R\R\ R\R\ R\R0rg\A\\4\BS '\ Rx\Rx\ Rf\Rf\Rf\Rf\ RRR\Rf\ R~\R~\ R\R\ Rz\Rz\ R|\R|\ R\R\ Rv\Rv\ Rt\Rt0 ri\A\\4\BS '\ R1rk\l\\BS '\%R\RR$R\%R\RR$R\%R\RR$R\%R\RR$R\%R\RR$R\%R\RR$R0rt\A\\\u4\4\BS'\ R\'\ R\'\ R\'\ R\(0rz\A\\4\BS'\ Rf\ RR$Rf0r|\A\\4\BS'\ Rf\ RR$R*Rf0r}\A\\4\BS'S\l\4Sjr~S\A\\44SjrS\A\\44SjrS\A\\44SjrS\A\\44SjrS*S\S\\A\\4S\S\4SjjrS+S\S\\A\\4S\4SjjrS\A\\44SjrS\A\\44SjrS\A\\44S jrS\A\\44S!jrS\l\4S"jrS\l\4S#jrS\A\\\u4\44S$jrS%\\\u4S\4S&jrS'\RGRS\\4S(jrS'\RGRS\4S)jrg),N)AnyCallableOptionalUnion)nn)+default_fixed_qparams_range_0to1_fake_quant.default_fixed_qparams_range_neg1to1_fake_quant) DeQuantStub QuantStub)get_combined_dict)type_before_parametrizations).DEFAULT_REFERENCE_STATIC_QUANT_MODULE_MAPPINGS$DEFAULT_STATIC_QUANT_MODULE_MAPPINGSDEFAULT_QAT_MODULE_MAPPINGS%DEFAULT_DYNAMIC_QUANT_MODULE_MAPPINGS,DEFAULT_FLOAT_TO_QUANTIZED_OPERATOR_MAPPINGS"DEFAULT_MODULE_TO_ACT_POST_PROCESS+DEFAULT_STATIC_SPARSE_QUANT_MODULE_MAPPINGS,DEFAULT_DYNAMIC_SPARSE_QUANT_MODULE_MAPPINGSno_observer_set(get_default_static_quant_module_mappings2get_default_static_quant_reference_module_mappings*get_embedding_static_quant_module_mappings/get_default_static_sparse_quant_module_mappingsget_static_quant_module_classget_dynamic_quant_module_classget_default_qat_module_mappings!get_embedding_qat_module_mappings)get_default_dynamic_quant_module_mappings0get_default_dynamic_sparse_quant_module_mappings$get_default_qconfig_propagation_list&get_default_compare_output_module_list0get_default_float_to_quantized_operator_mappingsget_quantized_operatorrrrr_INCLUDE_QCONFIG_PROPAGATE_LISTrrrrreturncn[RR[RR1nU$)z8These modules cannot have observers inserted by default.)r quantizableLSTMMultiheadAttention) no_observerss c/var/www/auris/envauris/lib/python3.13/site-packages/torch/ao/quantization/quantization_mappings.pyrrs&NN'')J)JKL c6[R"[5$)z8Get module mapping for post training static quantization)copydeepcopyrr-r,rrs === >>r-c6[R"[5$)zBGet reference module mapping for post training static quantization)r/r0rr1r-r,rrs ==G HHr-c[R"[5n[RU[ R'[R U[ R 'U$)z7Get module mapping, including mapping for embedding QAT)r/r0rnnq EmbeddingBagnnqat Embeddingmappings r,rrs>mm@AG"%"2"2GE  "}}GEOO Nr-c6[R"[5$)z?Get module mapping for post training static sparse quantization)r/r0rr1r-r,rrs ==D EEr-float_module_classadditional_static_quant_mapping is_referencecUc0n[U(a[O[U5nURUS5nUcS[ U53S-5e[ R "U5$)z]n Get the statically quantized module class corresponding to the floating point module class NFloating point module class 5 does not have a corresponding quantized module class)r rrgetstrr/r0)r;r<r= all_mappingsstatic_quant_module_classs r,rrs}'.*,'$  7 1' L !- 0 01CT J $ 0 &s+='>&?@ A B 0 ==2 33r- additional_dynamic_quant_mappingcUc0n[[U5nURUS5nUcS[U53S-5e[R "U5$)z^n Get the dynamically quantized module class corresponding to the floating point module class Nr?r@)r rrArBr/r0)r;rErCdynamic_quant_module_classs r,rrsu(/+-($-/OL".!1!12Dd!K % 1 &s+='>&?@ A B 1 ==3 44r-c6[R"[5$)z:Get default module mapping for quantization aware training)r/r0rr1r-r,rrs ==4 55r-c[R"[5n[RU[ R'[R U[ R 'U$)zGet module mapping for quantization aware training This is includes default values in addition to enabling qat for embeddings. )r/r0rr6r5rr7r8s r,rrs> mm78G$11GBOO!OOGBLL Nr-c[$)z9Get module mapping for post training dynamic quantization)rr1r-r,rr*s 00r-c[$)z@Get module mapping for post training dynamic sparse quantization)rr1r-r,r r /s 77r-c[[R55[[R55-[[R55-[ -n[ R"U5$)zWGet the default list of module types that we'll attach qconfig attribute to in prepare )setrkeysrrr%r/r0)#QCONFIG_PROPAGATE_MODULE_CLASS_LISTs r,r!r!4se 0 5 5 78 )..0 1 2 388: ; < * *( ==< ==r-c[[R55[[R55-[[R55-[[R 55-[[R 55-[[R 55-[ -n[R"U5$)zKGet list of module class types that we will record output in numeric suite ) rMrvaluesrrrNr%r/r0).NUMERIC_SUITE_COMPARE_MODEL_OUTPUT_MODULE_LISTs r,r"r"As 0 7 7 9: )002 3 4 3::< = > 2779 : ; )..0 1  2 388: ;  < *  *3 ==G HHr-c6[R"[5$N)r/r0rr1r-r,r#r#Qs ==E FFr-float_opc`[RUS5nUcS[U5S35eU$)z>Get the quantized operator corresponding to the float operatorNz Operator z) does not have corresponding quantized op)rrArB)rU quantized_ops r,r$r$Xs@?CCHdSL L 3x=/!JKL r-modulec@[R[U5S5$)zGet the special activation post process for `module`, this has higher priority than the activation post process in `qconfig` e.g. input: torch.nn.Sigmoid output: default_affine_fixed_qparam_fake_quant N)rrAr rXs r,_get_special_act_post_processr[as! . 1 1$V,d r-cLUR=(a [U5[;$rT)trainingtyperrZs r,_has_special_act_post_processr_ms ?? QtF|/QQQr-)NFrT)r/typingrrrrtorch torch.ao.nnaorao_nntorch.ao.nn.intrinsic intrinsicnnitorch.ao.nn.intrinsic.qatqatnniqattorch.ao.nn.intrinsic.quantized quantizednniq'torch.ao.nn.intrinsic.quantized.dynamicdynamicnniqdtorch.ao.nn.qatr6torch.ao.nn.qat.dynamicnnqatdtorch.ao.nn.quantizedr4torch.ao.nn.quantized.dynamicnnqdtorch.ao.nn.quantized.reference referencennqrtorch.ao.nn.sparsetorch.nn.functional functionalF#torch.ao.quantization.fake_quantizerr torch.ao.quantization.stubsr r torch.ao.quantization.utilsr torch.nn.utils.parametrizer __all__Quantize DeQuantizeLinearConv1dConv2dConv3dConvTranspose1dConvTranspose2dConvTranspose3dr7r5GRUCellLSTMCellRNNCellr)rdict__annotations__ BatchNorm2d BatchNorm3dDropoutELU GroupNorm HardswishInstanceNorm1dInstanceNorm2dInstanceNorm3d LayerNorm LeakyReLUmoduleslinearNonDynamicallyQuantizableLinearReLU6PReLUFloatFunctional QFunctionalBNReLU2dBNReLU3d ConvReLU1d ConvReLU2d ConvReLU3d ConvAdd2d ConvAddReLU2d LinearReLULinearLeakyReLU LinearTanhConvBn1dConvBn2dConvBn3d ConvBnReLU1d ConvBnReLU2d ConvBnReLU3d LinearBn1drrGRUr Sequentialr%rMeluops hardswish instance_norm layer_norm leaky_reludropoutrrB HardsigmoidSigmoidSoftmaxTanhrsparserrrrrrrboolrrrrrr r!r"r#r$Moduler[r_r1r-r,rs  11 ##**..777((##,,..?9C 8s||IIt{{IIt{{IIt{{IIt{{,,,,,,LL$..OOT&&JJ KKJJ GGTYYG.Xs]0C&6= s||6=6=NNCOO6=NNCOO 6= JJ 6= IIszz 6=IIszz6=IIszz6=++6=++6=++6=FFCGG6=LL#--6=OOS%%6=LL#--6= LL#--!6="s))#6=$s))%6=&s))'6=(LL#--)6=*LL#--+6=,JJ55szz-6=.IIszz/6=0HHcii16=2JJ 36=4HHcii56=896=<LL$--=6=>LL$--?6=@NNDOOA6=BNNDOOC6=DNNDOOE6=FMM4>>G6=Ht))I6=JNNDOOK6=L--M6=NNNDOOO6=P OOSZZQ6=R OOSZZS6=T OOSZZU6=V W6=X Y6=Z [6=\ t]6=^ t_6=` ta6=b szzc6=f LL#**g6=h LL#**i6=j LL#**k6=$d8S=&96tIIu||IIu||IIu||JJ55u||LL&//LL&//LL&//f))f))f))NNF%%NNF%%NNF%%NNF%%4T(C-0(JJ IIt{{ MM4;;JJ55t{{GGTYYFFDHHKKJJ NNE$$OOS%%LL#-->%tHcM':0MM2XEE599   " "KK$$..OOUYY((66LL%))%%00LL%))%%00IIuyy""** V,d533G3Q.RNN?JJ;JJ;GG ; @"D8);$<IIu||%%,,D+T(C--@ IIu||%%--44E,d8S=.A S ?$x}2E? ID314# 3F1 8$x}:M8 >c(m > IH I Gx} x '(GU8S=%9h %((// hx>P R%((//RdRr-