a
    ¢º”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© )ÚmodelZ
calib_dataÚdataÚ_targetr   r   úI/var/www/auris/lib/python3.9/site-packages/torch/quantization/__init__.pyÚdefault_eval_fn   s    r	   )2ZQuantWrapperZ	QuantStubZDeQuantStubÚquantizeZquantize_dynamicÚquantize_qatÚprepareÚconvertÚ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_moduler	   Z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   r   r   N)Zfake_quantizer   Zfuser_method_mappingsr   ZqconfigZ
quant_typeZquantization_mappingsr
   r   Zstubsr	   Ú__all__r   r   r   r   Ú<module>   s   	