U <ºcþã@sNddlZddlmmmZddlmmmZGdd„dej ƒZ dS)éNcsjeZdZdZejZdejf‡fdd„ Z ej ej dœdd„Z dd „Z e ‡fd d „ƒZe ‡fd d „ƒZ‡ZS)Ú LinearReLUa® A LinearReLU module fused from Linear and ReLU modules We adopt the same interface as :class:`torch.ao.nn.quantized.Linear`. Attributes: Same as torch.ao.nn.quantized.Linear Examples:: >>> # xdoctest: +SKIP >>> m = nn.intrinsic.LinearReLU(20, 30) >>> input = torch.randn(128, 20) >>> output = m(input) >>> print(output.size()) torch.Size([128, 30]) Tcstƒ ||||¡dS©N)ÚsuperÚ__init__)ÚselfZ in_featuresZ out_featuresZbiasZdtype©Ú __class__©úT/tmp/pip-unpacked-wheel-gikjz4vx/torch/nn/intrinsic/quantized/modules/linear_relu.pyrszLinearReLU.__init__)ÚxÚreturncCstjj ||jj|j|j¡Sr)ÚtorchÚopsÚ quantizedZ linear_reluZ_packed_paramsZscaleZ zero_point)rr r r r Úforwards ÿzLinearReLU.forwardcCsdS)NZQuantizedLinearReLUr )rr r r Ú _get_name szLinearReLU._get_namecstt|ƒ |¡Sr)rrÚ from_float)ÚclsÚmodrr r r#szLinearReLU.from_floatcstƒ |d||¡S)Nr)rÚfrom_reference)rZref_linear_reluZ output_scaleZoutput_zero_pointrr r r'szLinearReLU.from_reference)Ú__name__Ú __module__Ú __qualname__Ú__doc__ÚnnirZ _FLOAT_MODULEr Zqint8rZTensorrrÚ classmethodrrÚ __classcell__r r rr rsr) r Ztorch.ao.nn.quantizedZaoÚnnrZnnqZtorch.ao.nn.intrinsicZ intrinsicrZLinearrr r r r Ús