8hM(SSKrSSKJrJrJrJrJrJrJrJ r SSK J r /SQr "SS\RR5r"SS\5r"S S \5r"S S \5r"S S\5r"SS\5r"SS\5r"SS\5r"SS\5r"SS\5r"SS\5r"SS\5r"SS\5r"SS \5r"S!S"\5r"S#S$\5r"S%S&\5r"S'S(\5r g))N) BatchNorm1d BatchNorm2d BatchNorm3dConv1dConv2dConv3dLinearReLU)type_before_parametrizations) ConvReLU1d ConvReLU2d ConvReLU3d LinearReLUConvBn1dConvBn2d ConvBnReLU1d ConvBnReLU2dConvBn3d ConvBnReLU3dBNReLU2dBNReLU3d LinearBn1dLinearLeakyReLU LinearTanh ConvAdd2d ConvAddReLU2dc\rSrSrSrg) _FusedModule&N)__name__ __module__ __qualname____firstlineno____static_attributes__r ]/var/www/fran/franai/venv/lib/python3.13/site-packages/torch/ao/nn/intrinsic/modules/fused.pyrr&sr&rc,^\rSrSrSrU4SjrSrU=r$)r *zThis is a sequential container which calls the Conv1d and ReLU modules. During quantization this will be replaced with the corresponding fused module.c>[U5[:Xa[U5[:XdS[U5[U535e[TU]X5 gNz!Incorrect types for input modules)r rr super__init__selfconvrelu __class__s r'r-ConvReLU1d.__init__.[ ( .& 8,T2d: 00LT0R/S+D12 4   ; $r&r r!r"r#r$__doc__r-r% __classcell__r2s@r'r r *V%%r&r c,^\rSrSrSrU4SjrSrU=r$)r 9zThis is a sequential container which calls the Conv2d and ReLU modules. During quantization this will be replaced with the corresponding fused module.c>[U5[:Xa[U5[:XdS[U5[U535e[TU]X5 gr+)r rr r,r-r.s r'r-ConvReLU2d.__init__=r4r&r r5r8s@r'r r 9r9r&r c,^\rSrSrSrU4SjrSrU=r$)rHzThis is a sequential container which calls the Conv3d and ReLU modules. During quantization this will be replaced with the corresponding fused module.c>[U5[:Xa[U5[:XdS[U5[U535e[TU]X5 gr+)r rr r,r-r.s r'r-ConvReLU3d.__init__Lr4r&r r5r8s@r'rrHr9r&rc,^\rSrSrSrU4SjrSrU=r$)rWzThis is a sequential container which calls the Linear and ReLU modules. During quantization this will be replaced with the corresponding fused module.c>[U5[:Xa[U5[:XdS[U5[U535e[TU]X5 gr+)r r r r,r-)r/linearr1r2s r'r-LinearReLU.__init__[s[ ( 0F :,T2d: 00LV0T/U+D12 4   ; &r&r r5r8s@r'rrWsV''r&rc,^\rSrSrSrU4SjrSrU=r$)rfzThis 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.c>[U5[:Xa[U5[:XdS[U5[U535e[TU]X5 gr+)r rrr,r-r/r0bnr2s r'r-ConvBn1d.__init__j\ ( .& 8,R0K? 00LT0R/S+B/0 2   @ "r&r r5r8s@r'rrfV##r&rc,^\rSrSrSrU4SjrSrU=r$)ruzThis 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.c>[U5[:Xa[U5[:XdS[U5[U535e[TU]X5 gr+)r rrr,r-rJs r'r-ConvBn2d.__init__yrMr&r r5r8s@r'rrurNr&rc,^\rSrSrSrU4SjrSrU=r$)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>[U5[:Xa&[U5[:Xa[U5[:Xd)S[U5[U5[U535e[TU]XU5 gr+)r rrr r,r-r/r0rKr1r2s r'r-ConvBnReLU1d.__init__u ( .& 8,R0K?,T2d: 00LT0R/S+B/0+D12 4   ; 4(r&r r5r8s@r'rrV ) )r&rc,^\rSrSrSrU4SjrSrU=r$)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>[U5[:Xa&[U5[:Xa[U5[:Xd)S[U5[U5[U535e[TU]XU5 gr+)r rrr r,r-rVs r'r-ConvBnReLU2d.__init__rXr&r r5r8s@r'rrrYr&rc,^\rSrSrSrU4SjrSrU=r$)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.c>[U5[:Xa[U5[:XdS[U5[U535e[TU]X5 gr+)r rrr,r-rJs r'r-ConvBn3d.__init__rMr&r r5r8s@r'rrrNr&rc,^\rSrSrSrU4SjrSrU=r$)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>[U5[:Xa&[U5[:Xa[U5[:Xd)S[U5[U5[U535e[TU]XU5 gr+)r rrr r,r-rVs r'r-ConvBnReLU3d.__init__rXr&r r5r8s@r'rrrYr&rc,^\rSrSrSrU4SjrSrU=r$)rzThis is a sequential container which calls the BatchNorm 2d and ReLU modules. During quantization this will be replaced with the corresponding fused module.c>[U5[:Xa[U5[:XdS[U5[U535e[TU]X5 gr+)r rr r,r-r/ batch_normr1r2s r'r-BNReLU2d.__init__[ ( 4 C,T2d: 00LZ0X/Y+D12 4   ; *r&r r5r8s@r'rrV++r&rc,^\rSrSrSrU4SjrSrU=r$)rzThis is a sequential container which calls the BatchNorm 3d and ReLU modules. During quantization this will be replaced with the corresponding fused module.c>[U5[:Xa[U5[:XdS[U5[U535e[TU]X5 gr+)r rr r,r-ris r'r-BNReLU3d.__init__rlr&r r5r8s@r'rrrmr&rc,^\rSrSrSrU4SjrSrU=r$)rzThis is a sequential container which calls the Linear and BatchNorm1d modules. During quantization this will be replaced with the corresponding fused module.c>[U5[:Xa[U5[:XdS[U5[U535e[TU]X5 gr+)r r rr,r-)r/rErKr2s r'r-LinearBn1d.__init__s\ ( 0F :,R0K? 00LV0T/U+B/0 2   @ $r&r r5r8s@r'rrr9r&rc,^\rSrSrSrU4SjrSrU=r$)rzThis is a sequential container which calls the Linear and LeakyReLU modules. During quantization this will be replaced with the corresponding fused module.c>[U5[:Xa'[U5[RR:XdS[U5[U535e[ TU]X5 gr+)typer torchnn LeakyReLUr,r-)r/rE leaky_relur2s r'r-LinearLeakyReLU.__init__sYF|v%$z*:ehh>P>P*P /V ~d:>N=O P P ,r&r r5r8s@r'rrsV--r&rc,^\rSrSrSrU4SjrSrU=r$)rzThis is a sequential container which calls the Linear and Tanh modules. During quantization this will be replaced with the corresponding fused module.c>[U5[:Xa'[U5[RR:XdS[U5[U535e[ TU]X5 gr+)ryr rzr{Tanhr,r-)r/rEtanhr2s r'r-LinearTanh.__init__sTF|v%$t* *E /V ~d4j\ J E &r&r r5r8s@r'rrsV''r&rc2^\rSrSrSrU4SjrSrSrU=r$)ri zThis is a sequential container which calls the Conv2d modules with extra Add. During quantization this will be replaced with the corresponding fused module.c0>[TU]U5 X lgN)r,r-add)r/r0rr2s r'r-ConvAdd2d.__init__ s r&c8URUS"U5U5$Nrrr/x1x2s r'forwardConvAdd2d.forwardsxxQ R((r&r r!r"r#r$r6r-rr%r7r8s@r'rr sV))r&rc2^\rSrSrSrU4SjrSrSrU=r$)rizThis is a sequential container which calls the Conv2d, add, Relu. During quantization this will be replaced with the corresponding fused module.c<>[TU]U5 X lX0lgr)r,r-rr1)r/r0rr1r2s r'r-ConvAddReLU2d.__init__s  r&cVURURUS"U5U55$r)r1rrs r'rConvAddReLU2d.forwards$yy$q'"+r233r&)rr1rr8s@r'rrsV 44r&r)!rztorch.nnrrrrrrr r torch.nn.utils.parametrizer __all__r{ Sequentialrr r rrrrrrrrrrrrrrrr r&r'rs    D , 588&&  % % % % % % ' ' #| # #| #)<)")<)" #| #)<)" +| + +| + % %-l-'' ) ) 4L 4r&