U <ºcÝ ã@sXddlZddlZddlZddlmmmZGdd„dejƒZ Gdd„dej ƒZdS)éNcs\eZdZdZejjjZd‡fdd„ Z dd„Z d d „Ze‡fdd„ƒZ e‡fd d„ƒZ‡ZS)ÚBNReLU2dzÙ A BNReLU2d module is a fused module of BatchNorm2d and ReLU We adopt the same interface as :class:`torch.ao.nn.quantized.BatchNorm2d`. Attributes: Same as torch.ao.nn.quantized.BatchNorm2d çñhãˆµøä>çš™™™™™¹?Ncstt|ƒj|||||ddS©N)ÚepsÚmomentumÚdeviceÚdtype)ÚsuperrÚ__init__©ÚselfZnum_featuresrrrr ©Ú __class__©úP/tmp/pip-unpacked-wheel-gikjz4vx/torch/nn/intrinsic/quantized/modules/bn_relu.pyrszBNReLU2d.__init__c Cs@t|jƒdkrtdƒ‚tjj ||j|j|j |j |j|j|j ¡S)Néz#Input shape must be `(N, C, H, W)`!)ÚlenÚshapeÚ ValueErrorÚtorchÚopsÚ quantizedZbatch_norm2d_reluÚweightÚbiasÚrunning_meanÚrunning_varrÚscaleÚ zero_point©r ÚinputrrrÚforwardsþzBNReLU2d.forwardcCsdS)NZQuantizedBNReLU2dr©r rrrÚ _get_name szBNReLU2d._get_namecstt|ƒ |¡S©N)r rÚ from_float©ÚclsÚmodrrrr%#szBNReLU2d.from_floatcstƒ |d||¡S©Nr©r Úfrom_reference©r'Zbn_reluZoutput_scaleZoutput_zero_pointrrrr+(szBNReLU2d.from_reference)rrNN)Ú__name__Ú __module__Ú__qualname__Ú__doc__rÚnnÚ intrinsicrÚ _FLOAT_MODULErr!r#Úclassmethodr%r+Ú __classcell__rrrrrs rcs\eZdZdZejjjZd‡fdd„ Z dd„Z d d „Ze‡fdd„ƒZ e‡fd d„ƒZ‡ZS)ÚBNReLU3dzÙ A BNReLU3d module is a fused module of BatchNorm3d and ReLU We adopt the same interface as :class:`torch.ao.nn.quantized.BatchNorm3d`. Attributes: Same as torch.ao.nn.quantized.BatchNorm3d rrNcstt|ƒj|||||ddSr)r r6rrrrrr8szBNReLU3d.__init__c Cs@t|jƒdkrtdƒ‚tjj ||j|j|j |j |j|j|j ¡S)Néz&Input shape must be `(N, C, D, H, W)`!)rrrrrrZbatch_norm3d_relurrrrrrrrrrrr!;sþzBNReLU3d.forwardcCsdS)NZQuantizedBNReLU3drr"rrrr#DszBNReLU3d._get_namecstt|ƒ |¡Sr$)r r6r%r&rrrr%GszBNReLU3d.from_floatcstƒ |d||¡Sr)r*r,rrrr+LszBNReLU3d.from_reference)rrNN)r-r.r/r0rr1r2r6r3rr!r#r4r%r+r5rrrrr6,s r6)rZtorch.ao.nn.intrinsicZtorch.nn.intrinsic.qatZtorch.ao.nn.quantizedZaor1rZnnqZBatchNorm2drZBatchNorm3dr6rrrrÚs $