U yh @sNddlZddlmZmZmZmZmZmZmZm Z m Z m Z m Z m Z ddlmZmZddddd d d d d ddg ZeejejejejdZeejejdZeejejejejdZeejejejejddZeejejejejddZeejejejejdZeejejejdZeejejejdZedddZeddd Z ddZ!ddZ"dS)N) _get_binary_op_configs_get_bn_configs_get_cat_config_get_conv_configs_get_default_op_configs_get_embedding_op_configs_get_fixed_qparams_op_configs_get_linear_configs_get_ln_configs_get_rnn_op_configs_get_share_qparams_op_configs_get_tensor_info_op_configs) BackendConfig DTypeConfig*get_test_only_legacy_native_backend_configdefault_op_quint8_dtype_configdefault_op_fp16_dtype_config!default_dynamic_int8_dtype_config$default_dynamic_float16_dtype_config%input_output_only_quint8_dtype_configweight_only_quint8_dtype_config!weight_only_quint4x2_dtype_configget_native_backend_configget_native_backend_config_dict/get_test_only_legacy_native_backend_config_dict) input_dtype output_dtype weight_dtype bias_dtype)rrT)rrrrZ is_dynamic)rrr)returnc Cstg}ttttg}ttg}tg}ttg}ttg}tg}ttg}ttg}tg} td t | t | t | t| t| t| t| t| t| t|  t| t|S)zr Return the `BackendConfig` for PyTorch Native backend (fbgemm/qnnpack) with various additional fp16 ops. Z_native_and_fp16)weighted_op_quint8_dtype_configrrrrrrrrset_backend_pattern_configsrr rset_backend_pattern_configrrr r rrr r r Zconv_dtype_configsZlinear_dtype_configsZbinary_op_dtype_configsZdefault_op_dtype_configsZfixed_qparams_op_dtype_configsZshare_qparams_op_dtype_configsZtensor_info_op_dtype_configsZrnn_op_dtype_configsZembedding_op_dtype_configsZlayer_norm_op_dtype_configsr%]/var/www/html/venv/lib/python3.8/site-packages/torch/ao/quantization/backend_config/native.pyrisd    c Cstg}tttg}tg}tg}tg}tg}tg}ttg}ttg}tg} tdt |t |t | t |t|t|t|t|t|t| t|t|S)zQ Return the `BackendConfig` for PyTorch Native backend (fbgemm/qnnpack). native)r!rrrrrrrr"rr rr#rrr r rrr r rr$r%r%r&rsT    cCs tS)zd Return the `BackendConfig` for PyTorch Native backend (fbgemm/qnnpack) in dictionary form. )rto_dictr%r%r%r&rscCs tS)z Return the `BackendConfig` for PyTorch Native backend (fbgemm/qnnpack) with various additional fp16 ops in dictionary form. )rr(r%r%r%r&rs)#ZtorchZ_common_operator_config_utilsrrrrrrr r r r r rZbackend_configrr__all__Zquint8Zqint8floatr!rZfloat16rrrrrZquint4x2rrrrrr%r%r%r&s8   2'