%VjiM(VddlZddlmZmZmZmZmZmZmZm Z ddl m Z gdZ Gddej jZGddeZGd d eZGd d eZGd deZGddeZGddeZGddeZGddeZGddeZGddeZGddeZGddeZGdd eZGd!d"eZGd#d$eZGd%d&eZGd'd(eZ dS))N) BatchNorm1d BatchNorm2d BatchNorm3dConv1dConv2dConv3dLinearReLU)type_before_parametrizations) ConvReLU1d ConvReLU2d ConvReLU3d LinearReLUConvBn1dConvBn2d ConvBnReLU1d ConvBnReLU2dConvBn3d ConvBnReLU3dBNReLU2dBNReLU3d LinearBn1dLinearLeakyReLU LinearTanh ConvAdd2d ConvAddReLU2dceZdZdS) _FusedModuleN)__name__ __module__ __qualname__]/root/voice-cloning/.venv/lib/python3.11/site-packages/torch/ao/nn/intrinsic/modules/fused.pyrr&sDr#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|tkrt|tks)Jdt|t|t||dSNz!Incorrect types for input modules)r rr super__init__selfconvrelu __class__s r$r)zConvReLU1d.__init__. ( . .& 8 8,T22d::: 40LT0R0R 4+D11 4 4;: ; t$$$$$r#rr r!__doc__r) __classcell__r.s@r$r r *DVV%%%%%%%%%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|tkrt|tks)Jdt|t|t||dSr')r rr r(r)r*s r$r)zConvReLU2d.__init__=r/r#r0r3s@r$r r 9r4r#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|tkrt|tks)Jdt|t|t||dSr')r rr r(r)r*s r$r)zConvReLU3d.__init__Lr/r#r0r3s@r$rrHr4r#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|tkrt|tks)Jdt|t|t||dSr')r r r r(r))r+linearr-r.s r$r)zLinearReLU.__init__[s ( 0 0F : :,T22d::: 40LV0T0T 4+D11 4 4;: ; &&&&&r#r0r3s@r$rrWsDVV'''''''''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|tkrt|tks)Jdt|t|t||dSr')r rrr(r)r+r,bnr.s r$r)zConvBn1d.__init__j ( . .& 8 8,R00K??? 20LT0R0R 2+B// 2 2@? @ r"""""r#r0r3s@r$rrfDVV#########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|tkrt|tks)Jdt|t|t||dSr')r rrr(r)r>s r$r)zConvBn2d.__init__yr@r#r0r3s@r$rrurAr#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.cNt|tkr0t|tkrt|tks8Jdt|t|t|t |||dSr')r rrr r(r)r+r,r?r-r.s r$r)zConvBnReLU1d.__init__ ( . .& 8 8,R00K??,T22d::: 40LT0R0R 4+B// 4+D11 4 4;: ; r4(((((r#r0r3s@r$rrDVV ) ) ) ) ) ) ) ) )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.cNt|tkr0t|tkrt|tks8Jdt|t|t|t |||dSr')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|tkrt|tks)Jdt|t|t||dSr')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.cNt|tkr0t|tkrt|tks8Jdt|t|t|t |||dSr')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|tkrt|tks)Jdt|t|t||dSr')r rr r(r)r+ batch_normr-r.s r$r)zBNReLU2d.__init__ ( 4 4 C C,T22d::: 40LZ0X0X 4+D11 4 4;: ; T*****r#r0r3s@r$rrDVV+++++++++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|tkrt|tks)Jdt|t|t||dSr')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|tkrt|tks)Jdt|t|t||dSr')r r rr(r))r+r;r?r.s r$r)zLinearBn1d.__init__s ( 0 0F : :,R00K??? 20LV0T0T 2+B// 2 2@? @ $$$$$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|tkr"t|tjjks)Jdt|t|t ||dSr')typer torchnn LeakyReLUr(r))r+r; leaky_relur.s r$r)zLinearLeakyReLU.__init__szF||v%%$z*:*:eh>P*P*P*P PV Pd:>N>N P P+Q*PP ,,,,,r#r0r3s@r$rrsDVV---------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|tkr"t|tjjks)Jdt|t|t ||dSr')r[r r\r]Tanhr(r))r+r;tanhr.s r$r)zLinearTanh.__init__suF||v%%$t** *E*E*E JV Jd4jj J J+F*EE &&&&&r#r0r3s@r$rrsDVV'''''''''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.cXt|||_dSN)r(r)add)r+r,rgr.s r$r)zConvAdd2d.__init__ s& r#cL||d||SNr)rgr+x1x2s r$forwardzConvAdd2d.forwards"xxQ R(((r#rr r!r1r)rmr2r3s@r$rr sSVV)))))))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.cft|||_||_dSrf)r(r)rgr-)r+r,rgr-r.s r$r)zConvAddReLU2d.__init__s-  r#cr|||d||Sri)r-rgrjs r$rmzConvAddReLU2d.forwards.yy'$q'"++r22333r#rnr3s@r$rrsSVV 4444444r#r)!r\torch.nnrrrrrrr r torch.nn.utils.parametrizer __all__r] Sequentialrr r rrrrrrrrrrrrrrrr"r#r$rvs                     DCCCCC   ,     58&    % % % % % % % % % % % % % % % % % % % % % % % % ' ' ' ' ' ' ' ' # # # # #| # # # # # # # #| # # #)))))<)))")))))<)))" # # # # #| # # #)))))<)))" + + + + +| + + + + + + + +| + + + % % % % % % % %-----l---'''''''' ) ) ) ) ) ) ) ) 4 4 4 4 4L 4 4 4 4 4r#