o ôùÎhúã@srddlmZmZddlZddlmZddlmZgd¢ZGdd„dejƒZ Gdd „d ejƒZ Gd d „d ejƒZ dS) é)ÚAnyÚOptionalN)Únn)ÚQConfig)Ú QuantStubÚ DeQuantStubÚ QuantWrappercóBeZdZdZd deef‡fdd„ Zdejdejfdd „Z ‡Z S) raQuantize 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 NÚqconfigcótƒ ¡|r ||_dSdS©N©ÚsuperÚ__init__r ©Úselfr ©Ú __class__©úa/home/www/facesmatcher.com/frenv_anti/lib/python3.10/site-packages/torch/ao/quantization/stubs.pyró  ÿzQuantStub.__init__ÚxÚreturncCó|Sr r©rrrrrÚforwardózQuantStub.forwardr ) Ú__name__Ú __module__Ú __qualname__Ú__doc__rrrÚtorchÚTensorrÚ __classcell__rrrrr órcr ) raDequantize 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 Nr cr r r rrrrr&rzDeQuantStub.__init__rrcCrr rrrrrr+rzDeQuantStub.forwardr ) rrrr rrrr!r"rr#rrrrrr$rcsZeZdZUdZeed<eed<ejed<dejf‡fdd„ Z de j de j fd d „Z ‡Z S) raç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ÚmodulecsRtƒ ¡t|ddƒ}| dt|ƒ¡| dt|ƒ¡| d|¡| |j¡dS)Nr r%r&r')rrÚgetattrZ add_modulerrÚtrainZtraining)rr'r rrrr?s   zQuantWrapper.__init__ÚXrcCs| |¡}| |¡}| |¡Sr )r%r'r&)rr*rrrrGs   zQuantWrapper.forward)rrrr rÚ__annotations__rrÚModulerr!r"rr#rrrrr/s   r) Útypingrrr!rZtorch.ao.quantizationrÚ__all__r,rrrrrrrÚs