ёiSrSSKJr SSKJr SSKJr \(aSSKJr SSK J r SSK J r "S S \5r "S S \5rg )z!Define stub used in quantization.) annotations) TYPE_CHECKING)Layer)Tensor) QuantConfig)QuanterFactoryc>^\rSrSrSrSSU4SjjjrSSjrSrU=r$) Stuba The stub is used as placeholders that will be replaced by observers before PTQ or QAT. It is hard to assign a quantization configuration to a functional API called in the forward of a layer. Instead, we can create a stub and add it to the sublayers of the layer. And call the stub before the functional API in the forward. The observer held by the stub will observe or quantize the inputs of the functional API. Args: observer(QuanterFactory): The configured information of the observer to be inserted. It will use a global configuration to create the observers if the 'observer' is none. Examples: .. code-block:: python >>> import paddle >>> from paddle.nn.quant import Stub >>> from paddle.quantization.quanters import FakeQuanterWithAbsMaxObserver >>> from paddle.nn import Conv2D >>> from paddle.quantization import QAT, QuantConfig >>> quanter = FakeQuanterWithAbsMaxObserver(moving_rate=0.9) >>> class Model(paddle.nn.Layer): ... def __init__(self, num_classes=10): ... super().__init__() ... self.conv = Conv2D(3, 6, 3, stride=1, padding=1) ... self.quant = Stub(quanter) ... ... def forward(self, inputs): ... out = self.conv(inputs) ... out = self.quant(out) ... return paddle.nn.functional.relu(out) >>> model = Model() >>> q_config = QuantConfig(activation=quanter, weight=quanter) >>> qat = QAT(q_config) >>> quant_model = qat.quantize(model) >>> print(quant_model) Model( (conv): QuantedConv2D( (weight_quanter): FakeQuanterWithAbsMaxObserverLayer() (activation_quanter): FakeQuanterWithAbsMaxObserverLayer() ) (quant): QuanterStub( (_observer): FakeQuanterWithAbsMaxObserverLayer() ) ) c.>[TU]5 XlgN)super__init__ _observer)selfobserver __class__s T/var/www/html/banglarbhumi/venv/lib/python3.13/site-packages/paddle/nn/quant/stub.pyr Stub.__init__Ns !cU$rrinputs rforward Stub.forwardRs rrr)rzQuanterFactory | NonereturnNonerrrr __name__ __module__ __qualname____firstlineno____doc__rr__static_attributes__ __classcell__rs@rr r s.`""rr c:^\rSrSrSrSU4SjjrSSjrSrU=r$) QuanterStubVa It is an identity layer with an observer observing the input. Before QAT or PTQ, the stub in the model will be replaced with an instance of QuanterStub. The user should not use this class directly. Args: layer(paddle.nn.Layer): The stub layer with an observer configure factory. If the observer of the stub layer is none, it will use 'q_config' to create an observer instance. q_config(QuantConfig): The quantization configuration for the current stub layer. c>[TU]5 SUlURb!URRU5UlgURb!URRU5Ulggr)rrr _instance activation)rlayerq_configrs rrQuanterStub.__init__bs^  ?? &"__66u=DN  ,%00::5ADN-rcBURbURU5$U$rrrs rrQuanterStub.forwardjs(,(Bt~~e$MMrr)r1r r2rrr r!r"r*s@rr,r,Vs BNNrr,N)r' __future__rtypingr layer.layersrpaddlerpaddle.quantizationrpaddle.quantization.factoryr r r,rrrr<s:(" /:656rN%Nr