a ‘º”húã@srddlmZmZddlZddlmZddlmZgd¢ZGdd„dejƒZ Gdd „d ejƒZ Gd d „d ejƒZ dS) é)ÚAnyÚOptionalN)Únn)ÚQConfig)Ú QuantStubÚ DeQuantStubÚ QuantWrappercs@eZdZdZd eedœ‡fdd„ Zejejdœdd„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©Úqconfigcstƒ ¡|r||_dS©N©ÚsuperÚ__init__r ©Úselfr ©Ú __class__©úI/var/www/auris/lib/python3.9/site-packages/torch/ao/quantization/stubs.pyrs zQuantStub.__init__©ÚxÚreturncCs|Sr r©rrrrrÚforwardszQuantStub.forward)N) Ú__name__Ú __module__Ú __qualname__Ú__doc__rrrÚtorchÚTensorrÚ __classcell__rrrrr srcs@eZdZdZd eedœ‡fdd„ Zejejdœdd„Z ‡Z S) 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 cstƒ ¡|r||_dSr r rrrrr&s zDeQuantStub.__init__rcCs|Sr rrrrrr+szDeQuantStub.forward)N) rrrrrrrrrrr rrrrrsrcsXeZdZUdZeed<eed<ejed<ejdœ‡fdd„ Z e j e j dœd 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Úmodule)r#csRtƒ ¡t|ddƒ}| dt|ƒ¡| dt|ƒ¡| d|¡| |j¡dS)Nr r!r"r#)r rÚgetattrZ add_modulerrZtrainZtraining)rr#r rrrr?s    zQuantWrapper.__init__)ÚXrcCs| |¡}| |¡}| |¡Sr )r!r#r")rr%rrrrGs  zQuantWrapper.forward)rrrrrÚ__annotations__rrÚModulerrrrr rrrrr/s   r) ÚtypingrrrrZtorch.ao.quantizationrÚ__all__r'rrrrrrrÚs