
    sh2                      @   S SK r SSKJrJrJrJrJrJrJrJ	r	J
r
JrJrJr  SSKJrJr  / SQr\" \ R$                  \ R$                  \ R&                  \ R(                  S9r\" \ R$                  \ R$                  S9r\" \ R.                  \ R.                  \ R.                  \ R.                  S9r\" \ R$                  \ R(                  \ R&                  \ R(                  SS	9r\" \ R.                  \ R(                  \ R.                  \ R(                  SS	9r\" \ R$                  \ R$                  \ R(                  \ R(                  S9r\" \ R(                  \ R(                  \ R$                  S
9r\" \ R(                  \ R(                  \ R:                  S
9rS\4S jrS\4S jr S r!S r"g)    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_configdefault_op_quint8_dtype_configdefault_op_fp16_dtype_config!default_dynamic_int8_dtype_config$default_dynamic_float16_dtype_config%input_output_only_quint8_dtype_configweight_only_quint8_dtype_config!weight_only_quint4x2_dtype_configget_native_backend_configget_native_backend_config_dict/get_test_only_legacy_native_backend_config_dict)input_dtypeoutput_dtypeweight_dtype
bias_dtype)r   r   T)r   r   r   r    
is_dynamic)r   r   r   returnc                  4   [         /n [         [        [        [        /n[        [        /n[        /n[        [        /n[        [        /n[        /n[        [        /n[
        [        /n[        /n	[        S5      R                  [        U 5      5      R                  [        U5      5      R                  [        U5      5      R                  [        U5      5      R                  [        U5      5      R                  [!        U5      5      R                  [#        U5      5      R                  [%        U5      5      R                  ['        U5      5      R                  [)        U	5      5      R                  [+        U5      5      R                  [-        U5      5      $ )zj
Return the `BackendConfig` for PyTorch Native backend (fbgemm/qnnpack) with various additional fp16 ops.
_native_and_fp16)weighted_op_quint8_dtype_configr   r   r   r   r   r   r   r   set_backend_pattern_configsr   r   r   set_backend_pattern_configr   r   r
   r   r   r   r   r   r	   
conv_dtype_configslinear_dtype_configsbinary_op_dtype_configsdefault_op_dtype_configsfixed_qparams_op_dtype_configsshare_qparams_op_dtype_configstensor_info_op_dtype_configsrnn_op_dtype_configsembedding_op_dtype_configslayer_norm_op_dtype_configss
             /Users/tiagomarins/Projetos/claudeai/copy_bank/venv/lib/python3.13/site-packages/torch/ao/quantization/backend_config/native.pyr   r   l   s    ::'),$	 	'$ !??&$&"
 	'$&"
 	'$  	*,
 	()" $I"I()	$	$%67I%J	K	$	$%89M%N	O	$	$%;<S%T	U	#	#O4L$M	N	$	$%<=U%V	W	$	$)*HI

 
%	$)*HI

 
%	$'(DE

 
%	$_5M%N	O	$	$_5P%Q	R	$	$%89M%N	O	$	$%&@A

'    c                     [         /n [         [        [        /n[        /n[        /n[        /n[        /n[        /n[        [        /n[        [
        /n[        /n	[        S5      R                  [        U 5      5      R                  [        U5      5      R                  [        U5      5      R                  [        U5      5      R                  [        U5      5      R                  [        U5      5      R                  [!        U5      5      R                  [#        U5      5      R                  [%        U5      5      R                  ['        U	5      5      R                  [)        U5      5      R                  [+        U5      5      $ )zI
Return the `BackendConfig` for PyTorch Native backend (fbgemm/qnnpack).
native)r%   r   r   r   r   r   r   r   r&   r   r   r   r'   r   r   r
   r   r   r   r   r   r	   r(   s
             r3   r   r      sg   
 ::'),
  >> >?&D%E"&D%E"$B#C ),
 	()" $I"Ih	$	$%67I%J	K	$	$%89M%N	O	$	$%;<S%T	U	#	#O4L$M	N	$	$%<=U%V	W	$	$)*HI

 
%	$)*HI

 
%	$'(DE

 
%	$_5M%N	O	$	$_5P%Q	R	$	$%89M%N	O	$	$%&@A

'r4   c                  2    [        5       R                  5       $ )z\
Return the `BackendConfig` for PyTorch Native backend (fbgemm/qnnpack) in dictionary form.
)r   to_dict r4   r3   r   r      s     %&..00r4   c                  2    [        5       R                  5       $ )z}
Return the `BackendConfig` for PyTorch Native backend (fbgemm/qnnpack) with various additional
fp16 ops in dictionary form.
)r   r8   r9   r4   r3   r   r      s    
 67??AAr4   )#torch_common_operator_config_utilsr   r   r   r   r   r	   r
   r   r   r   r   r   backend_configr   r   __all__quint8qint8floatr%   r   float16r   r   r   r   r   quint4x2r   r   r   r   r   r9   r4   r3   <module>rD      s       7( #.{{	#  "-" 
  +}}	   %0{{ 	% ! (3{{ 	( $ )4{{	) % #.#  %0% !:M :z/= /d1Br4   