U <ºc ã@sDddlZddlZddlmZGdd„dejƒZGdd„dejƒZdS)éNcs$eZdZ‡fdd„Zdd„Z‡ZS)ÚLinearReluFunctionalChildcsNtƒ ¡t t ||¡¡|_t t |¡¡|_tjj j |jt   d¡ddS©Né)Úa) ÚsuperÚ__init__ÚnnÚ ParameterÚtorchÚemptyÚw1ÚzerosÚb1ÚinitÚkaiming_uniform_ÚmathÚsqrt©ÚselfÚN©Ú __class__©ú]/tmp/pip-unpacked-wheel-gikjz4vx/torch/testing/_internal/quantization_torch_package_models.pyrs z"LinearReluFunctionalChild.__init__cCs(tjj ||j|j¡}tjj |¡}|S©N)r rÚ functionalÚlinearr rÚrelu©rÚxrrrÚforwardsz!LinearReluFunctionalChild.forward©Ú__name__Ú __module__Ú __qualname__rr Ú __classcell__rrrrrs rcs$eZdZ‡fdd„Zdd„Z‡ZS)ÚLinearReluFunctionalcsXtƒ ¡t|ƒ|_t t ||¡¡|_t t  |¡¡|_ tjj j |jt  d¡ddSr)rrrÚchildrr r r r r rrrrrrrrrrs   zLinearReluFunctional.__init__cCs2| |¡}tjj ||j|j¡}tjj |¡}|Sr)r'r rrrr rrrrrrr s zLinearReluFunctional.forwardr!rrrrr&s r&)rr Ztorch.nnrÚModulerr&rrrrÚs