Ph%ddlZddlmZddlmZmZmZddlmZddl m Z m Z m Z ddl mZmZmZgdZe dej$j&j(ZGd d ej$j&j(ZGd d eej,ZGd deej.ZGddeej0Zy)N)_single_pair_triple) _FusedModule)TupleTypeVarUnion) _size_1_t _size_2_t _size_3_t)Conv1dConv2dConv3dMOD)boundceZdZeZ ddededeedfdeedfdeedfdeedfd ed eedfd ed ed eddfdZ dZ e dZ dZ y)_ConvNdN in_channels out_channels kernel_size.stridepaddingdilation transposedoutput_paddinggroupsbias padding_modereturnc| |d}tjjjj|||||||||| | | f i|| sJd| |_| j ||_y)N)devicedtypez'qconfig must be provided for QAT module)factory_kwargs)nnmodulesconvr__init__qconfigweightweight_fake_quant)selfrrrrrrrrrrrr(r!r"r#s gC:\Users\daisl\Desktop\realtime-object-detection\venv\Lib\site-packages\torch/ao/nn/qat/modules/conv.pyr'z_ConvNd.__init__sx%+U; (({L+)/(J)7| gWe gAAAw !(~!Ncn|j||j|j|jSN _conv_forwardr*r)rr+inputs r,forwardz_ConvNd.forward+*!!%)?)? )LdiiXXr-c Rt||jk(s.Jd|jzdz|jjzt|dsJd|jsJdt t|t r|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 r(z,Input float module must have qconfig definedz,Input float module must have a valid qconfigrN)rrrrrrr()type _FLOAT_MODULE__name__hasattrr( issubclassrrrrrrrrrrr))clsmodr(qat_convs r, from_floatz_ConvNd.from_float.sCyC--- ll + ,(( ) - sI&V(VV&{{JJJ{ d3i .a&C++s(8(8#//!jj#++ !jjsxxt/C$'$4$4gG** r-c t|}|j|j|j|j|j |j |j|j|jdu|j }tjj|jj|_|j 99 **499+;+;+=>DI c< (fG3 45 55))+D NN4 %C%%w/E KK &LKr-)NNN)r9 __module__ __qualname__rr8intrboolstrr'r4 staticmethodr?rKr-r,rrsMO!O"O$CHoOsCx O  S/ O !c? O"O"'sCxOOO #O!%O.Y4r-rceZdZdZej Zej Z ddedede de de e e fde d ed e d e d dffd Z efdZxZS)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 Nrrrrrrrrrrc t|} t|}t|tr|n t|}t|}t|||| |||dtd||| | | | yNFr) rrrrrrrrr(r!r")r isinstancerPsuperr'r+rrrrrrrrrr(r!r" kernel_size_stride_padding_ dilation_ __class__s r,r'zConv1d.__init__ww{+ &/(#67GGrys ::.((AA  e2::??223Ubjjoo%%Un3,Wbii3,j8,Wbii8,t8,Wbii8,r-