U KºcŸã @sBddlZddlmZmZmZmZmZmZmZm Z ddl m Z dddddd d d d d dddg Z Gdd„dej jƒZGdd„deƒZGdd„deƒZGdd„deƒZGdd„deƒZGdd„deƒZGdd „d eƒZGdd „d eƒZGdd „d eƒZGdd „d eƒZGdd „d eƒZGdd„deƒZGdd„deƒZGdd„deƒZdS) éN)ÚConv1dÚConv2dÚConv3dÚReLUÚLinearÚ BatchNorm1dÚ BatchNorm2dÚ BatchNorm3d)Útype_before_parametrizationsÚ ConvReLU1dÚ ConvReLU2dÚ ConvReLU3dÚ LinearReLUÚConvBn1dÚConvBn2dÚ ConvBnReLU1dÚ ConvBnReLU2dÚConvBn3dÚ ConvBnReLU3dÚBNReLU2dÚBNReLU3dÚ LinearBn1dc@s eZdZdS)Ú _FusedModuleN)Ú__name__Ú __module__Ú __qualname__©rrúG/tmp/pip-unpacked-wheel-gikjz4vx/torch/ao/nn/intrinsic/modules/fused.pyr srcs 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.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__©rrrÚ__doc__r"Ú __classcell__rrr'rr scs eZdZdZ‡fdd„Z‡ZS)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.csBt|ƒtkrt|ƒtks0td t|ƒt|ƒ¡ƒ‚tƒ ||¡dSr)r rrrr r!r"r#r'rrr"sÿÿzConvReLU2d.__init__r)rrr'rr scs eZdZdZ‡fdd„Z‡ZS)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.csBt|ƒtkrt|ƒtks0td t|ƒt|ƒ¡ƒ‚tƒ ||¡dSr)r rrrr r!r"r#r'rrr"!sÿÿzConvReLU3d.__init__r)rrr'rr scs eZdZdZ‡fdd„Z‡ZS)rzš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 rrrr r!r")r$Úlinearr&r'rrr"*sÿÿzLinearReLU.__init__r)rrr'rr'scs eZdZdZ‡fdd„Z‡ZS)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.csBt|ƒtkrt|ƒtks0td t|ƒt|ƒ¡ƒ‚tƒ ||¡dSr)r rrrr r!r"©r$r%Úbnr'rrr"3sÿÿzConvBn1d.__init__r)rrr'rr0scs eZdZdZ‡fdd„Z‡ZS)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.csFt|ƒtkrt|ƒtks0td t|ƒt|ƒ¡ƒ‚tt|ƒ ||¡dSr)r rrrr r!rr"r-r'rrr"<sÿÿzConvBn2d.__init__r)rrr'rr9scs eZdZdZ‡fdd„Z‡ZS)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.csVt|ƒtkr$t|ƒtkr$t|ƒtksBtd t|ƒt|ƒt|ƒ¡ƒ‚tƒ |||¡dS©Nz'Incorrect types for input modules{}{}{})r rrrrr r!r"©r$r%r.r&r'rrr"Es ÿÿÿzConvBnReLU1d.__init__r)rrr'rrBscs eZdZdZ‡fdd„Z‡ZS)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.csVt|ƒtkr$t|ƒtkr$t|ƒtksBtd t|ƒt|ƒt|ƒ¡ƒ‚tƒ |||¡dSr/)r rrrrr r!r"r0r'rrr"Ns ÿÿÿzConvBnReLU2d.__init__r)rrr'rrKscs eZdZdZ‡fdd„Z‡ZS)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.csBt|ƒtkrt|ƒtks0td t|ƒt|ƒ¡ƒ‚tƒ ||¡dSr)r rr rr r!r"r-r'rrr"WsÿÿzConvBn3d.__init__r)rrr'rrTscs eZdZdZ‡fdd„Z‡ZS)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.csVt|ƒtkr$t|ƒtkr$t|ƒtksBtd t|ƒt|ƒt|ƒ¡ƒ‚tƒ |||¡dSr/)r rr rrr r!r"r0r'rrr"`s ÿÿÿzConvBnReLU3d.__init__r)rrr'rr]scs eZdZdZ‡fdd„Z‡ZS)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.csBt|ƒtkrt|ƒtks0td t|ƒt|ƒ¡ƒ‚tƒ ||¡dSr)r rrrr r!r"©r$Z batch_normr&r'rrr"jsÿÿzBNReLU2d.__init__r)rrr'rrgscs eZdZdZ‡fdd„Z‡ZS)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.csBt|ƒtkrt|ƒtks0td t|ƒt|ƒ¡ƒ‚tƒ ||¡dSr)r r rrr r!r"r1r'rrr"ssÿÿzBNReLU3d.__init__r)rrr'rrpscs eZdZdZ‡fdd„Z‡ZS)rz¡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 rrrr r!r")r$r,r.r'rrr"}sÿzLinearBn1d.__init__r)rrr'rrzs)ZtorchZtorch.nnrrrrrrrr Ztorch.nn.utils.parametrizer Ú__all__ÚnnZ Sequentialrr r r rrrrrrrrrrrrrrÚs2(  þ