^jb':ddlmZmZddlZddlmZddlmZddlm Z m Z m Z ddl m Z mZmZgdZGddej"j$j&ZGd d eej(ZGd d eej*ZGd deej,Zy))ClassVarLiteralN) _FusedModule) _size_1_t _size_2_t _size_3_t)_pair_single_triple)Conv1dConv2dConv3dceZdZUeeej jje d< dde de de e dfde e dfde e e dfzd e e dfd e d e e dfd e d e dedddfdZdZeddZdZy)_ConvNd _FLOAT_MODULEN in_channels out_channels kernel_size.stridepaddingdilation transposedoutput_paddinggroupsbias padding_modezerosreflect replicatecircularreturnc| |d}tjjjj|||||||||| | | f i|| s t d| |_| j||_y)N)devicedtypez'qconfig must be provided for QAT module)factory_kwargs) nnmodulesconvr__init__AssertionErrorqconfigweightweight_fake_quant)selfrrrrrrrrrrrr,r$r%r&s g/var/www/ramen.bs-engineer-server.com/venv/lib/python3.12/site-packages/torch/ao/nn/qat/modules/conv.pyr*z_ConvNd.__init__s"%+U; ((               !JK K !(~!Ncn|j||j|j|jSN _conv_forwardr.r-rr/inputs r0forwardz_ConvNd.forward7*!!%)?)? )LdiiXXr1c t||jurEtd|jd|jjdt|jt |ds td|j s tdt t|tr|d}|j }||j|j|j|j|j|j|j|jdu|j | }|j"|_|j|_|S) zCreate a qat module from a float module Args: `mod`: a float module, either produced by torch.ao.quantization utilities or directly from user zqat.z.from_float only works for z, got r,z,Input float module must have qconfig definedz,Input float module must have a valid qconfigrN)rrrrrrr,)typerr+__name__hasattrr, issubclassrrrrrrrrrrr-)clsmoduse_precomputed_fake_quantr,qat_convs r0 from_floatz_ConvNd.from_float:s 9C-- - s||n$?$$--.fT#Y5G5G4HJ sI& !OP P{{ !OP P d3i .a&C++ OO    OO::KK\\::%))  ** r1c t|}|j|j|j|j|j |j |j|j|jdu|j }tjj|jj|_|jrr=r+r<rEappendrtraintraining)r/r?r)r(relufuseds r0to_floatz_ConvNd.to_float^s=4j%%          KK LL MM KK IIT !     hh((););)=> 99 **499+;+;+=>DI c< (fG3 45$||n$KL))+D NN4 &C%%w/E KK &LKr1)NNNF)r< __module__ __qualname__rr;r'r(r)r__annotations__inttuplestrboolrr*r8 staticmethodrCrOr1r0rrsD!8!89::$O$O$O38_ $O c3h $O uS#X& $OS/$O$Oc3h$O$O$OIJ$O !$OLY!!F"r1rceZdZUdZej Zeeej e d<ej Z eeej e d< dde de de de d e e zd e d e d ed edddffd Zedfd ZxZS)r aZ A Conv1d module attached with FakeQuantize modules for weight, used for quantization aware training. We adopt the same interface as :class:`~torch.nn.Conv1d` Similar to :class:`~torch.nn.Conv2d`, with FakeQuantize modules initialized to default. Attributes: weight_fake_quant: fake quant module for weight rrFNrrrrrrrrrrr"c t|} t|}t|tr|n t|}t|}t|||| |||dtd||| | | | yNFr) rrrrrrrrr,r$r%)r isinstancerVsuperr*r/rrrrrrrrrr,r$r% kernel_size_stride_padding_ dilation_ __class__s r0r*zConv1d.__init__w{+ &/(#67GGrsy$ .AA:: )rbjjoo%%rj9 Wbii9 x> Wbii> B> Wbii> r1