o ‡ZŽh?&ã@slddlZddlmZmZmZmZmZmZmZm Z ddl m Z gd¢Z Gdd„dej jƒZGdd„deƒZGd d „d eƒZGd d „d eƒZGd d„deƒZGdd„deƒZGdd„deƒZGdd„deƒZGdd„deƒZGdd„deƒZGdd„deƒZGdd„deƒZGdd„deƒZGdd „d eƒZGd!d"„d"eƒZGd#d$„d$eƒZGd%d&„d&eƒZGd'd(„d(eƒZ dS))éN)Ú BatchNorm1dÚ BatchNorm2dÚ BatchNorm3dÚConv1dÚConv2dÚConv3dÚLinearÚReLU)Útype_before_parametrizations)Ú ConvReLU1dÚ ConvReLU2dÚ ConvReLU3dÚ LinearReLUÚConvBn1dÚConvBn2dÚ ConvBnReLU1dÚ ConvBnReLU2dÚConvBn3dÚ ConvBnReLU3dÚBNReLU2dÚBNReLU3dÚ LinearBn1dÚLinearLeakyReLUÚ LinearTanhÚ ConvAdd2dÚ ConvAddReLU2dc@s eZdZdS)Ú _FusedModuleN)Ú__name__Ú __module__Ú __qualname__©r r úR/var/www/auris/lib/python3.10/site-packages/torch/ao/nn/intrinsic/modules/fused.pyr&srcó eZdZdZ‡fdd„Z‡ZS)r zšThis is a sequential container which calls the Conv1d and ReLU modules. During quantization this will be replaced with the corresponding fused module.cóDt|ƒtkr t|ƒtksJdt|ƒ›t|ƒ›ƒ‚tƒ ||¡dS©Nz!Incorrect types for input modules)r rr ÚsuperÚ__init__©ÚselfÚconvÚrelu©Ú __class__r r!r&.ó ÿzConvReLU1d.__init__©rrrÚ__doc__r&Ú __classcell__r r r+r!r *ór cr")r zšThis is a sequential container which calls the Conv2d and ReLU modules. During quantization this will be replaced with the corresponding fused module.cr#r$)r rr r%r&r'r+r r!r&:r-zConvReLU2d.__init__r.r r r+r!r 6r1r cr")r zšThis is a sequential container which calls the Conv3d and ReLU modules. During quantization this will be replaced with the corresponding fused module.cr#r$)r rr r%r&r'r+r r!r&Fr-zConvReLU3d.__init__r.r r r+r!r Br1r cr")rzšThis is a sequential container which calls the Linear and ReLU modules. During quantization this will be replaced with the corresponding fused module.cr#r$)r rr r%r&)r(Úlinearr*r+r r!r&Rr-zLinearReLU.__init__r.r r r+r!rNr1rcr")rz¤This 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.cr#r$)r rrr%r&©r(r)Úbnr+r r!r&^r-zConvBn1d.__init__r.r r r+r!rZr1rcr")rz¤This 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.cr#r$)r rrr%r&r3r+r r!r&jr-zConvBn2d.__init__r.r r r+r!rfr1rcr")rz«This 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óZt|ƒtkrt|ƒtkrt|ƒtks#Jdt|ƒ›t|ƒ›t|ƒ›ƒ‚tƒ |||¡dSr$)r rrr r%r&©r(r)r4r*r+r r!r&vó  ÿzConvBnReLU1d.__init__r.r r r+r!rrr1rcr")rz«This 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.cr5r$)r rrr r%r&r6r+r r!r&ƒr7zConvBnReLU2d.__init__r.r r r+r!rr1rcr")rz¤This 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.cr#r$)r rrr%r&r3r+r r!r&r-zConvBn3d.__init__r.r r r+r!rŒr1rcr")rz«This 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.cr5r$)r rrr r%r&r6r+r r!r&œr7zConvBnReLU3d.__init__r.r r r+r!r˜r1rcr")rz This is a sequential container which calls the BatchNorm 2d and ReLU modules. During quantization this will be replaced with the corresponding fused module.cr#r$)r rr r%r&©r(Z batch_normr*r+r r!r&©r-zBNReLU2d.__init__r.r r r+r!r¥r1rcr")rz This is a sequential container which calls the BatchNorm 3d and ReLU modules. During quantization this will be replaced with the corresponding fused module.cr#r$)r rr r%r&r8r+r r!r&µr-zBNReLU3d.__init__r.r r r+r!r±r1rcr")rz¡This is a sequential container which calls the Linear and BatchNorm1d modules. During quantization this will be replaced with the corresponding fused module.cr#r$)r rrr%r&)r(r2r4r+r r!r&Ár-zLinearBn1d.__init__r.r r r+r!r½r1rcr")rzŸThis is a sequential container which calls the Linear and LeakyReLU modules. During quantization this will be replaced with the corresponding fused module.cóHt|ƒtkrt|ƒtjjksJdt|ƒ›t|ƒ›ƒ‚tƒ ||¡dSr$)ÚtyperÚtorchÚnnZ LeakyReLUr%r&)r(r2Z leaky_relur+r r!r&ÍóÿzLinearLeakyReLU.__init__r.r r r+r!rÉr1rcr")rzšThis is a sequential container which calls the Linear and Tanh modules. During quantization this will be replaced with the corresponding fused module.cr9r$)r:rr;r<ZTanhr%r&)r(r2Útanhr+r r!r&Ør=zLinearTanh.__init__r.r r r+r!rÔr1rcó(eZdZdZ‡fdd„Zdd„Z‡ZS)rz This is a sequential container which calls the Conv2d modules with extra Add. During quantization this will be replaced with the corresponding fused module.cstƒ |¡||_dS©N)r%r&Úadd)r(r)rAr+r r!r&ãs  zConvAdd2d.__init__cCs| |d|ƒ|¡S©Nr)rA©r(Úx1Zx2r r r!ÚforwardçszConvAdd2d.forward©rrrr/r&rEr0r r r+r!rßs rcr?)rz”This is a sequential container which calls the Conv2d, add, Relu. During quantization this will be replaced with the corresponding fused module.cstƒ |¡||_||_dSr@)r%r&rAr*)r(r)rAr*r+r r!r&ïs  zConvAddReLU2d.__init__cCs| | |d|ƒ|¡¡SrB)r*rArCr r r!rEôszConvAddReLU2d.forwardrFr r r+r!rës r)!r;Ztorch.nnrrrrrrrr Ztorch.nn.utils.parametrizer Ú__all__r<Z Sequentialrr r r rrrrrrrrrrrrrrr r r r!Ús,(