U Kc@s<ddlZddlmZmZmZmZmZmZmZm Z m Z m Z m Z ddl mZmZeejejejejdZeejejdZeejejejejdZeejejejejddZeejejejejddZeejejejejdZeejejejd Zeejejejd Zed d d Zed d dZddZddZ d dddgZ!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) BackendConfig DTypeConfig) input_dtype output_dtype weight_dtype bias_dtype)rrT)rrrrZ is_dynamic)rrr)returnc Cstg}ttttg}ttg}tg}ttg}ttg}ttg}ttg}tg}td 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_int8_dtype_config!default_dynamic_int8_dtype_config$default_dynamic_float16_dtype_configdefault_op_fp16_dtype_configdefault_op_quint8_dtype_configweight_only_quint8_dtype_config!weight_only_quint4x2_dtype_config%input_output_only_quint8_dtype_configrset_backend_pattern_configsrr rset_backend_pattern_configrrr r rr r r Zconv_dtype_configsZlinear_dtype_configsZbinary_op_dtype_configsZdefault_op_dtype_configsZfixed_qparams_op_dtype_configsZshare_qparams_op_dtype_configsZrnn_op_dtype_configsZembedding_op_dtype_configsZlayer_norm_op_dtype_configsr O/tmp/pip-unpacked-wheel-gikjz4vx/torch/ao/quantization/backend_config/native.py*get_test_only_legacy_native_backend_configZs\   r"c Cstg}tttg}tg}tg}tg}tg}ttg}ttg}tg}tdt |t |t | t |t|t|t|t|t|t|t|S)zQ Return the `BackendConfig` for PyTorch Native backend (fbgemm/qnnpack). Znative)rrrrrrrrrrr rrrrr r rr r rrr r r!get_native_backend_configsN   r#cCs tS)zd Return the `BackendConfig` for PyTorch Native backend (fbgemm/qnnpack) in dictionary form. )r#to_dictr r r r!get_native_backend_config_dictsr%cCs tS)z Return the `BackendConfig` for PyTorch Native backend (fbgemm/qnnpack) with various additional fp16 ops in dictionary form. )r"r$r r r r!/get_test_only_legacy_native_backend_config_dictsr&)"ZtorchZ_common_operator_config_utilsrrrrrrr r r r r Zbackend_configrrZquint8Zqint8floatrrZfloat16rrrrrZquint4x2rr"r#r%r&__all__r r r r!sr4     .%