EiG.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 `/home/mande/repo/quber/.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[:Xd7[S[U5RS[U5R35e[ TU]X5 gN#Incorrect types for input modules: z and )r rr AssertionErrorr!super__init__selfconvrelu __class__s r'r/ConvReLU1d.__init__.f ( .& 8,T2d: 5/5>>?u/5>>?A  $r&r r!r"r#r$__doc__r/r% __classcell__r4s@r'r r *V % %r&r c,^\rSrSrSrU4SjrSrU=r$)r ;zThis 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[:Xd7[S[U5RS[U5R35e[ TU]X5 gr+)r rr r-r!r.r/r0s r'r/ConvReLU2d.__init__?r6r&r r7r:s@r'r r ;r;r&r c,^\rSrSrSrU4SjrSrU=r$)rLzThis 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[:Xd7[S[U5RS[U5R35e[ TU]X5 gr+)r rr r-r!r.r/r0s r'r/ConvReLU3d.__init__Pr6r&r r7r:s@r'rrLr;r&rc,^\rSrSrSrU4SjrSrU=r$)r]zThis 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[:Xd7[S[U5RS[U5R35e[ TU]X5 gr+)r r r r-r!r.r/)r1linearr3r4s r'r/LinearReLU.__init__asf ( 0F :,T2d: 5/7@@A/5>>?A  &r&r r7r:s@r'rr]sV ' 'r&rc,^\rSrSrSrU4SjrSrU=r$)rnzThis 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[:Xd7[S[U5RS[U5R35e[ TU]X5 gr+)r rrr-r!r.r/r1r2bnr4s r'r/ConvBn1d.__init__re ( .& 8,R0K? 5/5>>?u/3<<=?  "r&r r7r:s@r'rrnV # #r&rc,^\rSrSrSrU4SjrSrU=r$)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.c>[U5[:Xa[U5[:Xd7[S[U5RS[U5R35e[ TU]X5 gr+)r rrr-r!r.r/rLs r'r/ConvBn2d.__init__rOr&r r7r:s@r'rrrPr&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 2>[U5[:Xa&[U5[:Xa[U5[:XdM[ S[U5R S[U5R S[U5R 35e[ TU]XU5 gNr,z, z, and )r rrr r-r!r.r/r1r2rMr3r4s r'r/ConvBnReLU1d.__init__ ( .& 8,R0K?,T2d: 5/5>>?r/3<<=V/5>>?A  4(r&r r7r:s@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 2>[U5[:Xa&[U5[:Xa[U5[:XdM[ S[U5R S[U5R S[U5R 35e[ TU]XU5 grX)r rrr r-r!r.r/rYs r'r/ConvBnReLU2d.__init__r[r&r r7r:s@r'rrr\r&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[:Xd7[S[U5RS[U5R35e[ TU]X5 gr+)r rrr-r!r.r/rLs r'r/ConvBn3d.__init__rOr&r r7r:s@r'rrrPr&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 2>[U5[:Xa&[U5[:Xa[U5[:XdM[ S[U5R S[U5R S[U5R 35e[ TU]XU5 grX)r rrr r-r!r.r/rYs r'r/ConvBnReLU3d.__init__r[r&r r7r:s@r'rrr\r&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[:Xd7[S[U5RS[U5R35e[ TU]X5 gr+)r rr r-r!r.r/r1 batch_normr3r4s r'r/BNReLU2d.__init__f ( 4 C,T2d: 5/ ;DDEU/5>>?A  *r&r r7r:s@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[:Xd7[S[U5RS[U5R35e[ TU]X5 gr+)r rr r-r!r.r/rls r'r/BNReLU3d.__init__ror&r r7r:s@r'rrrpr&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[:Xd7[S[U5RS[U5R35e[ TU]X5 gr+)r r rr-r!r.r/)r1rGrMr4s r'r/LinearBn1d.__init__se ( 0F :,R0K? 5/7@@A/3<<=?  $r&r r7r:s@r'rrr;r&rc,^\rSrSrSrU4SjrSrU=r$)ri zThis is a sequential container which calls the Linear and LeakyReLU modules. During quantization this will be replaced with the corresponding fused module.c>[U5[La&[U5[RRLd7[ S[U5R S[U5R 35e[TU]!X5 gr+) typer torchnn LeakyReLUr-r!r.r/)r1rG leaky_relur4s r'r/LinearLeakyReLU.__init__slV &4 +;uxx?Q?Q+Q 5<(()tJ/?/H/H.IK  ,r&r r7r:s@r'rr sV--r&rc,^\rSrSrSrU4SjrSrU=r$)rizThis is a sequential container which calls the Linear and Tanh modules. During quantization this will be replaced with the corresponding fused module.c>[U5[La&[U5[RRLd7[ S[U5R S[U5R 35e[TU]!X5 gr+) r{r r|r}Tanhr-r!r.r/)r1rGtanhr4s r'r/LinearTanh.__init__shV &4:+F 5<(()tDz/B/B.CE  &r&r r7r:s@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)r1r2rr4s r'r/ConvAdd2d.__init__+s r&c8URUS"U5U5$)z4Applies convolution to x1 and adds the result to x2.rrr1x1x2s r'forwardConvAdd2d.forward/sxxQ R((r&r r!r"r#r$r8r/rr%r9r:s@r'rr'sV))r&rc2^\rSrSrSrU4SjrSrSrU=r$)ri4zThis 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/rr3)r1r2rr3r4s r'r/ConvAddReLU2d.__init__8s  r&cVURURUS"U5U55$)zCApplies convolution to x1, adds the result to x2, and applies ReLU.r)r3rrs r'rConvAddReLU2d.forward=s$yy$q'"+r233r&)rr3rr:s@r'rr4sV 44r&r)!r|torch.nnrrrrrrr r torch.nn.utils.parametrizer __all__r} Sequentialrr r rrrrrrrrrrrrrrrr r&r'rs    D , 588&& %%"%%"%%"''"#|#"#|#")<)&)<)&#|#")<)&+|+"+|+"%%" -l - ' ' ) ) 4L 4r&