sg &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__V/var/www/html/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(sJdt|t|t|||yNz!Incorrect types for input modules)r rr super__init__selfconvrelu __class__s r$r)zConvReLU1d.__init__.^ ( .& 8,T2d: x//KD/Q.RSoptSuRv w x ; 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 rr r(r)r*s r$r)zConvReLU2d.__init__:r/r#r0r3s@r$r r 6r4r#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 rr r(r)r*s r$r)zConvReLU3d.__init__Fr/r#r0r3s@r$rrBr4r#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(sJdt|t|t|||yr')r r r r(r))r+linearr-r.s r$r)zLinearReLU.__init__Rs^ ( 0F :,T2d: z//KF/S.TUqrvUwTx y z ; &r#r0r3s@r$rrNsV''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__^_ ( .& 8,R0K? v//KD/Q.RSoprSsRt u v @ r"r#r0r3s@r$rrZV##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 rrr(r)r>s r$r)zConvBn2d.__init__jr@r#r0r3s@r$rrfrAr#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(s)Jdt|t|t|t||||yr')r rrr r(r)r+r,r?r-r.s r$r)zConvBnReLU1d.__init__v ( .& 8,R0K?,T2d: Z//KD/Q.RSoprSsRtvRSWvXuY Z  Z ; r4(r#r0r3s@r$rrrV))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(s)Jdt|t|t|t||||yr')r rrr r(r)rFs r$r)zConvBnReLU2d.__init__rGr#r0r3s@r$rrrHr#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 rrr(r)r>s r$r)zConvBn3d.__init__r@r#r0r3s@r$rrrAr#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(s)Jdt|t|t|t||||yr')r rrr r(r)rFs r$r)zConvBnReLU3d.__init__rGr#r0r3s@r$rrrHr#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(sJdt|t|t|||yr')r rr r(r)r+ batch_normr-r.s r$r)zBNReLU2d.__init__^ ( 4 C,T2d: ~//KJ/W.XYuvzY{X| } ~ ; T*r#r0r3s@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(sJdt|t|t|||yr')r rr r(r)rQs r$r)zBNReLU3d.__init__rSr#r0r3s@r$rrrTr#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 r rr(r))r+r;r?r.s r$r)zLinearBn1d.__init__s_ ( 0F :,R0K? x//KF/S.TUqrtUuTv w x @ $r#r0r3s@r$rrr4r#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')typer torchnn LeakyReLUr(r))r+r; leaky_relur.s r$r)zLinearLeakyReLU.__init__s[ LF "tJ'7588;M;M'M P .tF|nT*=Mrvs    D , 588&&  % % % % % % ' ' #| # #| # )< ) )< ) #| # )< ) +| + +| + % %-l-'' ) ) 4L 4r#