o
    0ñh^
  ã                   @   sh   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lT d dlT d dl	T dd„ Z
g d¢ZdS )é   )Ú*)Ú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© )ÚmodelÚ
calib_dataÚdataÚ_targetr   r   úO/var/www/vscode/kcb/lib/python3.10/site-packages/torch/quantization/__init__.pyÚdefault_eval_fn   s   
ÿr
   )2ÚQuantWrapperÚ	QuantStubÚDeQuantStubÚquantizeÚquantize_dynamicÚquantize_qatÚprepareÚconvertÚprepare_qatÚquantize_jitÚquantize_dynamic_jitÚ_prepare_ondevice_dynamic_jitÚ_convert_ondevice_dynamic_jitÚ_quantize_ondevice_dynamic_jitÚ	QuantTypeÚ(get_default_static_quant_module_mappingsÚget_static_quant_module_classÚ)get_default_dynamic_quant_module_mappingsÚget_default_qat_module_mappingsÚ$get_default_qconfig_propagation_listÚ&get_default_compare_output_module_listÚget_quantized_operatorÚget_fuser_methodÚpropagate_qconfig_Úadd_quant_dequantÚswap_moduler
   ÚObserverBaseÚWeightObserverÚHistogramObserverÚobserverÚdefault_observerÚdefault_weight_observerÚdefault_placeholder_observerÚ#default_per_channel_weight_observerÚdefault_fake_quantÚdefault_weight_fake_quantÚ.default_fixed_qparams_range_neg1to1_fake_quantÚ+default_fixed_qparams_range_0to1_fake_quantÚ%default_per_channel_weight_fake_quantÚdefault_histogram_fake_quantÚQConfigÚdefault_qconfigÚdefault_dynamic_qconfigÚfloat16_dynamic_qconfigÚ!float_qparams_weight_only_qconfigÚdefault_qat_qconfigr   r   r   N)Úfake_quantizer   Úfuser_method_mappingsr(   ÚqconfigÚ
quant_typeÚquantization_mappingsr   r   Ústubsr
   Ú__all__r   r   r   r	   Ú<module>   s   	