j<ddlmZGddejZGddejZGddejZdS) )nnc*eZdZdZdfd ZdZxZS) QuantStubaQuantize stub module, before calibration, this is same as an observer, it will be swapped as `nnq.Quantize` in `convert`. Args: qconfig: quantization configuration for the tensor, if qconfig is not provided, we will get qconfig from parent modules Nc^t|r ||_dSdSNsuper__init__qconfigselfr __class__s LC:\Projects\Historical\venv\Lib\site-packages\torch/ao/quantization/stubs.pyr zQuantStub.__init__6   #"DLLL # #c|Srr xs rforwardzQuantStub.forwardrr__name__ __module__ __qualname____doc__r r __classcell__rs@rrrV###### rrc*eZdZdZdfd ZdZxZS) DeQuantStubaDequantize stub module, before calibration, this is same as identity, this will be swapped as `nnq.DeQuantize` in `convert`. Args: qconfig: quantization configuration for the tensor, if qconfig is not provided, we will get qconfig from parent modules Nc^t|r ||_dSdSrrr s rr zDeQuantStub.__init__!rrc|Srrrs rrzDeQuantStub.forward&rrrrrs@rr!r!rrr!cReZdZUdZeed<eed<ejed<fdZ dZ xZ S) QuantWrapperaA wrapper class that wraps the input module, adds QuantStub and DeQuantStub and surround the call to module with call to quant and dequant modules. This is used by the `quantization` utility functions to add the quant and dequant modules, before `convert` function `QuantStub` will just be observer, it observes the input tensor, after `convert`, `QuantStub` will be swapped to `nnq.Quantize` which does actual quantization. Similarly for `DeQuantStub`. quantdequantmodulecVtt|dd}|dt ||dt ||d|||jdS)Nr r&r'r()r r getattr add_modulerr!traintraining)r r(r rs rr zQuantWrapper.__init__9s &)T22 7!3!3444  ;w#7#7888 &))) 6?#####rc||}||}||Sr)r&r(r')r Xs rrzQuantWrapper.forwardAs0 JJqMM KKNN||Ar) rrrrr__annotations__r!rModuler rrrs@rr%r%*s{     I$$$$$rr%N)torchrr1rr!r%rrrr3s $")$29r