|jddlmZddlZddlmZmZddlmZerddlm Z ddl mZ ddl m Z Gddeej ZdS) ) annotationsN) TYPE_CHECKINGAny)Layer)Iterable)TensorceZdZdZdfd ZejddZejdd Zejdd Z ejdd Z ejdd Z xZ S) BaseQuanterzw Built-in quanters and customized quanters should extend this base quanter and implement abstract methods. returnNonecHtdSN)super__init__)self __class__s p/lsinfo/ai/hellotax_ai/data_center/backend/venv/lib/python3.11/site-packages/paddle/quantization/base_quanter.pyrzBaseQuanter.__init__#s inputrTensor | npt.NDArray[Any]cdSr)rrs rforwardzBaseQuanter.forward&s rcdS)z Get the scales used for quantization. It can be none which means the quanter didn't hold scales for quantization. Nrrs rscaleszBaseQuanter.scales* rcdS)z Get the zero points used for quantization. It can be none which means the quanter didn't hold zero points for quantization. Nrrs r zero_pointszBaseQuanter.zero_points2rrint | Iterable[int]cdS)zT Get the axis of quantization. None means tensor-wise quantization. Nrrs r quant_axiszBaseQuanter.quant_axis: rcdS)z5 Get the bit length of quantization. Nrrs r bit_lengthzBaseQuanter.bit_lengthAr#r)r r )rrr r)r r)r r ) __name__ __module__ __qualname____doc__rabcabstractmethodrrrr"r% __classcell__)rs@rr r s                              rr ) metaclass) __future__rr*typingrr paddle.nnrcollections.abcr numpy.typingnptpaddlerABCMetar rrrr6s#""""" %%%%%%%%(((((() ) ) ) ) %3;) ) ) ) ) ) r