U ‰dÌã@s$ddlZddlmZmZmZmZmZmZmZm Z ddl m Z Gdd„dej j ƒZGdd„deƒZGdd „d eƒZGd d „d eƒZGd d „d eƒZGdd„deƒZGdd„deƒZGdd„deƒZGdd„deƒZGdd„deƒZGdd„deƒZGdd„deƒZGdd„deƒZGdd„deƒZdS) éN)ÚConv1dÚConv2dÚConv3dÚReLUÚLinearÚ BatchNorm1dÚ BatchNorm2dÚ BatchNorm3d)Útype_before_parametrizationsc@s eZdZdS)Ú _FusedModuleN)Ú__name__Ú __module__Ú __qualname__©rrúD/tmp/pip-unpacked-wheel-ua33x9lu/torch/nn/intrinsic/modules/fused.pyr sr cs eZdZdZ‡fdd„Z‡ZS)Ú ConvReLU1dzšThis is a sequential container which calls the Conv1d and ReLU modules. During quantization this will be replaced with the corresponding fused module.csBt|ƒtkrt|ƒtks0td t|ƒt|ƒ¡ƒ‚tƒ ||¡dS©Nz%Incorrect types for input modules{}{})r rrÚAssertionErrorÚformatÚsuperÚ__init__©ÚselfÚconvÚrelu©Ú __class__rrr sÿÿzConvReLU1d.__init__©r r rÚ__doc__rÚ __classcell__rrrrr srcs eZdZdZ‡fdd„Z‡ZS)Ú ConvReLU2dzšThis is a sequential container which calls the Conv2d and ReLU modules. During quantization this will be replaced with the corresponding fused module.csBt|ƒtkrt|ƒtks0td t|ƒt|ƒ¡ƒ‚tƒ ||¡dSr)r rrrrrrrrrrrsÿÿzConvReLU2d.__init__rrrrrr sr cs eZdZdZ‡fdd„Z‡ZS)Ú ConvReLU3dzšThis is a sequential container which calls the Conv3d and ReLU modules. During quantization this will be replaced with the corresponding fused module.csBt|ƒtkrt|ƒtks0td t|ƒt|ƒ¡ƒ‚tƒ ||¡dSr)r rrrrrrrrrrrsÿÿzConvReLU3d.__init__rrrrrr!sr!cs eZdZdZ‡fdd„Z‡ZS)Ú LinearReLUzšThis is a sequential container which calls the Linear and ReLU modules. During quantization this will be replaced with the corresponding fused module.csBt|ƒtkrt|ƒtks0td t|ƒt|ƒ¡ƒ‚tƒ ||¡dSr)r rrrrrr)rÚlinearrrrrr'sÿÿzLinearReLU.__init__rrrrrr"$sr"cs eZdZdZ‡fdd„Z‡ZS)ÚConvBn1dz¤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.csBt|ƒtkrt|ƒtks0td t|ƒt|ƒ¡ƒ‚tƒ ||¡dSr)r rrrrrr©rrÚbnrrrr0sÿÿzConvBn1d.__init__rrrrrr$-sr$cs eZdZdZ‡fdd„Z‡ZS)ÚConvBn2dz¤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.csFt|ƒtkrt|ƒtks0td t|ƒt|ƒ¡ƒ‚tt|ƒ ||¡dSr)r rrrrrr'rr%rrrr9sÿÿzConvBn2d.__init__rrrrrr'6sr'cs eZdZdZ‡fdd„Z‡ZS)Ú ConvBnReLU1dz«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.csVt|ƒtkr$t|ƒtkr$t|ƒtksBtd t|ƒt|ƒt|ƒ¡ƒ‚tƒ |||¡dS©Nz'Incorrect types for input modules{}{}{})r rrrrrrr©rrr&rrrrrBs ÿÿÿzConvBnReLU1d.__init__rrrrrr(?sr(cs eZdZdZ‡fdd„Z‡ZS)Ú ConvBnReLU2dz«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.csVt|ƒtkr$t|ƒtkr$t|ƒtksBtd t|ƒt|ƒt|ƒ¡ƒ‚tƒ |||¡dSr))r rrrrrrrr*rrrrKs ÿÿÿzConvBnReLU2d.__init__rrrrrr+Hsr+cs eZdZdZ‡fdd„Z‡ZS)ÚConvBn3dz¤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.csBt|ƒtkrt|ƒtks0td t|ƒt|ƒ¡ƒ‚tƒ ||¡dSr)r rr rrrrr%rrrrTsÿÿzConvBn3d.__init__rrrrrr,Qsr,cs eZdZdZ‡fdd„Z‡ZS)Ú ConvBnReLU3dz«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.csVt|ƒtkr$t|ƒtkr$t|ƒtksBtd t|ƒt|ƒt|ƒ¡ƒ‚tƒ |||¡dSr))r rr rrrrrr*rrrr]s ÿÿÿzConvBnReLU3d.__init__rrrrrr-Zsr-cs eZdZdZ‡fdd„Z‡ZS)ÚBNReLU2dz This is a sequential container which calls the BatchNorm 2d and ReLU modules. During quantization this will be replaced with the corresponding fused module.csBt|ƒtkrt|ƒtks0td t|ƒt|ƒ¡ƒ‚tƒ ||¡dSr)r rrrrrr©rZ batch_normrrrrrgsÿÿzBNReLU2d.__init__rrrrrr.dsr.cs eZdZdZ‡fdd„Z‡ZS)ÚBNReLU3dz This is a sequential container which calls the BatchNorm 3d and ReLU modules. During quantization this will be replaced with the corresponding fused module.csBt|ƒtkrt|ƒtks0td t|ƒt|ƒ¡ƒ‚tƒ ||¡dSr)r r rrrrrr/rrrrpsÿÿzBNReLU3d.__init__rrrrrr0msr0cs eZdZdZ‡fdd„Z‡ZS)Ú LinearBn1dz¡This is a sequential container which calls the Linear and BatchNorm1d modules. During quantization this will be replaced with the corresponding fused module.csBt|ƒtkrt|ƒtks0td t|ƒt|ƒ¡ƒ‚tƒ ||¡dSr)r rrrrrr)rr#r&rrrrzsÿzLinearBn1d.__init__rrrrrr1wsr1)ZtorchZtorch.nnrrrrrrrr Ztorch.nn.utils.parametrizer ÚnnZ Sequentialr rr r!r"r$r'r(r+r,r-r.r0r1rrrrÚs (