o
    i^
                     @   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   X/home/ubuntu/veenaModal/venv/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   	