a d@sFddlmZGdddejZGdddejZGdddejZdS) )nncs*eZdZdZdfdd ZddZZS) 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 Ncst|r||_dSNsuper__init__qconfigselfr __class__d/var/www/html/stable-diffusion-webui/venv/lib/python3.9/site-packages/torch/ao/quantization/stubs.pyr s zQuantStub.__init__cCs|Srr r xr r rforwardszQuantStub.forward)N__name__ __module__ __qualname____doc__rr __classcell__r r r rrsrcs*eZdZdZdfdd ZddZZS) 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 Ncst|r||_dSrrr r r rrs zDeQuantStub.__init__cCs|Srr rr r rr"szDeQuantStub.forward)Nrr r r rrsrcsDeZdZUdZeed<eed<ejed<fddZ ddZ Z 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`. quantdequantmodulecsZtt|dr|jnd}|dt||dt||d|||jdS)Nrrrr) rrhasattrrZ add_modulerrZtrainZtraining)r rrr r rr5s   zQuantWrapper.__init__cCs||}||}||Sr)rrr)r Xr r rr=s  zQuantWrapper.forward) rrrrr__annotations__rrModulerrrr r r rr&s    rN)Ztorchrr rrrr r r rs