9jG.ddlZddlmZmZmZmZmZmZmZm Z ddl m Z gdZ GddejjZGddeZGd d eZGd d eZGd deZGddeZGddeZGddeZGddeZGddeZGddeZGddeZGddeZGdd eZGd!d"eZGd#d$eZGd%d&eZGd'd(eZ y))N) BatchNorm1d BatchNorm2d BatchNorm3dConv1dConv2dConv3dLinearReLU)type_before_parametrizations) ConvReLU1d ConvReLU2d ConvReLU3d LinearReLUConvBn1dConvBn2d ConvBnReLU1d ConvBnReLU2dConvBn3d ConvBnReLU3dBNReLU2dBNReLU3d LinearBn1dLinearLeakyReLU LinearTanh ConvAdd2d ConvAddReLU2dc eZdZy) _FusedModuleN)__name__ __module__ __qualname__c/media/conek/DATA/Code/OCR/venv/lib/python3.12/site-packages/torch/ao/nn/intrinsic/modules/fused.pyrr&sr#rc"eZdZdZfdZxZS)r zThis is a sequential container which calls the Conv1d and ReLU modules. During quantization this will be replaced with the corresponding fused module.ct|tk(rt|tk(s7tdt|jdt|jt |||yN#Incorrect types for input modules: z and )r rr AssertionErrorrsuper__init__selfconvrelu __class__s r$r+zConvReLU1d.__init__.h ( .& 8,T2d: 5/5>>?u/5>>?A  t$r#rr r!__doc__r+ __classcell__r0s@r$r r *V % %r#r c"eZdZdZfdZxZS)r zThis is a sequential container which calls the Conv2d and ReLU modules. During quantization this will be replaced with the corresponding fused module.ct|tk(rt|tk(s7tdt|jdt|jt |||yr')r rr r)rr*r+r,s r$r+zConvReLU2d.__init__?r1r#r2r5s@r$r r ;r6r#r c"eZdZdZfdZxZS)rzThis is a sequential container which calls the Conv3d and ReLU modules. During quantization this will be replaced with the corresponding fused module.ct|tk(rt|tk(s7tdt|jdt|jt |||yr')r rr r)rr*r+r,s r$r+zConvReLU3d.__init__Pr1r#r2r5s@r$rrLr6r#rc"eZdZdZfdZxZS)rzThis is a sequential container which calls the Linear and ReLU modules. During quantization this will be replaced with the corresponding fused module.ct|tk(rt|tk(s7tdt|jdt|jt |||yr')r r r r)rr*r+)r-linearr/r0s r$r+zLinearReLU.__init__ash ( 0F :,T2d: 5/7@@A/5>>?A  &r#r2r5s@r$rr]sV ' 'r#rc"eZdZdZfdZxZS)rzThis is a sequential container which calls the Conv 1d and Batch Norm 1d modules. During quantization this will be replaced with the corresponding fused module.ct|tk(rt|tk(s7tdt|jdt|jt |||yr')r rrr)rr*r+r-r.bnr0s r$r+zConvBn1d.__init__rg ( .& 8,R0K? 5/5>>?u/3<<=?  r"r#r2r5s@r$rrnV # #r#rc"eZdZdZfdZxZS)rzThis is a sequential container which calls the Conv 2d and Batch Norm 2d modules. During quantization this will be replaced with the corresponding fused module.ct|tk(rt|tk(s7tdt|jdt|jt |||yr')r rrr)rr*r+r@s r$r+zConvBn2d.__init__rBr#r2r5s@r$rrrCr#rc"eZdZdZfdZxZS)rzThis is a sequential container which calls the Conv 1d, Batch Norm 1d, and ReLU modules. During quantization this will be replaced with the corresponding fused module.c .t|tk(r$t|tk(rt|tk(sMt dt|j dt|j dt|j t ||||yNr(z, z, and )r rrr r)rr*r+r-r.rAr/r0s r$r+zConvBnReLU1d.__init__ ( .& 8,R0K?,T2d: 5/5>>?r/3<<=V/5>>?A  r4(r#r2r5s@r$rrV ) )r#rc"eZdZdZfdZxZS)rzThis is a sequential container which calls the Conv 2d, Batch Norm 2d, and ReLU modules. During quantization this will be replaced with the corresponding fused module.c .t|tk(r$t|tk(rt|tk(sMt dt|j dt|j dt|j t ||||yrH)r rrr r)rr*r+rIs r$r+zConvBnReLU2d.__init__rJr#r2r5s@r$rrrKr#rc"eZdZdZfdZxZS)rzThis is a sequential container which calls the Conv 3d and Batch Norm 3d modules. During quantization this will be replaced with the corresponding fused module.ct|tk(rt|tk(s7tdt|jdt|jt |||yr')r rrr)rr*r+r@s r$r+zConvBn3d.__init__rBr#r2r5s@r$rrrCr#rc"eZdZdZfdZxZS)rzThis is a sequential container which calls the Conv 3d, Batch Norm 3d, and ReLU modules. During quantization this will be replaced with the corresponding fused module.c .t|tk(r$t|tk(rt|tk(sMt dt|j dt|j dt|j t ||||yrH)r rrr r)rr*r+rIs r$r+zConvBnReLU3d.__init__rJr#r2r5s@r$rrrKr#rc"eZdZdZfdZxZS)rzThis is a sequential container which calls the BatchNorm 2d and ReLU modules. During quantization this will be replaced with the corresponding fused module.ct|tk(rt|tk(s7tdt|jdt|jt |||yr')r rr r)rr*r+r- batch_normr/r0s r$r+zBNReLU2d.__init__h ( 4 C,T2d: 5/ ;DDEU/5>>?A  T*r#r2r5s@r$rrV + +r#rc"eZdZdZfdZxZS)rzThis is a sequential container which calls the BatchNorm 3d and ReLU modules. During quantization this will be replaced with the corresponding fused module.ct|tk(rt|tk(s7tdt|jdt|jt |||yr')r rr r)rr*r+rTs r$r+zBNReLU3d.__init__rVr#r2r5s@r$rrrWr#rc"eZdZdZfdZxZS)rzThis is a sequential container which calls the Linear and BatchNorm1d modules. During quantization this will be replaced with the corresponding fused module.ct|tk(rt|tk(s7tdt|jdt|jt |||yr')r r rr)rr*r+)r-r=rAr0s r$r+zLinearBn1d.__init__sg ( 0F :,R0K? 5/7@@A/3<<=?  $r#r2r5s@r$rrr6r#rc"eZdZdZfdZxZS)rzThis is a sequential container which calls the Linear and LeakyReLU modules. During quantization this will be replaced with the corresponding fused module.ct|tur%t|tjjus7t dt|j dt|j t|!||yr') typer torchnn LeakyReLUr)rr*r+)r-r= leaky_relur0s r$r+zLinearLeakyReLU.__init__snV &4 +;uxx?Q?Q+Q 5<(()tJ/?/H/H.IK  ,r#r2r5s@r$rr sV--r#rc"eZdZdZfdZxZS)rzThis is a sequential container which calls the Linear and Tanh modules. During quantization this will be replaced with the corresponding fused module.ct|tur%t|tjjus7t dt|j dt|j t|!||yr') r^r r_r`Tanhr)rr*r+)r-r=tanhr0s r$r+zLinearTanh.__init__sjV &4:+F 5<(()tDz/B/B.CE  &r#r2r5s@r$rrsV''r#rc(eZdZdZfdZdZxZS)rzThis is a sequential container which calls the Conv2d modules with extra Add. During quantization this will be replaced with the corresponding fused module.c2t||||_yN)r*r+add)r-r.rjr0s r$r+zConvAdd2d.__init__+s r#c8|j|d||S)z4Applies convolution to x1 and adds the result to x2.r)rjr-x1x2s r$forwardzConvAdd2d.forward/sxxQ R((r#rr r!r3r+ror4r5s@r$rr'sV)r#rc(eZdZdZfdZdZxZS)rzThis is a sequential container which calls the Conv2d, add, Relu. During quantization this will be replaced with the corresponding fused module.c@t||||_||_yri)r*r+rjr/)r-r.rjr/r0s r$r+zConvAddReLU2d.__init__8s  r#cV|j|j|d||S)zCApplies convolution to x1, adds the result to x2, and applies ReLU.r)r/rjrls r$rozConvAddReLU2d.forward=s&yy'$q'"+r233r#rpr5s@r$rr4sV 4r#r)!r_torch.nnrrrrrrr r torch.nn.utils.parametrizer __all__r` Sequentialrr r rrrrrrrrrrrrrrrr"r#r$rxs    D , 588&& %%"%%"%%"''"#|#"#|#")<)&)<)&#|#")<)&+|+"+|+"%%" -l - ' ' ) ) 4L 4r#