Ph:%ddlZddlmZmZmZmZmZmZmZm Z ddl m Z gdZ GddejjZGddeZGd d eZGd d eZGd deZGddeZGddeZGddeZGddeZGddeZGddeZGddeZGddeZGdd eZGd!d"eZGd#d$eZGd%d&eZGd'd(eZ y))N)Conv1dConv2dConv3dReLULinear BatchNorm1d BatchNorm2d BatchNorm3d)type_before_parametrizations) ConvReLU1d ConvReLU2d ConvReLU3d LinearReLUConvBn1dConvBn2d ConvBnReLU1d ConvBnReLU2dConvBn3d ConvBnReLU3dBNReLU2dBNReLU3d LinearBn1dLinearLeakyReLU LinearTanh ConvAdd2d ConvAddReLU2dc eZdZy) _FusedModuleN)__name__ __module__ __qualname__nC:\Users\daisl\Desktop\realtime-object-detection\venv\Lib\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(sJdt|t|t|||yNz!Incorrect types for input modules)r rrsuper__init__selfconvrelu __class__s r$r)zConvReLU1d.__init__Z+D1V;@\]a@bfj@j y/0LT0R/STpquTvSw x yj t$r#rr r!__doc__r) __classcell__r.s@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(sJdt|t|t|||yr')r rrr(r)r*s r$r)zConvReLU2d.__init__r/r#r0r3s@r$r r r4r#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(sJdt|t|t|||yr')r rrr(r)r*s r$r)zConvReLU3d.__init__ r/r#r0r3s@r$rrr4r#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(s)Jdjt|t|t|||yNz%Incorrect types for input modules{}{})r rrformatr(r))r+linearr-r.s r$r)zLinearReLU.__init__(s_+F3v=B^_cBdhlBl Z 3 : :,V46RSW6X Z Zl &r#r0r3s@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(sJdt|t|t|||yr')r rrr(r)r+r,bnr.s r$r)zConvBn1d.__init__1Z+D1V;@\]_@`do@o w/0LT0R/STpqsTtSu v wo r"r#r0r3s@r$rr.V##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(sJdt|t|t|||yr')r rr r(r)r@s r$r)zConvBn2d.__init__9rBr#r0r3s@r$rr6rCr#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.ct|tk(r$t|tk(rt|tk(s3Jdj t|t|t|t ||||yNz'Incorrect types for input modules{}{}{})r rrrr<r(r)r+r,rAr-r.s r$r)zConvBnReLU1d.__init__Au+D1V;@\]_@`do@o ( .$ 6 ~8a V068TUW8XZvw{Z| } ~7 r4(r#r0r3s@r$rr>V))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.ct|tk(r$t|tk(rt|tk(s3Jdj t|t|t|t ||||yrH)r rr rr<r(r)rIs r$r)zConvBnReLU2d.__init__JrJr#r0r3s@r$rrGrKr#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(sJdt|t|t|||yr')r rr r(r)r@s r$r)zConvBn3d.__init__SrBr#r0r3s@r$rrPrCr#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.ct|tk(r$t|tk(rt|tk(s3Jdj t|t|t|t ||||yrH)r rr rr<r(r)rIs r$r)zConvBnReLU3d.__init__[rJr#r0r3s@r$rrXrKr#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(s)Jdjt|t|t|||yr;)r r rr<r(r)r+ batch_normr-r.s r$r)zBNReLU2d.__init__e_+J7;FKghlKmquKu ^ 3 : :,Z8:VW[:\ ^ ^u T*r#r0r3s@r$rrbV++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(s)Jdjt|t|t|||yr;)r r rr<r(r)rTs r$r)zBNReLU3d.__init__nrVr#r0r3s@r$rrkrWr#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(sJdt|t|t|||yr')r rrr(r))r+r=rAr.s r$r)zLinearBn1d.__init__xsZ+F3v=B^_aBbfqBq y/0LV0T/UVrsuVvUw x yq $r#r0r3s@r$rrur4r#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|tk(r&t|tjjk(sJdt|t|t |||yr')typertorchnn LeakyReLUr(r))r+r= leaky_relur.s r$r)zLinearLeakyReLU.__init__s[F|v%$z*:ehh>P>P*P Q/V ~d:>N=O P QP ,r#r0r3s@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|tk(r&t|tjjk(sJdt|t|t |||yr')r^rr_r`Tanhr(r))r+r=tanhr.s r$r)zLinearTanh.__init__sVF|v%$t* *E K/V ~d4j\ J KE &r#r0r3s@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,rjr.s r$r)zConvAdd2d.__init__s r#c8|j|d||SNr)rjr+x1x2s r$forwardzConvAdd2d.forwardsxxQ R((r#rr r!r1r)rpr2r3s@r$rrsV)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-r.s r$r)zConvAddReLU2d.__init__s  r#cV|j|j|d||Srl)r-rjrms r$rpzConvAddReLU2d.forwards&yy'$q'"+r233r#rqr3s@r$rrsV 4r#r)!r_torch.nnrrrrrrr r torch.nn.utils.parametrizer __all__r` Sequentialrr r rrrrrrrrrrrrrrrr"r#r$rys ```C X  588&& %%%%%%''#|##|#)<))<)#|#)<)+|++|+%%-l-'') ) 4L 4r#