a OSic"ã@sFddlmZGdd„dejƒZGdd„dejƒZGdd„dejƒZdS) é)Únncs*eZdZdZd‡fdd„ Zdd„Z‡ZS)Ú 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 Ncstt|ƒ ¡|r||_dS©N)ÚsuperrÚ__init__Úqconfig©Úselfr©Ú __class__©úW/var/www/html/django/DPS/env/lib/python3.9/site-packages/torch/ao/quantization/stubs.pyr szQuantStub.__init__cCs|Srr ©r Úxr r r ÚforwardszQuantStub.forward)N©Ú__name__Ú __module__Ú __qualname__Ú__doc__rrÚ __classcell__r r r r rsrcs*eZdZdZd‡fdd„ Zdd„Z‡ZS)Ú 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 Ncstt|ƒ ¡|r||_dSr)rrrrrr r r rszDeQuantStub.__init__cCs|Srr rr r r r"szDeQuantStub.forward)Nrr r r r rsrcsDeZdZUdZeed<eed<ejed<‡fdd„Z dd„Z ‡Z S) Ú QuantWrapperaçA 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`. ÚquantÚdequantÚmodulecs^tt|ƒ ¡t|dƒr|jnd}| dt|ƒ¡| dt|ƒ¡| d|¡| |j ¡dS)Nrrrr) rrrÚhasattrrÚ add_modulerrÚtrainÚtraining)r rrr r r r5s  zQuantWrapper.__init__cCs| |¡}| |¡}| |¡Sr)rrr)r ÚXr r r r=s  zQuantWrapper.forward) rrrrrÚ__annotations__rrÚModulerrrr r r r r&s    rN)Útorchrr"rrrr r r r Ús