EiSSKJr SSKrSSKJr SSKJr /SQr"SS\R5r"SS \R5r "S S \R5r g) )AnyN)nn)QConfig) QuantStub DeQuantStub QuantWrappercx^\rSrSrSrS S\S-4U4SjjjrS\RS\R4Sjr S r U=r $) r aQuantize 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 Nqconfigc>>[TU]5 U(aXlggNsuper__init__r selfr __class__s X/home/mande/repo/quber/.venv/lib/python3.13/site-packages/torch/ao/quantization/stubs.pyrQuantStub.__init__  "L xreturncU$r rrs rforwardQuantStub.forwardrr r ) __name__ __module__ __qualname____firstlineno____doc__rrtorchTensorr__static_attributes__ __classcell__rs@rrr s=#$## %,,rrcx^\rSrSrSrS S\S-4U4SjjjrS\RS\R4Sjr S r U=r $) 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 c>>[TU]5 U(aXlggr rrs rrDeQuantStub.__init__&rrrrcU$r rrs rrDeQuantStub.forward+rrr r ) r!r"r#r$r%rrr&r'rr(r)r*s@rrrs=#d ## %,,rrc^\rSrSr%Sr\\S'\\S'\R\S'S\R4U4Sjjr S\ RS\ R4S jr S rU=r$) r/aA 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`. quantdequantmodulec>[TU]5 [USS5nURS[ U55 URS[ U55 URSU5 UR UR5 g)Nr r3r4r5)rrgetattr add_modulerrtraintraining)rr5r rs rrQuantWrapper.__init__?s` &)T2 7!34  ;w#78 &) 6??#rXrchURU5nURU5nURU5$r )r3r5r4)rr<s rrQuantWrapper.forwardGs* JJqM KKN||Arr)r!r"r#r$r%r__annotations__rrModulerr&r'rr(r)r*s@rrr/sP    II$ryy$%,,rr) typingrr&rtorch.ao.quantizationr__all__r@rrrrrrrDsH ) 7 $"))$299r