U yh=$@sddlZddlmZmZmZmZmZmZmZm Z ddl m Z dddddd d d d d dddddddgZ Gdddej jZGdddeZGdddeZGdddeZGdddeZGdddeZGdd d eZGdd d eZGdd d eZGdd d eZGd d d eZGd!ddeZGd"ddeZGd#ddeZGd$ddeZGd%ddeZGd&ddeZGd'ddeZ dS)(N)Conv1dConv2dConv3dReLULinear BatchNorm1d BatchNorm2d BatchNorm3d)type_before_parametrizations ConvReLU1d ConvReLU2d ConvReLU3d LinearReLUConvBn1dConvBn2d ConvBnReLU1d ConvBnReLU2dConvBn3d ConvBnReLU3dBNReLU2dBNReLU3d LinearBn1dLinearLeakyReLU LinearTanh ConvAdd2d ConvAddReLU2dc@s eZdZdS) _FusedModuleN)__name__ __module__ __qualname__r r U/var/www/html/venv/lib/python3.8/site-packages/torch/ao/nn/intrinsic/modules/fused.pyr srcs eZdZdZfddZZS)r zThis is a sequential container which calls the Conv1d and ReLU modules. During quantization this will be replaced with the corresponding fused module.csDt|tkrt|tks2tdt|t|t||dSNz!Incorrect types for input modules)r rrAssertionErrorsuper__init__selfconvrelu __class__r r!r%szConvReLU1d.__init__rrr__doc__r% __classcell__r r r*r!r scs eZdZdZfddZZS)r zThis is a sequential container which calls the Conv2d and ReLU modules. During quantization this will be replaced with the corresponding fused module.csDt|tkrt|tks2tdt|t|t||dSr")r rrr#r$r%r&r*r r!r%szConvReLU2d.__init__r,r r r*r!r scs eZdZdZfddZZS)r zThis is a sequential container which calls the Conv3d and ReLU modules. During quantization this will be replaced with the corresponding fused module.csDt|tkrt|tks2tdt|t|t||dSr")r rrr#r$r%r&r*r r!r%!szConvReLU3d.__init__r,r r r*r!r scs eZdZdZfddZZS)rzThis is a sequential container which calls the Linear and ReLU modules. During quantization this will be replaced with the corresponding fused module.csDt|tkrt|tks2tdt|t|t||dSr")r rrr#r$r%)r'linearr)r*r r!r%)szLinearReLU.__init__r,r r r*r!r&scs eZdZdZfddZZS)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.csDt|tkrt|tks2tdt|t|t||dSr")r rrr#r$r%r'r(bnr*r r!r%1szConvBn1d.__init__r,r r r*r!r.scs eZdZdZfddZZS)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.csDt|tkrt|tks2tdt|t|t||dSr")r rrr#r$r%r0r*r r!r%9szConvBn2d.__init__r,r r r*r!r6scs eZdZdZfddZZS)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.csZt|tkr$t|tkr$t|tksFtdt|t|t|t|||dSr")r rrrr#r$r%r'r(r1r)r*r r!r%As  zConvBnReLU1d.__init__r,r r r*r!r>scs eZdZdZfddZZS)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.csZt|tkr$t|tkr$t|tksFtdt|t|t|t|||dSr")r rrrr#r$r%r2r*r r!r%Is  zConvBnReLU2d.__init__r,r r r*r!rFscs eZdZdZfddZZS)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.csDt|tkrt|tks2tdt|t|t||dSr")r rr r#r$r%r0r*r r!r%QszConvBn3d.__init__r,r r r*r!rNscs eZdZdZfddZZS)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.csZt|tkr$t|tkr$t|tksFtdt|t|t|t|||dSr")r rr rr#r$r%r2r*r r!r%Ys  zConvBnReLU3d.__init__r,r r r*r!rVscs eZdZdZfddZZS)rzThis is a sequential container which calls the BatchNorm 2d and ReLU modules. During quantization this will be replaced with the corresponding fused module.csDt|tkrt|tks2tdt|t|t||dSr")r rrr#r$r%r'Z batch_normr)r*r r!r%bszBNReLU2d.__init__r,r r r*r!r_scs eZdZdZfddZZS)rzThis is a sequential container which calls the BatchNorm 3d and ReLU modules. During quantization this will be replaced with the corresponding fused module.csDt|tkrt|tks2tdt|t|t||dSr")r r rr#r$r%r3r*r r!r%jszBNReLU3d.__init__r,r r r*r!rgscs eZdZdZfddZZS)rzThis is a sequential container which calls the Linear and BatchNorm1d modules. During quantization this will be replaced with the corresponding fused module.csDt|tkrt|tks2tdt|t|t||dSr")r rrr#r$r%)r'r/r1r*r r!r%sszLinearBn1d.__init__r,r r r*r!rpscs eZdZdZfddZZS)rzThis is a sequential container which calls the Linear and LeakyReLU modules. During quantization this will be replaced with the corresponding fused module.csHt|tkrt|tjjks6tdt|t|t||dSr")typertorchnnZ LeakyReLUr#r$r%)r'r/Z leaky_relur*r r!r%{szLinearLeakyReLU.__init__r,r r r*r!rxscs eZdZdZfddZZS)rzThis is a sequential container which calls the Linear and Tanh modules. During quantization this will be replaced with the corresponding fused module.csHt|tkrt|tjjks6tdt|t|t||dSr")r4rr5r6ZTanhr#r$r%)r'r/tanhr*r r!r%szLinearTanh.__init__r,r r r*r!rscs(eZdZdZfddZddZZS)rzThis is a sequential container which calls the Conv2d modules with extra Add. During quantization this will be replaced with the corresponding fused module.cst|||_dSN)r$r%add)r'r(r9r*r r!r%s zConvAdd2d.__init__cCs||d||SNr)r9r'x1Zx2r r r!forwardszConvAdd2d.forwardrrrr-r%r=r.r r r*r!rs cs(eZdZdZfddZddZZS)rzThis is a sequential container which calls the Conv2d, add, Relu. During quantization this will be replaced with the corresponding fused module.cst|||_||_dSr8)r$r%r9r))r'r(r9r)r*r r!r%s zConvAddReLU2d.__init__cCs|||d||Sr:)r)r9r;r r r!r=szConvAddReLU2d.forwardr>r r r*r!rs )!r5Ztorch.nnrrrrrrrr Ztorch.nn.utils.parametrizer __all__r6Z Sequentialrr r r rrrrrrrrrrrrrrr r r r!sB(