U ,-eZ@s|ddlmZmZddlmZmZddlmZmZm Z m Z m Z m Z ddl Z ddl mZddlmZddlmZddlmZd d d d d dddgZe je je je je jgZe je je je je j gZ!Gdd d eZ"edddGdd d e"Z#edddGdd d e"Z$e e eefefZ%de%_&edddGddde"Z'edddGddde"Z(eGdddZ)Gdd d eZ*dS))ABCabstractmethod) dataclassfield)CallableDictListOptionalTupleUnionN)Tensor)ObserverOrFakeQuantize)"_ObserverOrFakeQuantizeConstructor)Node QuantizerQuantizationSpecBaseQuantizationSpecFixedQParamsQuantizationSpec EdgeOrNodeSharedQuantizationSpecDerivedQuantizationSpecQuantizationAnnotationc@seZdZdZdS)rzBase class for different types of quantization specs that allows users to specify how to quantize a Tensor (input/output of a Node) in the model N)__name__ __module__ __qualname____doc__rrj/var/www/html/Darija-Ai-Train/env/lib/python3.8/site-packages/torch/ao/quantization/quantizer/quantizer.pyr!sT)eqfrozenc@szeZdZUdZejed<eed<dZe e ed<dZ e e ed<dZ e ej ed<dZ e e ed<d Zeed <d d ZdS) rzQuantization spec for common operators that allows user to specify how to quantize a Tensor, this includes dtype, quant_min, quant_max etc. dtypeZobserver_or_fake_quant_ctrN quant_min quant_maxqschemech_axisF is_dynamiccCs|jtkrtd|jd|jdk rV|jdk rV|j|jkrVtd|jd|jd|jdk r||jtkr|td|jd|jdk r|jdkrtddS)NzUnsupported dtype .z quant_min z must be <= quant_max zUnsupported qscheme rzCh_axis is < 0.) r SUPPORTED_DTYPES TypeErrorr!r" ValueErrorr#SUPPORTED_QSCHEMESr$)selfrrr __post_init__;s  zQuantizationSpec.__post_init__)rrrrtorchr __annotations__rr!r intr"r#r$r%boolr,rrrrr)s   c@sZeZdZUejed<eed<eed<dZe eed<dZ e eed<dZ e ej ed<dS)rr scaleZ zero_pointNr!r"r#) rrrr-r r.floatr/r!r r"r#rrrrrTs  z)torch.ao.quantization.quantizer.quantizerc@seZdZUdZeed<dS)rzg Quantization spec for the Tensors whose quantization parameters are shared with other Tensors Z edge_or_nodeN)rrrrrr.rrrrrhs c@seZdZUdZeeed<eeege e e ffed<e j ed<dZ eeed<dZeeed<dZee jed<dZeeed <dS) rz^Quantization spec for the Tensors whose quantization parameters are derived from other TensorsZ derived_fromZderive_qparams_fnr Nr!r"r#r$)rrrrrrr.rr r r r-r r!r r/r"r#r$rrrrrqs   c@sHeZdZUdZeedZeee fe d<dZ e e e d<dZ ee d<dS)rzHow are input arguemnt or output should be quantized, expressed as QuantizationSpec, this corresponds to how a Tensor in the operator Graph is observed (PTQ) or fake quantized (QAT) )default_factoryinput_qspec_mapN output_qspecF _annotated)rrrrrdictr4rrrr.r5r r6r0rrrrr~s c@s@eZdZeejjejjdddZeejjddddZdS)r)modelreturncCsdSNrr+r8rrrannotateszQuantizer.annotateNcCsdSr:rr;rrrvalidateszQuantizer.validate) rrrrr-ZfxZ GraphModuler<r=rrrrrs)+abcrr dataclassesrrtypingrrrr r r r-r Ztorch.ao.quantizationr Ztorch.ao.quantization.qconfigrZtorch.fxr__all__Zuint8Zint8Zint32Zfloat16Zfloat32r'Zper_tensor_affineZper_tensor_symmetricZper_channel_affineZper_channel_symmetricZ per_channel_affine_float_qparamsr*rrrrrrrrrrrrrsJ        *