U
    Ç9%e`
  ã                2   @   sÈ   d dl T d dlT d dlT d dlT d dlmZ d dlT d dlT d dlT d dlT d dl	T dd„ Z
ddddd	d
ddddddddddddddddddddddd d!d"d#d$d%d&d'd(d)d*d+d,d-d.d/d0d1d2dd
d3g2Zd4S )5é   )Ú*)Úfuse_modulesc                 C   s   |D ]\}}| |ƒ qdS )z†
    Default evaluation function takes a torch.utils.data.Dataset or a list of
    input Tensors and run the model on the dataset
    N© )ÚmodelZ
calib_dataÚdataÚtargetr   r   úZ/var/www/html/Darija-Ai-API/env/lib/python3.8/site-packages/torch/quantization/__init__.pyÚdefault_eval_fn   s    r	   ZQuantWrapperZ	QuantStubZDeQuantStubÚquantizeZquantize_dynamicZquantize_qatÚprepareÚconvertZprepare_qatÚquantize_jitZquantize_dynamic_jitZ_prepare_ondevice_dynamic_jitZ_convert_ondevice_dynamic_jitZ_quantize_ondevice_dynamic_jitZ	QuantTypeZ(get_default_static_quant_module_mappingsZget_static_quant_module_classZ)get_default_dynamic_quant_module_mappingsZget_default_qat_module_mappingsZ$get_default_qconfig_propagation_listZ&get_default_compare_output_module_listZget_quantized_operatorZget_fuser_methodZpropagate_qconfig_Zadd_quant_dequantZswap_moduleZObserverBaseZWeightObserverZHistogramObserverÚobserverZdefault_observerZdefault_weight_observerZdefault_placeholder_observerZ#default_per_channel_weight_observerZdefault_fake_quantZdefault_weight_fake_quantZ.default_fixed_qparams_range_neg1to1_fake_quantZ+default_fixed_qparams_range_0to1_fake_quantZ%default_per_channel_weight_fake_quantZdefault_histogram_fake_quantZQConfigZdefault_qconfigZdefault_dynamic_qconfigZfloat16_dynamic_qconfigZ!float_qparams_weight_only_qconfigZdefault_qat_qconfigr   N)r
   r   ZqconfigZfake_quantizer   ZstubsZ
quant_typer   Zquantization_mappingsZfuser_method_mappingsr	   Ú__all__r   r   r   r   Ú<module>   sz   
Â