
    [Th|G                     V   S SK r S SKJrJr  S SKrS SKJs  Js  Jr	  S SK
Js  Js  Js  Jr  S SKJs  Js  Js  Jr  S SKJs  Js  Js  Js  Jr  S SKJs  Js  Jr  S SKJs  Js  Js  Jr  S SKJs  Js  Jr  S SKJs  Js  Js  Jr  S SKJs  Js  Js  Jr  S SKJs  Js  J r   S SK!Jr  S SK"Js  J#r$  S SK%J&r&  SSK'J(r(  \RR                  R                  r*S\+\,\-\(   4   4S jr.S\+\,\-\(   4   S	\(S\\,   4S
 jr/S\+\,\-\(   4   S	\(S\\(   SS4S jr0S\+\,\-\(   4   4S jr1S\+\,\-\(   4   4S jr2g)    N)CallableOptional)get_native_backend_config   )NSNodeTargetTypereturnc                     / [         R                  1P[         R                  1P[         R                  1P[        R
                  1P[        R                  1P[        R                  1P[         R                  1P[        R                  1P[         R                  [        R                  1P[         R                  [        R                  R                  R                   1P[         R"                  [        R                  R                  R$                  1P[         R&                  [        R(                  1P[         R*                  [        R,                  1P[         R.                  [        R0                  1P[         R2                  1P[        R4                  [6        R4                  1P[        R8                  1P[        R:                  [6        R:                  1P[        R<                  [         R>                  SS[        R<                  1P[         R@                  [        RB                  1P[         RD                  [        RF                  1P[         RH                  [        RJ                  1P[        RL                  SS[         RN                  [        RL                  1P[         RP                  1P[         RR                  1P[         RT                  1P[         RV                  1P[         RX                  1P[        RZ                  1P[        R\                  1P[        R^                  1P[         R`                  1P[         Rb                  1P[         Rd                  1P[         Rf                  1P[         Rh                  1P[         Rj                  1P[         Rl                  1P[         Rn                  1P[         Rp                  1P[         Rr                  1P[         Rt                  [        Rv                  1P[        Rx                  1P[        Rz                  1P[        R|                  1P[        R~                  1P[        R                  1P[        R                  1P[         R                  [        R                  1P[         R                  [        R                  1P[         R                  [        R                  [        R                  SS1PSS[        R                  [         R                  1P[         R                  [        R                  [        R                  1P[6        R                  1P[        R                  1P[        R                  1P[        R                  1P[        R                  1P[        R                  1P[        R                  1P[        R                  1P[        R                  1P[        R                  1P[        R                  1P[        R                  1P[        R                  1P[        R                   R                  R                  1P[         R                  1P[        R                  1P[        R                  1P[         R                  1P[         R                  [        R                  1P[        R                  [        R                  1P[         R                  1P[        R                  1P[         R                  1P[        R                  1P[        R                  1Pn [        5       n[         R                  [         R                  R                  R                  4/nUR                  R                  5        H  u  p4Un[        U[        [        45      (       a"  US	   n[        U[        [        45      (       a  M"  UR                  b  UR                  XTR                  45        UR                  b  UR                  XTR                  45        UR                  c  M  UR                  XTR                  45        M     [        R                  [        R                  [        R                  [        R                  4 H.  nUR                  5        H  u  pxUR                  Xx45        M     M0     [        R                  [        R                  [        R                  4 HC  n	U	R                  5        H,  u  nu  pUR                  Xz45        UR                  X{45        M.     ME     [        R                  R                  5        H-  u  nu  n
nUR                  Xz45        UR                  X{45        M/     [        R                  [        R                  G[         GR                  4 H.  nUR                  5        H  u  pxUR                  Xx45        M     M0     G[         GR                  4 H.  nUR                  5        H  u  pxUR                  Xx45        M     M0     U H=  u  pU  H2  nX;   d  X;   d  M  UR5                  U5        UR5                  U5          M;     M?     0 nG[        U 5       H  u  nnG[	        U5      nXU'   M     U$ )
Nrelurelu_sigmoidsigmoid_tanh_tanhhardsigmoid_hardsigmoid)nnConv1dConv2dConv3dFconv1dconv2dconv3dLinearlinear	AvgPool1dtorch
avg_pool1d	AvgPool2d_C_nn
avg_pool2d	AvgPool3d
avg_pool3dAdaptiveAvgPool1dadaptive_avg_pool1dAdaptiveAvgPool2dadaptive_avg_pool2dAdaptiveAvgPool3dadaptive_avg_pool3dLSTMaddoperatorcatmulr
   ReLU	MaxPool1d
max_pool1d	MaxPool2d
max_pool2d	MaxPool3d
max_pool3dr   SigmoidBatchNorm2dBatchNorm3dConvTranspose1dConvTranspose2dConvTranspose3dconv_transpose1dconv_transpose2dconv_transpose3dELU	EmbeddingEmbeddingBag	GroupNorm	HardswishInstanceNorm1dInstanceNorm2dInstanceNorm3d	LayerNorm	LeakyReLUReLU6relu6elu	hardswish
group_norminstance_norm
layer_norm
leaky_reluSiLUsiluMishmishTanhr   r   HardsigmoidHardtanhhardtanh	hardtanh_floordiv	unsqueezestacksqueezesortrepeat_interleaveminmeanmax	transposeflattenclampchunk
functionalinterpolateDropoutdropoutmatmulSoftmaxPReLUnnqprelutoqPixelShufflepixel_shufflePixelUnshufflepixel_unshufflenarrowr   modulesNonDynamicallyQuantizableLinear!_pattern_complex_format_to_configitems
isinstancelisttuplefused_moduleappend
qat_modulereference_quantized_module_lower_to_native_backendSTATIC_LOWER_MODULE_MAPDYNAMIC_LOWER_MODULE_MAPWEIGHT_ONLY_LOWER_MODULE_MAP SPECIAL_PATTERN_LOWER_MODULE_MAPSTATIC_LOWER_FUSED_MODULE_MAP(STATIC_LOWER_FUSED_MODULE_TWO_INPUTS_MAPDYNAMIC_LOWER_FUSED_MODULE_MAPSTATIC_LOWER_FUNCTIONAL_MAPQBIN_OP_MAPPINGQBIN_RELU_OP_MAPPINGquantization_mappings,DEFAULT_FLOAT_TO_QUANTIZED_OPERATOR_MAPPINGS%DEFAULT_DYNAMIC_QUANT_MODULE_MAPPINGS	enumeratestr)sets_of_related_opsbackend_confignew_connectionspatternconfigfirst_elementsource_to_targetsourcetargetsource_to_double_targettarget1target2item1item2set_of_related_ops base_name_to_sets_of_related_opscounter	base_names                     O/var/www/auris/envauris/lib/python3.13/site-packages/torch/ao/ns/fx/mappings.py$get_base_name_to_sets_of_related_opsr      s
   G8 II	
G8 II	
G8 II	
G8 HH	
G8  HH	
G8& HH	
%G8. II	
-G86 HH	
5G8> LL	
=G8F LLHHLL##	
EG8N LLHHLL##	
MG8X   !!	
WG8`   !!	
_G8h   !!	
gG8r GG	
qG8z IILL	
yG8D II	
CG8L IILL	
KG8V FFGGJJ	
UG8f LLLL	
eG8n LLLL	
mG8v LLLL	
uG8@ MMJJII	
G8P NN	
OG8V NN	
UG8^ 	
]G8d 	
cG8j 	
iG8r 	
qG8x 	
wG8~ 	
}G8F FF	
EG8N LL	
MG8V OO	
UG8^ LL	
]G8f LL	
eG8n 	
mG8t 	
sG8z 	
yG8B LL	
AG8J LL	
IG8R HHGG	
QG8\ EE	
[G8d KK	
cG8l LL	
kG8t OO	
sG8| LL	
{G8D LL	
CG8L GGFF	
KG8V GGFF	
UG8` GGFFJJ	
_G8p MMNN		
oG8~ KKJJKK	
}G8J 	
IG8R OO	
QG8Z KK	
YG8b MM	
aG8j JJ	
iG8r ##	
qG8z II	
yG8B JJ	
AG8J II	
IG8R OO	
QG8Z MM	
YG8b KK	
aG8j KK	
iG8r HH++	
qG8z JJ	
yG8B	 II	
A	G8J	 LL	
I	G8R	 JJ	
Q	G8Z	 HHII	
Y	G8d	 GGII	
c	G8n	 OO	
m	G8t	 OO	
s	G8|	 	
{	G8B
 	
A
G8J
 LL	
I
G8V
 /0N 
BJJ%%EEF8O
 *KKQQSu66)"-M u66 * ""M3F3F#GH( ""M3D3D#EF,,8 ""M3T3T#UV) T6 	!88 99 == AA	 /446NF""F#34 7 	!>> II ??$
 +B*G*G*I&F&W""F#45""F#45 +J$  
"	=	=	C	C	E	 0101 
F
 	!00 55JJ
 /446NF""F#34 7 	CC /446NF""F#34 7 ("5*e.I"&&u-"&&u-	 #6 ( JL$'01D'E##L	6H3 (F ,+    r   opc                 H    U R                  5        H  u  p#X;   d  M  Us  $    g )N)r{   )r   r   r   r   s       r   get_base_name_for_opr     s-     *J)O)O)Q%	# *R r   
related_opc                     Ub=  U R                  5        H  nX#;   d  M
  UR                  U5          g    [        U S35      eSn[        U5      U ;   a  US-  n[        U5      U ;   a  M  U1U [        U5      '   g )Nz was not foundr   r   )valuesr-   AssertionErrorr   )r   r   r   r   r   s        r   add_op_to_sets_of_related_opsr     s    
 "B"I"I"K/"&&r* #L
 
|>:;;'l>>qLG 'l>>:<(W6r   c                  T   [         R                  [         R                  [         R                  [         R                  [
        R                  [         R                  [         R                  [         R                  [         R                  [         R                  [         R                  [         R                  [         R                  [        R                   [
        R                   [        R"                  [
        R"                  [
        R$                  [         R&                  1n [)        5       n[*        R                  [*        R,                  [*        R                  [*        R.                  [*        R                  [*        R0                  [*        R                  [*        R2                  [*        R                  [*        R                  [*        R                  [*        R                  [*        R                  [*        R                  [*        R                  [*        R&                  1n1 [         R4                  i[         R6                  i[
        R6                  i[         R8                  i[
        R8                  i[         R:                  i[        R<                  i[
        R>                  i[         R@                  i[         RB                  i[         R                  i[         RD                  i[         RF                  i[         RH                  i[         RJ                  i[         RL                  i[         RN                  i[         RP                  i[         RR                  i[         RT                  i[
        RV                  i[
        RX                  RZ                  R\                  i[
        RX                  RZ                  R^                  i[
        R                  i[
        R`                  i[
        Rb                  i[
        Rd                  i[
        Rf                  i[
        Rh                  i[
        Rj                  i[
        Rl                  i[
        Rn                  i[
        Rp                  i[
        Rr                  i[
        Rt                  i[
        Rv                  i[
        Rx                  i[        R                   in1 [z        R|                  i[~        R|                  i[        R|                  i[        R|                  i[
        Rz                  R                  R                  R                  i[z        R                  i[z        R                  i[z        R                  i[~        R                  i[~        R                  i[~        R                  i[~        R                  i[~        R                  i[z        R                  i[        R                  i[z        R                  i[z        R                  i[z        R                  i[z        R                  i[z        R                  i[z        R                  i[z        R                  i[z        R                  i[z        R                  i[z        R                  i[z        R                  i[z        R                  i[z        R                  i[z        R                  i[z        R                  i[z        R                  i[z        R                  i[z        R                  i[z        R                  i[        R                  i[        R                  i[        R                  i[        R                  i[        R                  i[        R                  i[        R                  i[        R                  i[        R                  i[        R                  i[        R                  i[        R                  i[        R                  i[        R                  i[        R                  i[        R                  i[        R                  i[        R                  i[        R                  i[        R                  i[        R                  i[        R                  i[        R                  i[        R                  i[        R                  i[        R                  in1 [        R|                  i[        R                  i[        R                  i[        R                  i[        R                  i[        R                  i[        R                  i[        R                  i[        R                  i[        R                  i[        R                  i[        R                  i[        R                  i[        R                  i[        R                  i[        R                  i[        R                  i[        R                  i[        R                  i[        R                  i[        R                  i[        R                  i[        R                  i[        R                  i[        R                  i[        R                  i[        R                  i[        R                  i[        R                  i[        R                  i[        R                  in[z        R                  [z        R                  [z        R                  [z        R                  [z        R                  [z        R                  [z        R                  [z        R                  [z        R                  [z        R                  [z        R                  [z        R                  [z        R                  [z        R                  [z        GR                   [z        GR                  [z        GR                  [z        GR                  [z        R                  1n1 SknU UUUUUUUS.$ )N>   r
   r   r   r   r   r   r   r   )funs_io_type_fp32funs_io_type_fp16funs_io_type_int8funs_io_type_fp32_or_int8mods_io_type_fp32mods_io_type_int8mods_io_type_fp32_or_int8meths_io_type_fp32_or_int8)r   r   r   r   r   r   r/   rM   rN   rP   rQ   rR   rl   rT   rV   r.   r-   r0   sumrq   setrr   linear_reluconv1d_reluconv2d_reluconv3d_relur
   r   r   r   r\   r'   r)   r+   rZ   r[   rj   r3   r5   r7   rL   rt   rv   r   r!   r"   r#   r%   rh   rg   rf   re   rd   rc   rb   rw   ra   r`   r_   r^   r]   r   r   nnqatnnqatdnnqdrx   ry   r   r   r   rB   rC   r,   r9   r:   rk   r;   r<   r=   rA   rD   rF   rG   rH   rI   rE   rJ   rK   rS   rU   rn   ro   nniBNReLU2dBNReLU3d
ConvReLU1d
ConvReLU2d
ConvReLU3d
LinearReLU
LinearBn1dConvBn1dConvBn2dConvBn3dnniqatConvBnReLU1dConvBnReLU2dConvBnReLU3dnniqdLinearLeakyReLU
LinearTanh	ConvAdd2dConvAddReLU2drp   nniqr1   rW   r8   rX   r&   r(   r*   r   r    r$   rY   Identityr2   r4   r6   rs   ru   )FUNS_IO_TYPE_FP32FUNS_IO_TYPE_FP16FUNS_IO_TYPE_INT8FUNS_IO_TYPE_FP32_OR_INT8MODS_IO_TYPE_FP32MODS_IO_TYPE_INT8MODS_IO_TYPE_FP32_OR_INT8METHS_IO_TYPE_FP32_OR_INT8s           r   get_node_type_to_io_type_mapr     s	   																							'0, 03u 	







		!0.'8	'8	'8 	

'8 	
			'8
 	'8 	
'8 	'8 	!!'8 	
'8 	
'8 	
		'8 	


'8 	
'8 	
'8 	
'8  	
!'8" 	
#'8$ 	
%'8& 	
''8( 	
)'8* 	+'8, 	-'8. 	/'80 			1'82 	3'84 	5'86 	7'88 	9'8: 			;'8< 	

='8> 			?'8@ 	A'8B 	C'8D 	

E'8F 	G'8H 	I'8J 	K'8L 	M'8R?0
		?0?0 	?0 		?0
 	???0 			?0 			?0 			?0 	?0 	?0 	?0 	?0 	?0 	?0" 			#?0$ 	%?0& 	'?0( 	

)?0* 	+?0, 	-?0. 	/?00 	1?02 	3?04 	5?06 	7?08 	9?0: 	;?0< 	=?0> 	??0@ 	A?0B 	C?0D 	E?0F 	

G?0H 	I?0J 	K?0L 	M?0N 	O?0P 	Q?0R 	S?0T 	U?0V 	W?0X 	Y?0Z 	[?0\ 	]?0^ 	_?0` 	a?0b 	c?0d 	e?0f 	g?0h 	i?0j 	k?0l 	m?0n 	o?0p 	q?0r 	s?0t 	u?0v 	w?0x 	y?0z 	{?0| 	}?0B 0

 0

 0 	

 0 	

	 0
 	 0 	 0 	 0 	 0 	 0 	 0 	 0 	 0 	 0 	 0 	 0  	! 0" 	# 0$ 	% 0& 	' 0( 	) 0* 			+ 0, 	- 0. 	/ 00 	1 02 	3 04 	5 06 	7 08 	9 0: 	; 0< 	= 0> 	? 0F 	





















'8,	9 /..%>..%>&@	 	r   c                  x    [         R                  [        R                  1n [        R
                  1n1 SknU UUS.$ )N>   torc   sizeviewrh   rg   numelshapedetachrepeatdetach_permutereshaperesize_r_   squeeze_re   r]   
contiguous
dequantize
unsqueeze_ra   )funs_unmatchablemods_unmatchablemeths_unmatchable)r   quantize_per_tensorr.   getitemr   r   )FUNS_UNMATCHABLEMODS_UNMATCHABLEMETHS_UNMATCHABLEs      r   get_unmatchable_types_mapr     sJ    !!/ 	/04 -,. r   )3r.   typingr   r   r   torch.ao.nn.intrinsicaor   	intrinsicr   torch.ao.nn.intrinsic.qatqatr   torch.ao.nn.intrinsic.quantized	quantizedr   'torch.ao.nn.intrinsic.quantized.dynamicdynamicr   torch.ao.nn.qatr   torch.ao.nn.qat.dynamicr   torch.ao.nn.quantizedrp   torch.ao.nn.quantized.dynamicr   1torch.ao.quantization.fx._lower_to_native_backendquantizationfxr   +torch.ao.quantization.quantization_mappingsr   torch.nntorch.nn.functionalri   r   $torch.ao.quantization.backend_configr   ns_typesr   opsrr   dictr   r   r   r   r   r   r    r   r   <module>r     s9    %  # # * * . . 7 7 7   ( ( # # , , T T K K    J & iik,d3<L8M3M.N k,\&*34D0E+E&F c]>&*34D0E+E&F>> )*> 
	>(fd34D0E+E&F fR'4S1A-B(B#C 'r   