
    h{                         S SK r S SKrS SKrS SKrS SKJrJr  / r " S S\R                  5      r	\
S:X  a  \ R                  " 5         gg)    N)TensorProtohelperc                      \ rS rSrS rSS jrSS jrSS jrSS jrSS	 jr	SS
 jr
SS jrSS jrSS jrSS jrSS jrSS jrSS jrSS jrSS jrSS jrSS jrSS jrSS jrSS jrSS jrSS jrSS jrSS jrSS jrSS jrSS jrSS  jr SS! jr!SS" jr"SS# jr#SS$ jr$SS% jr%SS& jr&SS' jr'SS( jr(SS) jr)SS* jr*SS+ jr+SS, jr,SS- jr-SS. jr.SS/ jr/SS0 jr0SS1 jr1SS2 jr2SS3 jr3SS4 jr4SS5 jr5SS6 jr6SS7 jr7SS8 jr8SS9 jr9SS: jr:SS; jr;SS< jr<SS= jr=SS> jr>SS? jr?SS@ jr@SSA jrASSB jrBSSC jrCSSD jrDSSE jrESSF jrFSSG jrGSSH jrHSSI jrISSJ jrJSSK jrKSSL jrLSSM jrMSSN jrNSSO jrOSSP jrPSSQ jrQSSR jrRSSS jrSSST jrTSSU jrUSSV jrVSSW jrWSSX jrXSSY jrYSSZ jrZSS[ jr[SS\ jr\SS] jr]SS^ jr^SS_ jr_SS` jr`SSa jraSSb jrbSSc jrcSSd jrdSSe jreSSf jrfSSg jrgSSh jrhSSi jriSSj jrjSSk jrkSSl jrlSSm jrmSSn jrnSSo jroSSp jrpSSq jrqSSr jrrSSs jrsSSt jrtSSu jruSSv jrvSSw jrwSSx jrxSSy jrySSz jrzSS{ jr{SS| jr|SS} jr}SS~ jr~SS jrSS jrSS jrSS jrSS jrSS jrSS jrSS jrSS jrSS jrSS jrSS jrSS jrSS jrSS jrSS jrSS jrSS jrSS jrSS jrSS jrSS jrSS jrSS jrSS jrSS jrSS jrSS jrSS jrSS jrSS jrSS jrSS jrSS jrSS jrSS jrSS jrSS jrSS jrSS jrSS jrSS jrSS jrSS jrSS jrSS jrSS jrSS jrSS jrSS jrSS jrSS jrSS jrSS jrSS jrSS jrSS jrSS jrSS jrSS jrSS jrSS jrSS jrSS jrSS jrSS jrSS jrSS jrSS jrSS jrSS jrSS jrSS jrSS jrSS jrSS jrSS jrSS jrSS jrSS jrSS jrSS jrS\SS4S jrSS jrSS jrSS jrSS jrS\SS4S jrSS jrSS jrSS jrSS jrSS jrSS jrSS jrSS jrSS jrSS jrSS jrSrg)TestAutomaticUpgrade   c                 b    [         R                  U5        U R                  " U/UQ70 UDSS0D6  g )N
is_upgradeT)
tested_opsappend_test_op_conversion)selfopargskwargss       l/var/www/fran/franai/venv/lib/python3.13/site-packages/onnx/test/version_converter/automatic_upgrade_test.py_test_op_upgrade%TestAutomaticUpgrade._test_op_upgrade   s-    "  FdFfFF    returnNc                 ,    U R                  SSSS/0S9  g )NAbs   consumed_inputsr   attrsr   r   s    r   test_AbsTestAutomaticUpgrade.test_Abs       eQ/@1#.FGr   c                 (    U R                  SS5        g )NAcosh	   r   r   s    r   
test_AcoshTestAutomaticUpgrade.test_Acosh       gq)r   c                 (    U R                  SS5        g )NAcos   r   r   s    r   	test_AcosTestAutomaticUpgrade.test_Acos       fa(r   c                     U R                  SSSS/SS//SS//[        R                  [        R                  /[        R                  /5        g )NAndr)         r   r   BOOLr   s    r   test_AndTestAutomaticUpgrade.test_And"   O    VaVVH{//0	
r   c                 (    U R                  SS5        g )NAsinhr#   r   r   s    r   
test_AsinhTestAutomaticUpgrade.test_Asinh-   r&   r   c                 (    U R                  SS5        g )NAtanhr#   r   r   s    r   
test_AtanhTestAutomaticUpgrade.test_Atanh0   r&   r   c                 :    U R                  SS/ SQ/ SQ/SS/0S9  g )NAddr   r0         r   r   r   r   r   s    r   
test_Add_1TestAutomaticUpgrade.test_Add_13   +    1y),5F4L 	 	
r   c                 :    U R                  SS/ SQS//S/SS.S9  g )Nr?   r   r@   rB   r   )r   	broadcastr   r   r   s    r   
test_Add_2TestAutomaticUpgrade.test_Add_28   s-    1y1#&1#TU.V 	 	
r   c           	      <    U R                  SS/ SQS//S/SSS.S9  g )Nr?   r   r@   r0   r   )r   rG   axisr   r   r   s    r   
test_Add_3TestAutomaticUpgrade.test_Add_3=   s1    '(c1E	 	 	
r   c           	      J    Su  pp4U R                  SSUSS/S//XUS//5        g )N)r/   r0   rB      
AffineGridr   r/   r0   rA   r   )r   N_HWs        r   test_AffineGrid_2D'TestAutomaticUpgrade.test_AffineGrid_2DE   s4    
alB!QQC0@A!Q<.Qr   c           
      L    Su  pp4nU R                  SSUSS/S//XXES//5        g )N)r/   r0   rA   rB   rO   rP   r   r0   rA   rB   r   )r   rQ   rR   DrS   rT   s         r   test_AffineGrid_3D'TestAutomaticUpgrade.test_AffineGrid_3DI   s8    %aAlB!QQC0@A!PQ?BSTr   c                 T    U R                  SS/ SQ// SQ/[        R                  /S9  g )NArgMaxr)   r/   r0   rA   r   r0   rA   output_typesr   r   INT64r   s    r   test_ArgMax_1"TestAutomaticUpgrade.test_ArgMax_1M   -    a)ykARAR@S 	 	
r   c           	      Z    U R                  SS/ SQ// SQ/[        R                  /SS0S9  g )Nr\   r)   r]   r/   r   rA   rK   r   r`   r   ra   r   s    r   test_ArgMax_2"TestAutomaticUpgrade.test_ArgMax_2R   8    KK%++,1+ 	 	
r   c                 T    U R                  SS/ SQ// SQ/[        R                  /S9  g )NArgMinr)   r]   r^   r_   ra   r   s    r   test_ArgMin_1"TestAutomaticUpgrade.test_ArgMin_1\   re   r   c           	      Z    U R                  SS/ SQ// SQ/[        R                  /SS0S9  g )Nrm   r)   r]   rg   rK   r   rh   ra   r   s    r   test_ArgMin_2"TestAutomaticUpgrade.test_ArgMin_2a   rk   r   c                 (    U R                  SS5        g )NAsinr)   r   r   s    r   	test_AsinTestAutomaticUpgrade.test_Asink   r,   r   c                 (    U R                  SS5        g )NAtanr)   r   r   s    r   	test_AtanTestAutomaticUpgrade.test_Atann   r,   r   c           	      >    U R                  SS/ SQ// SQ/SSS/0S9  g )NAveragePoolr   r   r   rB   rB   r   r   rA   rA   kernel_shaper/   r   r   r   s    r   test_AveragePool%TestAutomaticUpgrade.test_AveragePoolq   s1    NN!Aq6* 	 	
r   c                 (    U R                  SS5        g )N	Bernoulli   r   r   s    r   test_Bernoulli#TestAutomaticUpgrade.test_Bernoulliz       k2.r   c           
          U R                  SSSS/SS//SS//[        R                  [        R                  /[        R                  /SS0S9  g )NBitShift   r/   r0   	directionRIGHTr   r   r   UINT8r   s    r   test_BitShift"TestAutomaticUpgrade.test_BitShift}   s\    VaVVH 1 12( 	 	
r   c           
      R    U R                  SSSS/S/S/S/S//SS//SS/SSS.S9  g )NBatchNormalizationr   r0   )r   is_testspatialr   r   r   s    r   test_BatchNormalization_1.TestAutomaticUpgrade.test_BatchNormalization_1   sL     VaS1#sQC(VH'(!fqI 	 	
r   c           	      R    U R                  SSSS/S/S/S/S//SS/S/S//SS0S9  g )Nr      r   r0   training_moder   r   r   s    r   test_BatchNormalization_2.TestAutomaticUpgrade.test_BatchNormalization_2   sO     VaS1#sQC(VaS1#"A& 	 	
r   c           	      Z    U R                  SSSS//SS//[        R                  /SS0S9  g )NCastrO   r/   r0   tor   r   ra   r   s    r   	test_CastTestAutomaticUpgrade.test_Cast   s;    AAx1a&K,=,=+>tQi 	 	
r   c                 ,    U R                  SSSS/0S9  g )NCeilr   r   r   r   r   r   s    r   	test_CeilTestAutomaticUpgrade.test_Ceil       fa0AA3/GHr   c                 (    U R                  SS5        g )NCelu   r   r   s    r   	test_CeluTestAutomaticUpgrade.test_Celu       fb)r   c                 ,    U R                  SSSS/0S9  g )NClipr   r   r   r   r   r   s    r   test_Clip_1 TestAutomaticUpgrade.test_Clip_1   r   r   c                 .    U R                  SSS/SS.S9  g )Nr   r   r   ffffff)r   minr   r   r   s    r   test_Clip_2 TestAutomaticUpgrade.test_Clip_2   s    faA3t/TUr   c                 .    U R                  SSS/SS.S9  g )Nr   r   r   @)r   maxr   r   r   s    r   test_Clip_3 TestAutomaticUpgrade.test_Clip_3   s    faA3s/STr   c                 0    U R                  SSS/SSS.S9  g )Nr   r   r   r   r   )r   r   r   r   r   r   s    r   test_Clip_4 TestAutomaticUpgrade.test_Clip_4   s%    A!T#N 	 	
r   c                 @    U R                  SS/ SQS/S/// SQ/5        g )NCol2Im   )r   rB   rB   r/   r}   r   r   s    r   test_Col2Im_4D#TestAutomaticUpgrade.test_Col2Im_4D   s#    hYaS,AL>Rr   c                 @    U R                  SS/ SQS/S/// SQ/5        g )Nr   r   )r   
   r   r0   )r   r/   r0   rA   rB   r   r   s    r   test_Col2Im_5D#TestAutomaticUpgrade.test_Col2Im_5D   s$    h[1#s,CoEVWr   c                     U R                  SSSS/S//S//[        R                  [        R                  /[        R                  /5        g )NCompressr#   rO   r)   r0   )r   r   FLOATr2   r   s    r   test_Compress"TestAutomaticUpgrade.test_Compress   sJ    VaSMSE 0 01	
r   c                 >    U R                  SSSS/SS//SS//5        g )NConcatr   r/   r0   rA   r)   r   r   s    r   test_Concat TestAutomaticUpgrade.test_Concat   )    hQFQF+;q!fXFr   c           
         [         R                  " S[        R                  / SQ[        R
                  R                  SSS5      R                  [        R                  5      R                  5       SS9nU R                  SS	/ S
U0S9  g )NValuer@   r0   rA   rB   TdimsvalsrawConstantr   valuer   r   make_tensorr   r   nprandomrandastypefloat32tobytesr   )r   r   s     r   test_constant"TestAutomaticUpgrade.test_constant   sn    ""1a(//

;CCE
 	j!R7GHr   c                 .    U R                  SSS//5        g )NConstantOfShaper#   r0   r   r   s    r   test_ConstantOfShape)TestAutomaticUpgrade.test_ConstantOfShape   s    /aSE:r   c                 B    U R                  SS/ SQ/ SQS/// SQ/5        g NConvr   r   r0   rB   rB   rA   r0   r/   r/   rA   r   rA   rA   rA   r   r   s    r   test_Conv_1 TestAutomaticUpgrade.test_Conv_1   $    AlQC8<.	
r   c                 B    U R                  SS/ SQ/ SQS/// SQ/5        g r   r   r   s    r   test_Conv_2 TestAutomaticUpgrade.test_Conv_2   r   r   c                 V    U R                  SS/ SQ/ SQS/// SQ/SS/S/ S	QSS/S
.S9  g )Nr   r   r   )rA   r   r/   r/   rA   )r   rA   r0   r)   r/   r0   )r   r   r/   r0   )	dilationsgrouppadsstridesr   r   r   s    r   test_Conv_3 TestAutomaticUpgrade.test_Conv_3   sE    <!-NV$q6	 	 	
r   c           	          U R                  SS/ SQ/ SQS/// SQ/[        R                  [        R                  [        R                  /[        R                  /5        g )NConvIntegerr   r   r   rA   r   )r   r   r   INT32r   s    r   test_Convinteger%TestAutomaticUpgrade.test_Convinteger   sO    <!-N 1 1;3D3DE	
r   c                 >    U R                  SS/ SQ/ SQ// SQ/5        g )NConvTransposer   r}   r   r   r0   r0   )r   r   r)   r)   r   r   s    r   test_ConvTranspose'TestAutomaticUpgrade.test_ConvTranspose   s     Q| <|n	
r   c                 D    U R                  SS/ SQ/ SQ/ SQ// SQ/5        g )N
DeformConv   r   )r   r   r/   r/   )r      r/   r/   r   r   s    r   test_DeformConv$TestAutomaticUpgrade.test_DeformConv   s%    <6N		
r   c                 (    U R                  SS5        g )NCoshr#   r   r   s    r   	test_CoshTestAutomaticUpgrade.test_Cosh  r,   r   c                 (    U R                  SS5        g )NCosr)   r   r   s    r   test_CosTestAutomaticUpgrade.test_Cos      eQ'r   c                 x    U R                  SS/ SQ/ // SQ/[        R                  [        R                  /5        g )NCumSumr   r@   r   r   r   rb   r   s    r   test_Cumsum TestAutomaticUpgrade.test_Cumsum	  s6    OK 1 12	
r   c                 :    U R                  SS/ SQ// SQ/SS0S9  g )NDepthToSpacer   )r   r  r0   r0   )r   r/   rO   rO   	blocksizer/   r   r   r   s    r   test_DepthToSpace&TestAutomaticUpgrade.test_DepthToSpace  s*    A~~kSTEU 	 	
r   c           	          U R                  SSSS// / /SS//[        R                  [        R                  [        R                  /5        g )NDequantizeLinearr   r/   r0   )r   r   INT8r   r   s    r   test_DequantizeLinear*TestAutomaticUpgrade.test_DequantizeLinear  sI    VRVH{00+2B2BC	
r   c                 6    U R                  SS/ SQ/S//5        g )NDetr   )r0   rB   rB   r0   r   r   s    r   
test_Det_1TestAutomaticUpgrade.test_Det_1   s    eR)se<r   c                 4    U R                  SSSS/// /5        g )Nr"  r   rB   r   r   s    r   
test_Det_2TestAutomaticUpgrade.test_Det_2#  s    eR1a&B48r   c           	          U R                  SS/ SQ// SQ/ / /[        R                  [        R                  [        R                  /S9  g )NDynamicQuantizeLinearr   r@   r_   )r   r   r   r   r   s    r   test_DynamicQuantizeLinear/TestAutomaticUpgrade.test_DynamicQuantizeLinear&  sF    #KB%++[->->@Q@QR 	 	
r   c                 :    U R                  SS/ SQ/ SQ/SS/0S9  g )NDivr   r@   )r0   r   rB   r   r   r   r   r   s    r   test_DivTestAutomaticUpgrade.test_Div/  rE   r   c                 .    U R                  SSS/SS.S9  g )NDropoutr   r   )r   r   r   r   r   s    r   test_Dropout!TestAutomaticUpgrade.test_Dropout4  s#    qQCA F 	 	
r   c                 @    U R                  SS/ SQ/ SQ// SQ/SS0S9  g )	NEinsumr   r@   r0   rB   rO   )r0   rA   rO   equationzbij, bjk -> bikr   r   r   s    r   test_Einsum_1"TestAutomaticUpgrade.test_Einsum_19  s1    	"K01 	 	
r   c                 :    U R                  SSSS//SS//SS0S9  g )Nr5  r   rA   rB   r7  zij->jir   r   r   s    r   test_Einsum_2"TestAutomaticUpgrade.test_Einsum_2B  s1    bAq6(aVHZ4J 	 	
r   c                 ,    U R                  SSSS/0S9  g )NElur   r   r   r   r   r   s    r   test_EluTestAutomaticUpgrade.test_EluG  r    r   c                 Z    U R                  SSSS/SS//SS//[        R                  /S9  g )NEqualr)   r/   r0   r_   r1   r   s    r   
test_EqualTestAutomaticUpgrade.test_EqualJ  s<    Q!Q!Q(Aq6(+BRBRAS 	 	
r   c                 (    U R                  SS5        g )NErfr#   r   r   s    r   test_ErfTestAutomaticUpgrade.test_ErfP  r  r   c                 ,    U R                  SSSS/0S9  g )NExpr   r   r   r   r   r   s    r   test_ExpTestAutomaticUpgrade.test_ExpS  r    r   c           	          [         R                  " S[        R                  S/[        R
                  " / SQ5      S9nU R                  SS/ SQS/// SQ/[        R                  [        R                  /U/S9  g )	NbrA   )rB   r/   rO   rA   r   r   Expandr  rg   initializerr   r   r   rb   r   arrayr   r   )r   shapes     r   test_Expand TestAutomaticUpgrade.test_ExpandV  so    """"!288L3I
 	N 1 12 	 	
r   c                 8    U R                  SSSS//SS//5        g )NEyeLiker#   rA   rB   r   r   s    r   test_EyeLike!TestAutomaticUpgrade.test_EyeLikec  s"    iaVH1vh?r   c                 :    U R                  SS/ SQ/SS//SS0S9  g )NFlattenr   r@   r0   r   rK   r   r   r   s    r   test_Flatten!TestAutomaticUpgrade.test_Flattenf  s(    iYK1b'6ST+Vr   c                 ,    U R                  SSSS/0S9  g )NFloorr   r   r   r   r   r   s    r   
test_FloorTestAutomaticUpgrade.test_Floori  s    gq1BQC0HIr   c                 |    U R                  SS/ SQSS/// SQ/[        R                  [        R                  /5        g )NGatherr   r@   rO   r)   )rO   r)   rA   rB   r  r   s    r   test_Gather TestAutomaticUpgrade.test_Gatherl  s;    AN 1 12	
r   c                 |    U R                  SS/ SQSS//SS//[        R                  [        R                  /5        g )NGatherElementsr   r@   rO   r)   r  r   s    r   test_GatherElements(TestAutomaticUpgrade.test_GatherElementsu  s?    AVH 1 12	
r   c                 >    U R                  SS/ SQ/ SQ/SS//5        g )NGatherNDr   r   r/   r0   r   r/   r   r   s    r   test_GatherND"TestAutomaticUpgrade.test_GatherND~  s!    j"y).D1vhOr   c                 *    U R                  SSSS0S9  g )NGelur   approximatetanhr   r   r   s    r   test_Gelu_approximate_tanh/TestAutomaticUpgrade.test_Gelu_approximate_tanh  s    fb0GHr   c                 (    U R                  SS5        g )Nrr  r   r   r   s    r   	test_GeluTestAutomaticUpgrade.test_Gelu  r   r   c                 B    U R                  SSSS/SS/S//SS//5        g )NGemmr   rB   rA   r0   r   r   s    r   	test_GemmTestAutomaticUpgrade.test_Gemm  s-    fa1a&1a&1#)>!QIr   c                 8    U R                  SS/ SQ// SQ/5        g )NGlobalAveragePoolr   r   r0   r   r   r   r0   r   r   r   r   s    r   test_GlobalAveragePool+TestAutomaticUpgrade.test_GlobalAveragePool  s    11~6FWr   c                 8    U R                  SS/ SQ// SQ/5        g )NGlobalMaxPoolr   r  r  r   r   s    r   test_GlobalMaxPool'TestAutomaticUpgrade.test_GlobalMaxPool  s    oq>2B\NSr   c                 8    U R                  SS/ SQ// SQ/5        g )NGlobalLpPoolr/   r  r  r   r   s    r   test_GlobalLpPool&TestAutomaticUpgrade.test_GlobalLpPool  s    na.1AL>Rr   c                 Z    U R                  SSSS/SS//SS//[        R                  /S9  g )NGreaterr)   r/   r0   r_   r1   r   s    r   test_Greater!TestAutomaticUpgrade.test_Greater  s<    qAq6Aq6*aVHKDTDTCU 	 	
r   c                 Z    U R                  SSSS/SS//SS//[        R                  /S9  g )NGreaterOrEqualr   r/   r0   r_   r1   r   s    r   test_GreaterOrEqual(TestAutomaticUpgrade.test_GreaterOrEqual  s@    VaVVH%**+ 	 	
r   c                     U R                  SS/ SQ/ SQ// SQ/[        R                  [        R                  /[        R                  /SSSS.S	9  g )
N
GridSample   r   )r   r0   r0   r/   nearestborderr   )modepadding_modealign_cornersinput_typesr`   r   r   r   r   r   s    r   test_GridSample$TestAutomaticUpgrade.test_GridSample  sS    <(N$**K,=,=>%++,$hQRS 	 	
r   c                 L    U R                  SS/ SQ/ SQ/ SQ// SQ/ SQ/SS0S	9  g )
NGRUr)   rB   r0   rA   r   r   rA   rB   r   r0   rO   r   r0   rO   hidden_sizerO   r   r   r   s    r   
test_GRU_1TestAutomaticUpgrade.test_GRU_1  s5    
J/9% !$ 	 	
r   c           	      N    U R                  SS/ SQ/ SQ/ SQ// SQ/ SQ/SSS	.S
9  g )Nr  r)   r  )r/   r   rA   rB   r/   r0   rO   r/   r0   rO   rO   bidirectionalr  r   r   r   r   s    r   
test_GRU_2TestAutomaticUpgrade.test_GRU_2  s5    
J/9%"#/B 	 	
r   c                    U R                  SS/ SQ/ SQ/ SQSS/S// SQ// S	Q/ S
Q/[        R                  [        R                  [        R                  [        R                  [        R                  [        R                  /SS0S9  g )Nr  r)   r  r  r      rB   r   rB   rO   r  r  r  rO   r   r  r   s    r   
test_GRU_3TestAutomaticUpgrade.test_GRU_3  s~    
JB!iH9%!!!!!!!!!!!! !!$ 	 	
r   c                 ,    U R                  SSSS/0S9  g )NHardSigmoidr   r   r   r   r   r   s    r   test_HardSigmoid%TestAutomaticUpgrade.test_HardSigmoid  s    mQ7H1#6NOr   c                 (    U R                  SS5        g )N	HardSwishr   r   r   s    r   test_HardSwish#TestAutomaticUpgrade.test_HardSwish  r   r   c                 (    U R                  SS5        g )NHardmaxr   r   r   s    r   test_Hardmax!TestAutomaticUpgrade.test_Hardmax  s    i+r   c                 (    U R                  SS5        g )NIdentityr   r   r   s    r   test_Identity"TestAutomaticUpgrade.test_Identity      j!,r   c           
      >   [         R                  " S[        R                  / SQ5      /n[         R                  " S[        R                  / SQ[
        R                  R                  SSS5      R                  [
        R                  5      R                  5       SS9n[         R                  " S	/ S/US
9n[         R                  " U/S/ U/ 5      n[         R                  " S[        R                  / SQ[
        R                  R                  SSS5      R                  [
        R                  5      R                  5       SS9n[         R                  " S	/ S/US
9n[         R                  " U/S/ U/ 5      nU R                  SSS/// SQ/[        R                  /XGS.S9  g )Noutr@   r   r0   rA   rB   Tr   r   )r   
then_graph
else_graphIfr   r   )then_branchelse_branchr   )r   make_tensor_value_infor   r   r   r   r   r   r   r   r   	make_node
make_graphr   r2   )r   
sub_outputthen_tensor	then_noder  else_tensor	else_noder  s           r   test_IfTestAutomaticUpgrade.test_If  s`   ))%1B1BIN

 ((1a(//

;CCE
 $$ZeWKP	&&	{L"jRTU
((1a(//

;CCE
 $$ZeWKP	&&	{L"jRTU
SEK",H 	 	
r   c           	      r    U R                  SSS /// SQ/[        R                  /[        R                  /S9  g )NImageDecoderr   )NNr0   r  r`   r   r   s    r   test_ImageDecoder&TestAutomaticUpgrade.test_ImageDecoder  s@    VH$**+%++, 	 	
r   c                 D    U R                  SSSS/S/S//SS//SS/0S9  g )NInstanceNormalizationr   r0   r   r   r   r   r   s    r   test_InstanceNormalization/TestAutomaticUpgrade.test_InstanceNormalization	  s@    #VaS1#VH$qc* 	 	
r   c                 T    U R                  SSSS//SS//[        R                  /S9  g )NIsInfr   r/   r0   r_   r1   r   s    r   
test_IsInfTestAutomaticUpgrade.test_IsInf  s5    R1a&QF8;;K;K:L 	 	
r   c                 T    U R                  SSSS//SS//[        R                  /S9  g )NIsNaNr#   r/   r0   r_   r1   r   s    r   
test_IsNaNTestAutomaticUpgrade.test_IsNaN  s5    Q!QAq6(+:J:J9K 	 	
r   c                 ,    U R                  SSSS/0S9  g )N	LeakyRelur   r   r   r   r   r   s    r   test_LeakyRelu#TestAutomaticUpgrade.test_LeakyRelu  s    k15F4LMr   c                 Z    U R                  SSSS/SS//SS//[        R                  /S9  g )NLessr)   r/   r0   r_   r1   r   s    r   	test_LessTestAutomaticUpgrade.test_Less  s<    AAA'1a&AQAQ@R 	 	
r   c                 Z    U R                  SSSS/SS//SS//[        R                  /S9  g )NLessOrEqualr   r/   r0   r_   r1   r   s    r   test_LessOrEqual%TestAutomaticUpgrade.test_LessOrEqual%  s@    VaVVH%**+ 	 	
r   c                 ,    U R                  SSSS/0S9  g )NLogr   r   r   r   r   r   s    r   test_LogTestAutomaticUpgrade.test_Log.  r    r   c                 (    U R                  SS5        g )N
LogSoftmaxr   r   r   s    r   test_LogSoftmax$TestAutomaticUpgrade.test_LogSoftmax1  s    lA.r   c                    [         R                  R                  S[         R                  R                  / 5      n[         R                  R                  S[         R                  R
                  / 5      n[         R                  R                  S[         R                  R                  S/5      n[         R                  R                  S[         R                  R
                  / 5      n[         R                  R                  S[         R                  R                  S/5      n[         R                  R                  S[         R                  R                  S/5      n[         R                  R                  S/ S	/[         R                  R                  S
[         R                  R                  S/[        R                  " S/5      R                  [        R                  5      R                  [        5      S9S9n[         R                  R                  SSS	/S/S9n[         R                  R                  SS/S/S9n	[         R                  R                  SS/S/S9n
[         R                  R                  XxX/SXU/XEU/5      nU R                  SS/ SS//S/SS//[        R                  [        R
                  [        R                  /SU0S9  g )N
iter_countcond_inx_inr   cond_outx_outx_scanr   oner   )name	data_typer   r   )inputsoutputsr   r?   r  r  r  	loop_bodyLoop rB   bodyr   )onnxr   r  r   rb   r2   r   r  r   r   rT  r   r   floatr  r   )r   r  r  r   r  r  r  constaddid_1id_2r
  s               r   test_Loop_1 TestAutomaticUpgrade.test_Loop_14  s_   [[77$**00"

 ++44t'',,b
 {{11&$:J:J:P:PSTRUV;;55((--r
 227D<L<L<R<RUVTWX33d&&,,qc
 %%G++))**00SXXqc]))"**5<<UC	 * 	 & 

 kk##E65/G9#U{{$$Z	H:$V{{$$	{ZL % 
 KK**$$'f%	
	 	aSMS1a&M 0 0+2C2CD9% 	 	
r   c           	         [         R                  R                  S[         R                  R                  / 5      n[         R                  R                  S[         R                  R
                  / 5      n[         R                  R                  S[         R                  R                  SS/5      n[         R                  R                  S[         R                  R
                  / 5      n[         R                  R                  S[         R                  R                  SS/5      n[         R                  R                  SS/S	/S/S
9n[         R                  R                  SS	/S/S/S
9n[         R                  R                  SS/S/S9n[         R                  R                  XgU/SXU/XE/5      n	U R                  SS/ SSS//SS//[        R                  [        R
                  [        R                  /SU	0S9  g )Nr  r  r   r/   r   r  r  Squeezesqueeze_out)r  r  axes	Unsqueezer  r	  r
  r  r   r  r  r   )
r  r   r  r   rb   r2   r   r  r  r   )
r   r  r  r   r  r  squeeze	unsqueezeidentityr
  s
             r   test_Loop_2 TestAutomaticUpgrade.test_Loop_2b  s   [[77$**00"

 ++44t'',,b
 {{11D$$**QF
 ;;55((--r
 22T%%++aV
 ++''vhqc ( 
 KK))'! * 
	 ;;((	{ZL ) 
 KK***$'	
	 	aVVH 0 0+2C2CD9% 	 	
r   c                 (    U R                  SS5        g )NLpNormalizationr   r   r   s    r   test_LpNormalization)TestAutomaticUpgrade.test_LpNormalization  s    /3r   c           	      >    U R                  SS/ SQ// SQ/SSS/0S9  g )NLpPoolr/   r}   r~   r   r   r   r   s    r   test_LpPool TestAutomaticUpgrade.test_LpPool  s0    a,,QRTUPV?W 	 	
r   c                 *    U R                  SSSS0S9  g )NLRNr   sizer0   r   r   r   s    r   
test_LRN_1TestAutomaticUpgrade.test_LRN_1  s    eQvqk:r   c                 :    U R                  SS/ SQ// SQ/SS0S9  g )Nr)  r   r/   r0   rA   rB   r*  r0   r   r   r   s    r   
test_LRN_2TestAutomaticUpgrade.test_LRN_2  s(    1|n|nVQK 	 	
r   c           	      R    U R                  SS/ SQ/ SQ/ SQ// SQ/ SQ/ SQ/SS0S	9  g )
NLSTMr)   r  r   r  rA   r  r  r  rO   r   r   r   s    r   test_LSTM_1 TestAutomaticUpgrade.test_LSTM_1  s7    
J/9i0 !$ 	 	
r   c           	      T    U R                  SS/ SQ/ SQ/ SQ// SQ/ SQ/ SQ/SSS	.S
9  g )Nr2  r)   r  )r/   r  rA   r  r  rO   r  r  r   r   r   s    r   test_LSTM_2 TestAutomaticUpgrade.test_LSTM_2  s7    
J/9i0"#/B 	 	
r   c                 `   U R                  SS/ SQ/ SQ/ SQSS/S// SQ/ SQSS	/// S
Q/ SQ/ SQ/[        R                  [        R                  [        R                  [        R                  [        R                  [        R                  [        R                  [        R                  /SS0S9  g )Nr2  r)   r  r3  r   0   rB   r  r   r  r  r  rO   r   r  r   s    r   test_LSTM_3 TestAutomaticUpgrade.test_LSTM_3  s    BB	 9i0!!!!!!!!!!!!!!!!	 !!$1 	 	
r   c                 >    U R                  SSSS/SS//SS//5        g )NMatMulr   r/   r0   rA   r   r   s    r   test_MatMul_1"TestAutomaticUpgrade.test_MatMul_1  r   r   c                 >    U R                  SS/ SQ/ SQ// SQ/5        g )Nr>  r   )rB   r/   r0   r  )rB   r/   rA   r   r   s    r   test_MatMul_2"TestAutomaticUpgrade.test_MatMul_2  s    hIy+AI;Or   c                     U R                  SSSS/SS//SS//[        R                  [        R                  /[        R                  /5        g NMatMulIntegerr   r/   r0   rA   r   r   r  r   r   s    r   test_MatMulInteger_1)TestAutomaticUpgrade.test_MatMulInteger_1  sO    VaVVH{//0	
r   c           
          U R                  SSSS/SS// / /SS//[        R                  [        R                  [        R                  [        R                  /[        R                  /5        g rE  rG  r   s    r   test_MatMulInteger_2)TestAutomaticUpgrade.test_MatMulInteger_2  sc    VaVR$VH{//1A1A;CSCST	
r   c           
          U R                  SSSS/SS/S/S//SS//[        R                  [        R                  [        R                  [        R                  /[        R                  /5        g rE  rG  r   s    r   test_MatMulInteger_3)TestAutomaticUpgrade.test_MatMulInteger_3  sg    VaVaS1#&VH{//1A1A;CSCST	
r   c                 B    U R                  SS/ SQ/ SQ// SQ/SS/0S9  g )NMaxr   r]   r   r   r   r   r   s    r   test_MaxTestAutomaticUpgrade.test_Max  2    	"K$qc* 	 	
r   c           	      >    U R                  SS/ SQ// SQ/SSS/0S9  g )NMaxPoolr   r}   r~   r   r/   r   r   r   s    r   test_MaxPool_1#TestAutomaticUpgrade.test_MaxPool_1  s0    q<.<.RSUVQW@X 	 	
r   c           
          U R                  SS/ SQ// SQ/ SQ/[        R                  [        R                  /SSS/0S9  g )NrV  r  r}   r~   r   r/   rh   r  r   s    r   test_MaxPool_2#TestAutomaticUpgrade.test_MaxPool_2  sH    N<(%++[->->?!Aq6* 	 	
r   c           	      D    U R                  SS/ SQSS/// SQ/SSS/0S	9  g )
N
MaxRoiPoolr   r/   r0   r   r   rA   rB   )rA   r0   r0   r0   pooled_shaper0   r   r   r   s    r   test_MaxRoiPool$TestAutomaticUpgrade.test_MaxRoiPool  s8    aV$N!Aq6* 	 	
r   c           
          U R                  SS/ SQ/ SQ// SQ/[        R                  [        R                  /SSS/0S9  g )N	MaxUnpoolr#   r}   )r   r   rO   rO   r   r/   r   r  r   s    r   test_MaxUnpool#TestAutomaticUpgrade.test_MaxUnpool  sH    <(N 1 12!Aq6* 	 	
r   c                 B    U R                  SS/ SQ/ SQ// SQ/SS/0S9  g )NMeanr   r]   r   r   r   r   r   s    r   	test_MeanTestAutomaticUpgrade.test_Mean  s2    	"K$qc* 	 	
r   c                 .    U R                  SSSSS/0S9  g )NMeanVarianceNormalizationr#   r  r   r/   r   r   r   s    r   test_MeanVarianceNormalization3TestAutomaticUpgrade.test_MeanVarianceNormalization$  s"    91VaQRVDTUr   c                 B    U R                  SS/ SQ/ SQ// SQ/SS/0S9  g )NMinr   r]   r   r   r   r   r   s    r   test_MinTestAutomaticUpgrade.test_Min'  rT  r   c                 (    U R                  SS5        g )NMishr   r   r   s    r   	test_MishTestAutomaticUpgrade.test_Mish0  r   r   c                 >    U R                  SSSS/SS//SS//5        g )NModr   r/   r0   r   r   s    r   
test_Mod_1TestAutomaticUpgrade.test_Mod_13  s)    eR1a&1a&)9QF8Dr   c                 @    U R                  SSSS/SS//SS//SS0S9  g )Nrw  r   r/   r0   fmodr   r   r   r   s    r   
test_Mod_2TestAutomaticUpgrade.test_Mod_26  s3    eR1a&1a&)9QF8FTU;Wr   c                 B    U R                  SS/ SQ/ SQ// SQ/SS/0S9  g )NMulr   r]   rg   r   r   r   r   r   s    r   test_MulTestAutomaticUpgrade.test_Mul9  rT  r   c           	      Z    U R                  SSSS//SS//[        R                  /SS0S9  g )NMultinomialr)   r0   rB   sample_sizerh   r   r   r   r   s    r   test_Multinomial%TestAutomaticUpgrade.test_MultinomialB  sA    VHVH%++, !$ 	 	
r   c                 ,    U R                  SSSS/0S9  g )NNegr   r   r   r   r   r   s    r   test_NegTestAutomaticUpgrade.test_NegL  r    r   c                 x    U R                  SS/ SQSS/// /[        R                  [        R                  /5        g )NNegativeLogLikelihoodLossr   r@   r0   rB   r  r   s    r    test_NegativeLogLikelihoodLoss_15TestAutomaticUpgrade.test_NegativeLogLikelihoodLoss_1O  s;    'AD 1 12	
r   c           	          U R                  SS/ SQSS/S/// /[        R                  [        R                  [        R                  /5        g )Nr  r   r@   r0   rB   rA   r  r   s    r    test_NegativeLogLikelihoodLoss_25TestAutomaticUpgrade.test_NegativeLogLikelihoodLoss_2X  sG    'A$D 1 1;3D3DE	
r   c                 Z    U R                  SS/ SQ/ SQ/SS//[        R                  /S9  g )NNonMaxSuppressionr   r]   r6  r/   r0   r_   ra   r   s    r   test_NonMaxSuppression+TestAutomaticUpgrade.test_NonMaxSuppressiona  s8    	"VH%++, 	 	
r   c                 T    U R                  SSSS//SS//[        R                  /S9  g )NNonZeror#   r0   r/   rA   r_   ra   r   s    r   test_NonZero!TestAutomaticUpgrade.test_NonZeroj  s5    qAq6(aVHK<M<M;N 	 	
r   c                 x    U R                  SSSS//SS//[        R                  /[        R                  /5        g )NNotr   r/   r0   r1   r   s    r   test_NotTestAutomaticUpgrade.test_Noto  s;    11vh!Q;+;+;*<{?O?O>P	
r   c                 >    U R                  SS/ SQ/ S/// SQ/5        g )NOneHotr#   r@   r/   r0   rA   rB   rO   r   r   s    r   test_OneHot TestAutomaticUpgrade.test_OneHott  s!    hIrA3+?,Pr   c                     U R                  SSSS/SS//SS//[        R                  [        R                  /[        R                  /5        g )NOrr)   r/   r0   r1   r   s    r   test_OrTestAutomaticUpgrade.test_Orw  sO    VaVVH{//0	
r   c           	      @    U R                  SSSS//SS/// SQSS	.S
9  g )NPadr/   r0   rA   rB   r  )r   r/   r   r/         ?)r   r   r   r   r   s    r   test_PadTestAutomaticUpgrade.test_Pad  s2    11vh!QPS0T 	 	
r   c                 >    U R                  SS/ SQ/ SQ// SQ/5        g )NPowr   r]   r   r   s    r   test_PowTestAutomaticUpgrade.test_Pow  s    eQI(>Lr   c                 B    U R                  SS/ SQ/ SQ// SQ/SS/0S9  g )NPRelur   r]   r   r   r   r   r   s    r   
test_PReluTestAutomaticUpgrade.test_PRelu  s2    	"K$qc* 	 	
r   c                 J    U R                  SS/ SQ/ / / SQ/ / / / // SQ/5        g )NQLinearConvr   r   r   r   r   r   s    r   test_QLinearConv%TestAutomaticUpgrade.test_QLinearConv  s/    2r<RR@N		
r   c                 J    U R                  SSSS// / SS// / / / /SS//5        g )NQLinearMatMulr   r/   r0   rA   r   r   s    r   test_QLinearMatMul'TestAutomaticUpgrade.test_QLinearMatMul  s9    R1a&"b1a&"b"b!IQPQF8	
r   c           	          U R                  SS/ SQ/ / // SQ/[        R                  [        R                  [        R                  /[        R                  /5        g )NQuantizeLinearr   r@   )r   r   r   r   r   s    r   test_QuantizeLinear(TestAutomaticUpgrade.test_QuantizeLinear  sM    BK 1 1;3D3DE	
r   c           	      8    U R                  SS/ / SQ/S/ SQ0S9  g )NRandomNormalr   r@   rU  r   r   r   s    r   test_RandomNormal&TestAutomaticUpgrade.test_RandomNormal  s'    ArI;w	6J 	 	
r   c           
         [         R                  " S[        R                  / SQ[        R
                  R                  SSS5      R                  [        R                  5      R                  5       SS9nU R                  SS	/ SQ// SQ/U/S
9  g )Nar@   r0   rA   rB   Tr   RandomNormalLiker   rQ  r   r   likes     r   test_RandomNormalLike*TestAutomaticUpgrade.test_RandomNormalLike  sv    !!1a(//

;CCE
 	I;$ 	 	
r   c           	      8    U R                  SS/ / SQ/S/ SQ0S9  g )NRandomUniformr   r@   rU  r   r   r   s    r   test_RandomUniform'TestAutomaticUpgrade.test_RandomUniform  s'    QYK7K 	 	
r   c           
         [         R                  " S[        R                  / SQ[        R
                  R                  SSS5      R                  [        R                  5      R                  5       SS9nU R                  SS	/ SQ// SQ/U/S
9  g )Nr  r@   r0   rA   rB   Tr   RandomUniformLiker   rQ  r   r  s     r   test_RandomUniformLike+TestAutomaticUpgrade.test_RandomUniformLike  sv    !!1a(//

;CCE
 	YK)4& 	 	
r   c           	         [         R                  " S[        R                  / [        R
                  " S/5      S9n[         R                  " S[        R                  / [        R
                  " S/5      S9n[         R                  " S[        R                  / [        R
                  " S/5      S9nU R                  SS	/ / / /S
//XU/S9  g )Nr  r   rO  rN  r   cr/   Ranger   rO   rQ  )r   r   r   r   r   rT  r   )r   startendsteps       r   
test_RangeTestAutomaticUpgrade.test_Range  s    ""3(9(9STRUW  k&7&7brxxQSPT~V!!#{'8'8rRSQTVR"b"u5t:L 	 	
r   c                 ,    U R                  SSSS/0S9  g )N
Reciprocalr   r   r   r   r   r   s    r   test_Reciprocal$TestAutomaticUpgrade.test_Reciprocal  s    lA6G!5MNr   c                 8    U R                  SS/ SQ// SQ/5        g )NReduceL1r   r@   r   r   r   r   r   s    r   test_ReduceL1"TestAutomaticUpgrade.test_ReduceL1      j!i[9+Fr   c                 8    U R                  SS/ SQ// SQ/5        g )NReduceL2r   r@   r  r   r   s    r   test_ReduceL2"TestAutomaticUpgrade.test_ReduceL2  r  r   c                 8    U R                  SS/ SQ// SQ/5        g )NReduceLogSumr   r@   r  r   r   s    r   test_ReduceLogSum&TestAutomaticUpgrade.test_ReduceLogSum  s    na)ykJr   c                 8    U R                  SS/ SQ// SQ/5        g )NReduceLogSumExpr   r@   r  r   r   s    r   test_ReduceLogSumExp)TestAutomaticUpgrade.test_ReduceLogSumExp      /YK)Mr   c                 8    U R                  SS/ SQ// SQ/5        g )N
ReduceMeanr   r@   r  r   r   s    r   test_ReduceMean$TestAutomaticUpgrade.test_ReduceMean      lA	{YKHr   c                 8    U R                  SS/ SQ// SQ/5        g )N	ReduceMaxr   r@   r  r   r   s    r   test_ReduceMax#TestAutomaticUpgrade.test_ReduceMax      k1ykI;Gr   c                 8    U R                  SS/ SQ// SQ/5        g )N	ReduceMinr   r@   r  r   r   s    r   test_ReduceMin#TestAutomaticUpgrade.test_ReduceMin  r  r   c                 8    U R                  SS/ SQ// SQ/5        g )N
ReduceProdr   r@   r  r   r   s    r   test_ReduceProd$TestAutomaticUpgrade.test_ReduceProd  r  r   c                 8    U R                  SS/ SQ// SQ/5        g )N	ReduceSumr   r@   r  r   r   s    r   test_ReduceSum#TestAutomaticUpgrade.test_ReduceSum  r  r   c                 8    U R                  SS/ SQ// SQ/5        g )NReduceSumSquarer   r@   r  r   r   s    r   test_ReduceSumSquare)TestAutomaticUpgrade.test_ReduceSumSquare  r  r   c                 ,    U R                  SSSS/0S9  g )NRelur   r   r   r   r   r   s    r   	test_ReluTestAutomaticUpgrade.test_Relu  r   r   c           	      B    U R                  SS/ SQ// SQ/S// SQS.S9  g )NReshaper   r@   )r0   r   r/   r   )r   rU  r   r   r   s    r   test_Reshape!TestAutomaticUpgrade.test_Reshape  s/    KL'(cJ? 	 	
r   c                 <    U R                  SS/ SQS/// SQ/5        g )NResizer   r@   r0   r0   r  r   r   r   s    r   test_Resize TestAutomaticUpgrade.test_Resize  s    hY,<zlKr   c                 z    U R                  SS/ SQS/// SQ/[        R                  [        R                  /5        g )NReverseSequencer   r@   rA   r  r   s    r   test_ReverseSequence)TestAutomaticUpgrade.test_ReverseSequence  s9    K 1 12	
r   c                 L    U R                  SS/ SQ/ SQ/ SQ// SQ/ SQ/SS0S	9  g )
NRNNr)   r  r   rO   rA   r  r  r  rO   r   r   r   s    r   
test_RNN_1TestAutomaticUpgrade.test_RNN_1  s5    	9-9% !$ 	 	
r   c           	      N    U R                  SS/ SQ/ SQ/ SQ// SQ/ SQ/SSS	.S
9  g )Nr  r)   r  )r/   rO   rA   r  r  rO   r  r  r   r   r   s    r   
test_RNN_2TestAutomaticUpgrade.test_RNN_2  s5    	9-9%"#/B 	 	
r   c                    U R                  SS/ SQ/ SQ/ SQSS/S// SQ// S	Q/ S
Q/[        R                  [        R                  [        R                  [        R                  [        R                  [        R                  /SS0S9  g )Nr  r)   r  r   r   r   rB   r  r  r  r  rO   r   r  r   s    r   
test_RNN_3TestAutomaticUpgrade.test_RNN_3"  s~    	9q"gsIF9%!!!!!!!!!!!! !!$ 	 	
r   c           	          U R                  SS/ SQSS/S/// SQ/[        R                  [        R                  [        R                  /5        g )NRoiAlignr   r^  rA   r   r0   r   r   r  r   s    r   test_RoiAlign_1$TestAutomaticUpgrade.test_RoiAlign_14  sG    b!Wrd+O 1 1;3D3DE	
r   c           	          U R                  SS/ SQSS/S/// SQ/[        R                  [        R                  [        R                  /SS0S	9  g )
Nr*  r  r^  r   rA   r+  coordinate_transformation_mode
half_pixelr   r  r   s    r   test_RoiAlign_2$TestAutomaticUpgrade.test_RoiAlign_2=  sT    b!Wrd+O 1 1;3D3DE3\B 	 	
r   c                 (    U R                  SS5        g )NRoundr   r   r   s    r   
test_RoundTestAutomaticUpgrade.test_RoundG      gr*r   c           	          U R                  SSSS/SS/SS//SS//[        R                  [        R                  [        R                  /[        R                  /5        g )NScatterr#   r/   r0   r   r  r   s    r   test_Scatter!TestAutomaticUpgrade.test_ScatterJ  s]    VaVaV$VH 1 1;3D3DE	
r   c           	          U R                  SSSS/SS/SS//SS//[        R                  [        R                  [        R                  /[        R                  /5        g )NScatterElementsr   r/   r0   r   r  r   s    r   test_ScatterElements_1+TestAutomaticUpgrade.test_ScatterElements_1T  s]    VaVaV$VH 1 1;3D3DE	
r   c           
          U R                  SSSS/SS/SS//SS//[        R                  [        R                  [        R                  /[        R                  /SS0S9  g )	Nr=  r  r/   r0   r   	reductionr  r   r  r   s    r   test_ScatterElements_2+TestAutomaticUpgrade.test_ScatterElements_2^  sj    VaVaV$VH 1 1;3D3DE& 	 	
r   c           	          U R                  SSSS/SS/SS//SS//[        R                  [        R                  [        R                  /[        R                  /5        g )N	ScatterNDr   r/   r0   r   r  r   s    r   test_ScatterND_1%TestAutomaticUpgrade.test_ScatterND_1i  s]    VaVaV$VH 1 1;3D3DE	
r   c           
          U R                  SSSS/SS/SS//SS//[        R                  [        R                  [        R                  /[        R                  /SS0S9  g )	NrE  r  r/   r0   r   rA  mulr   r  r   s    r   test_ScatterND_2%TestAutomaticUpgrade.test_ScatterND_2s  sj    VaVaV$VH 1 1;3D3DE& 	 	
r   c           	         [         R                  R                  S[         R                  R                  S/5      n[         R                  R                  S[         R                  R                  S/5      n[         R                  R                  S[         R                  R                  S/5      n[         R                  R                  S[         R                  R                  S/5      n[         R                  R                  SSS/S/S9n[         R                  R                  SS/S/S9n[         R                  R                  XV/S	X/X4/5      nU R                  S
SSSS// SQ/SS// SQ/USS.S9  g )Nsum_inr/   next_insum_outscan_outr?   r	  r  	scan_bodyScanr  r  r   )r   r0   r/   )r  num_scan_inputsr   )r  r   r  r   r   r  r  r   )r   rM  rN  rO  rP  add_nodeid_noder  s           r   	test_ScanTestAutomaticUpgrade.test_Scan~  s`   33d&&,,qc
 ++44t''--s
 ++44t''--s
 ;;55((..
 ;;((8Y/) ) 
 ++''	{ZL ( 
 {{%%v.?'AT
 	!Q#VYA6 	 	
r   c                 ,    U R                  SSSS/0S9  g )NSelur   r   r   r   r   r   s    r   	test_SeluTestAutomaticUpgrade.test_Selu  r   r   c                 R    U R                  SS/ SQ/S//[        R                  /S9  g )NShaper   r@   r0   r_   ra   r   s    r   
test_ShapeTestAutomaticUpgrade.test_Shape  s/    QqcU+:K:K9L 	 	
r   c                 (    U R                  SS5        g )NShrinkr#   r   r   s    r   test_Shrink TestAutomaticUpgrade.test_Shrink  s    h*r   c                 ,    U R                  SSSS/0S9  g )NSigmoidr   r   r   r   r   r   s    r   test_Sigmoid!TestAutomaticUpgrade.test_Sigmoid  s    i3Dqc2JKr   c                 (    U R                  SS5        g )NSignr#   r   r   s    r   	test_SignTestAutomaticUpgrade.test_Sign  r,   r   c                 (    U R                  SS5        g )NSinhr#   r   r   s    r   	test_SinhTestAutomaticUpgrade.test_Sinh  r,   r   c                 (    U R                  SS5        g )NSinr)   r   r   s    r   test_SinTestAutomaticUpgrade.test_Sin  r  r   c                 P    U R                  SS/ SQ// /[        R                  /S9  g )NSizer   r@   r_   ra   r   s    r   	test_SizeTestAutomaticUpgrade.test_Size  s-    A	{RD8I8I7J 	 	
r   c           
      J    U R                  SS/ SQ// SQ/SS/SS/SS/S.S	9  g )
NSlicer   r@   )r0   r/   r/   r/   r   r0   )r  startsendsr   r   r   s    r   
test_SliceTestAutomaticUpgrade.test_Slice  s;    KKq6aVaVD 	 	
r   c                 *    U R                  SSSS0S9  g )NSoftmaxr   rK   r   r   r   r   s    r   test_Softmax_0#TestAutomaticUpgrade.test_Softmax_0      i61+>r   c                 *    U R                  SSSS0S9  g )Nr  r   rK   r   r   r   s    r   test_Softmax_1#TestAutomaticUpgrade.test_Softmax_1  r  r   c                 *    U R                  SSSS0S9  g )Nr  r   rK   r/   r   r   r   s    r   test_Softmax_2#TestAutomaticUpgrade.test_Softmax_2  r  r   c                 *    U R                  SSSS0S9  g )Nr  r   rK   r   r   r   s    r   test_Softmax_3#TestAutomaticUpgrade.test_Softmax_3      i62,?r   c                 *    U R                  SSSS0S9  g )Nr  r   rK   r   r   r   s    r   test_Softmax_4#TestAutomaticUpgrade.test_Softmax_4  r  r   c                 *    U R                  SSSS0S9  g )Nr  r   rK   r   r   r   s    r   test_Softmax_5#TestAutomaticUpgrade.test_Softmax_5  r  r   c                 (    U R                  SS5        g )NSoftplusr   r   r   s    r   test_Softplus"TestAutomaticUpgrade.test_Softplus  r  r   c                 (    U R                  SS5        g )NSoftsignr   r   r   s    r   test_Softsign"TestAutomaticUpgrade.test_Softsign  r  r   c                 x    U R                  SS/ SQSS/// /[        R                  [        R                  /5        g )NSoftmaxCrossEntropyLossr   r  r0   rO   r  r   s    r   test_SoftmaxCrossEntropyLoss1TestAutomaticUpgrade.test_SoftmaxCrossEntropyLoss  s;    %Aq6"D 1 12	
r   c                 :    U R                  SS/ SQ// SQ/SS0S9  g )NSpaceToDepthr   )r   r0   r  r  )r   r   rA   rA   r  r/   r   r   r   s    r   test_SpaceToDepth&TestAutomaticUpgrade.test_SpaceToDepth  s*    A~{TUFV 	 	
r   c           	      L    U R                  SS/ SQ// SQ/ SQ/ SQ/S/ SQS.S	9  g )
NSplitr/   )r0   rA   r)   r0   rA   r/   )r0   rA   r   )r0   rA   rA   rg   )rK   splitr   r   r   s    r   
test_SplitTestAutomaticUpgrade.test_Split  s2    K	9-y1 	 	
r   c                 ,    U R                  SSSS/0S9  g )NSqrtr   r   r   r   r   r   s    r   	test_SqrtTestAutomaticUpgrade.test_Sqrt  r   r   c                 8    U R                  SS/ SQ// SQ/5        g )Nr  r   )r/   r   r0   rA   r   r]   r   r   s    r   test_Squeeze!TestAutomaticUpgrade.test_Squeeze  s    i_,=	{Kr   c           
      z    U R                  SSSS//SS//[        R                  /[        R                  /SS0S9  g )NStringNormalizerr   r   r0   case_change_actionLOWERr   )r   r   STRINGr   s    r   test_StringNormalizer*TestAutomaticUpgrade.test_StringNormalizer  sM    VHVH  '1 	 	
r   c                 B    U R                  SS/ SQ/ SQ// SQ/SS/0S9  g )NSubr   r]   r   r   r   r   r   s    r   test_SubTestAutomaticUpgrade.test_Sub  rT  r   c                 B    U R                  SS/ SQ/ SQ// SQ/SS/0S9  g )NSumr   r]   r   r   r   r   r   s    r   test_SumTestAutomaticUpgrade.test_Sum  rT  r   c                 ,    U R                  SSSS/0S9  g )NTanhr   r   r   r   r   r   s    r   	test_TanhTestAutomaticUpgrade.test_Tanh  r   r   c                 (    U R                  SS5        g )NTanr)   r   r   s    r   test_TanTestAutomaticUpgrade.test_Tan  r  r   c                 H    U R                  SSS//S//SSSSSS	/SS
/S.S9  g )NTfIdfVectorizerr#   r0   rB   r   r/   TFIDFr   r   rA   )max_gram_lengthmax_skip_countmin_gram_lengthr  ngram_countsngram_indexesr   r   r   s    r   test_TfIdfVectorizer)TestAutomaticUpgrade.test_TfIdfVectorizer  sH    SESE#$"##$!"B"#Q 	 	
r   c                 (    U R                  SS5        g )NThresholdedRelur   r   r   s    r   test_ThresholdedRelu)TestAutomaticUpgrade.test_ThresholdedRelu*  s    /4r   c           	          [         R                  " S[        R                  S/[        R
                  " / SQ5      S9nU R                  SS/ SQS/// SQ/[        R                  [        R                  /U/S	9  g )
NrN  r0   rn  rO  TilerO   r@   r  rQ  rS  )r   repeatss     r   	test_TileTestAutomaticUpgrade.test_Tile-  so    $$""!288I3F
 	L 1 12 	 	 	
r   c           	      ~    U R                  SS/ SQ// SQ/ SQ/[        R                  [        R                  /SS0S9  g )NTopKr   r@   r  kr/   rh   r  r   s    r   	test_TopKTestAutomaticUpgrade.test_TopK;  sC    K	"%++[->->?( 	 	
r   c           	      >    U R                  SS/ SQ// SQ/S/ SQ0S9  g )N	Transposer   )r   r/   rB   r0   r)   )r   r)   rB   r/   r0   perm)r   rA   r/   r   r0   r   r   r   s    r   test_Transpose#TestAutomaticUpgrade.test_TransposeE  s/    ?+ 	 	
r   c                 (    U R                  SS5        g )NTrilur   r   r   s    r   
test_TriluTestAutomaticUpgrade.test_TriluN  r7  r   c                 6    U R                  SS/ SQ/S //5        g )NUniquer   r@   r   r   s    r   test_Unique_1"TestAutomaticUpgrade.test_Unique_1Q  s    hYK4&Br   c                 :    U R                  SS/ SQ// SQ/SS0S9  g )Nr  r   r@   )r0   NrB   rK   r   r   r   r   s    r   test_Unique_2"TestAutomaticUpgrade.test_Unique_2T  s(    b9+~fa[ 	 	
r   c                 <    U R                  SS/ SQ// SQ/SS/0S9  g )Nr  r   r@   )r0   rA   r   rB   r  r/   r   r   r   s    r   test_Unsqueeze#TestAutomaticUpgrade.test_UnsqueezeY  s*    YK,} 	 	
r   c           	      <    U R                  SS/ SQ// SQ/SSS.S9  g )	NUpsampler   )r   r0   rA   rB   )r   r0   rO   r   g       @r  )width_scaleheight_scaler   r   r   s    r   test_Upsample"TestAutomaticUpgrade.test_Upsample^  s-    NO"%s; 	 	
r   c           	          U R                  SSSS/SS/SS//SS//[        R                  [        R                  [        R                  /5        g )NWherer#   r/   r0   )r   r   r2   r   r   s    r   
test_WhereTestAutomaticUpgrade.test_Whereg  sQ    VaVaV$VH{00+2C2CD	
r   c                     U R                  SSSS/SS//SS//[        R                  [        R                  /[        R                  /5        g )NXorr)   r/   r0   r1   r   s    r   test_XorTestAutomaticUpgrade.test_Xorp  r5   r   c           	          U R                  SS/ SQ/ SQ// SQ/[        R                  [        R                  /[        R                  /S9  g )NCastLiker   r]   rg   r  )r   r   r   FLOAT16r   s    r   test_CastLike"TestAutomaticUpgrade.test_CastLike{  sH    	"K$**K,?,?@%--. 	 	
r   c           
          U R                  SS/ SQSS/SS/// SQ/[        R                  [        R                  [        R                  /[        R                  /SS0S9  g )	NLayerNormalization   r.  rA   rB   rK   r/   r  r  r   s    r   test_LayerNormalization,TestAutomaticUpgrade.test_LayerNormalization  sa     Aq6Aq6*N$**K,=,={?P?PQ%++,1+ 	 	
r   window_function_namec           	          [         R                  " S[        R                  / [        R
                  " S/5      S9nU R                  US/ /S//[        R                  /U/S9  g )Nr  r   rO  r
  rQ  )r   r   r   rb   r   rT  r   )r   r  r*  s      r   _test_window_function*TestAutomaticUpgrade._test_window_function  s`    !!#{'8'8rRTQUW DTF 	 	
r   c                 &    U R                  S5        g )NBlackmanWindowr  r   s    r   test_BlackmanWindow(TestAutomaticUpgrade.test_BlackmanWindow  s    ""#34r   c                 &    U R                  S5        g )N
HannWindowr  r   s    r   test_HannWindow$TestAutomaticUpgrade.test_HannWindow  s    ""<0r   c                 &    U R                  S5        g )NHammingWindowr  r   s    r   test_HammingWindow'TestAutomaticUpgrade.test_HammingWindow  s    ""?3r   c           	         U R                  SS/ SQ/ // SQ/5        U R                  SS/ SQ/ // SQ/5        U R                  SS/ SQ/ // SQ/SS0S9  U R                  SS/ SQ/ // SQ/SS0S9  U R                  SS/ SQ/ // SQ/S	S0S9  U R                  SS/ SQ/ // SQ/S	S0S9  U R                  SS/ SQ/ // SQ/SS
S.S9  g )NDFTr
  )r/   r  r   )r/   r  r/   )r/   r#   r/   onesidedr   r   inverser   )r!  rK   r   r   s    r   test_DFTTestAutomaticUpgrade.test_DFT  s   eR*b)9J<HeR*b)9J<H2
B')ZO 	 	
 	2
B')ZO 	 	
 	2
B'*i^ 	 	
 	2
B'*i^ 	 	
 	2
B'*TU=V 	 	
r   operator_namec                 
   [         R                  " S[        R                  SS/[        R
                  R                  SS5      R                  [        R                  5      S9n[         R                  " S[        R                  S/[        R                  " S/5      S9n[         R                  " S[        R                  S	/[        R                  " S	5      R                  [        R                  5      S9nU R                  US
SS/S/S	/// SQ/[        R                  [        R                  [        R                  [        R                  /X#U/S9  [         R                  " S[        R                  SS/[        R
                  R                  SS5      R                  [        R                  5      S9n[         R                  " S[        R                  S/[        R                  " S/5      S9n[         R                  " S[        R                  S	/[        R                  " S	5      R                  [        R                  5      S9nU R                  US
SS/S/S	/// SQ/[        R                  [        R                  [        R                  [        R                  /SS0X#U/S9  [         R                  " S[        R                  / SQ[        R
                  R                  SSS5      R                  [        R                  5      S9n[         R                  " S[        R                  S/[        R                  " S/5      S9n[         R                  " S[        R                  S	/[        R                  " S	5      R                  [        R                  5      S9nU R                  US
/ SQS/S	/// SQ/[        R                  [        R                  [        R                  [        R                  /X#U/S9  [         R                  " S[        R                  / SQ[        R
                  R                  SSS5      R                  [        R                  5      S9n[         R                  " S[        R                  S/[        R                  " S/5      S9n[         R                  " S[        R                  S	/[        R                  " S	5      R                  [        R                  5      S9n[         R                  " S[        R                  S/[        R                  " S	/5      S9nU R                  US
/ SQS/S	/// SQ/[        R                  [        R                  [        R                  [        R                  /SS0X#XE/S9  g )Nr  r/   @   rO  rN  r   r  r  r  r
  )r/   r)   r  r/   rQ  )r/   r)   r#   r/   r   )r   rR  )r/   r&  r/   e)r   r   r   r   r   r   r   r   r   rb   rT  onesr   )r   r$  signal
frame_stepwindowframe_lengths         r   "_test_short_time_fourier_transform7TestAutomaticUpgrade._test_short_time_fourier_transform  sw   ##R2&--bjj9	
 ''""!288QC=

 ##"""BGGBK4F4Frzz4R
 	WqcB4 O!!!!!!!!	  V4 	 	
 ##R2&--bjj9	
 ''""!288QC=

 ##"""BGGBK4F4Frzz4R
 	WqcB4 N!!!!!!!!	 q/V4 	 	
  ##2q)00<	
 ''""!288QC=

 ##"""BGGBK4F4Frzz4R
 	!rd#O!!!!!!!!	  V4 	 	
 ##2q)00<	
 ''""!288QC=

 ##"""BGGBK4F4Frzz4R
 ))""!288RD>
 	!rd#N!!!!!!!!	 q/VB 	 	
r   c                 &    U R                  S5        g )NSTFT)r-  r   s    r   	test_STFTTestAutomaticUpgrade.test_STFT)  s    //7r   c                 <   [         R                  " S[        R                  / [        R
                  " S/5      S9n[         R                  " S[        R                  / [        R
                  " S/5      S9n[         R                  " S[        R                  / [        R
                  " S/5      S9n[         R                  " S[        R                  / [        R
                  " S/5      S9n[         R                  " S	[        R                  / [        R
                  " S
/5      S9nU R                  SS/ / / / / /SS//[        R                  [        R                  [        R                  [        R                  [        R                  /UUUUU/S9  [         R                  " S[        R                  / [        R
                  " S/5      S9n[         R                  " S[        R                  / [        R
                  " S/5      S9n[         R                  " S[        R                  / [        R
                  " S/5      S9n[         R                  " S[        R                  / [        R
                  " S/5      S9n[         R                  " S	[        R                  / [        R
                  " S
/5      S9nU R                  SS/ / / / / /SS//[        R                  [        R                  [        R                  [        R                  [        R                  /UUUUU/S9  g )Nr  r   rO  rN  r&  r  r   dr'  r   MelWeightMatrixr
  !   rQ  r      r  )r   r   r   rb   r   rT  r   r   )r   num_mel_bins
dft_lengthsample_ratelower_edge_hertzupper_edge_hertzs         r   test_MelWeightMatrix)TestAutomaticUpgrade.test_MelWeightMatrix,  s   ))"""((B4.
 ''"""((B4.

 (("""((A3-
 "--"""((A3-
 "--"""((A3-
 	RR "XJ!!!!!!!!!!    	 	
* ))"""((B4.
 ''"""((B4.

 (("""((A3-
 "--"""((A3-
 "--"""((A3-
 	RR "XJ!!!!!!!!!!    	 	
r   c           	      f   [         R                  " S[        R                  SS/[        R
                  " / SQ5      S9n[         R                  " S[        R                  S/[        R
                  " SS/5      S9nU R                  SS	/ / /SS//[        R                  [        R                  /X/S
9  g )Ninputr/   rA   )r   r/   r0   rA   rB   rO   r)   r  rO  rU  r0   CenterCropPadr   rQ  )r   r   r   r   r   rT  rb   r   )r   input_rU  s      r   test_CenterCropPad'TestAutomaticUpgrade.test_CenterCropPadw  s    ##Q23	
 ""[&&aSrxxA7G
 	HVH 1 12 	 	
r   c                 x    U R                  SSSS//SS//[        R                  /[        R                  /5        g )N
BitwiseNotr   r/   r0   r  r   s    r   test_BitwiseNot$TestAutomaticUpgrade.test_BitwiseNot  s@    VHVH	
r   c                     U R                  SSSS/SS//SS//[        R                  [        R                  /[        R                  /5        g )N
BitwiseAndr   r/   r0   r   r   INT16r   s    r   test_BitwiseAnd$TestAutomaticUpgrade.test_BitwiseAnd  O    VaVVH 1 12	
r   c                     U R                  SSSS/SS//SS//[        R                  [        R                  /[        R                  /5        g )N	BitwiseOrr   r/   r0   rK  r   s    r   test_BitwiseOr#TestAutomaticUpgrade.test_BitwiseOr  sO    VaVVH 1 12	
r   c                     U R                  SSSS/SS//SS//[        R                  [        R                  /[        R                  /5        g )N
BitwiseXorr   r/   r0   rK  r   s    r   test_BitwiseXor$TestAutomaticUpgrade.test_BitwiseXor  rO  r   c           	      D    U R                  SS/ SQS/S/// SQ/SSS.S9  g )	NGroupNormalizationr   )r0   rA   r/   r/   r   gh㈵>r/   )epsilon
num_groupsr   r   r   s    r   test_GroupNormalization,TestAutomaticUpgrade.test_GroupNormalization  s6     A3$N"!4 	 	
r   c                 >    U R                  SSSS/SS//SS//5        g )NStringConcatr   r/   r0   r   r   s    r   test_StringConcat&TestAutomaticUpgrade.test_StringConcat  s/    VaVVH		
r   c                 x    U R                  SSSS//SS//[        R                  /[        R                  /5        g )NRegexFullMatchr   r/   r0   )r   r   r  r2   r   s    r   test_RegexFullMatch(TestAutomaticUpgrade.test_RegexFullMatch  s@    VHVH 	
r   c                 V   [         R                  R                  5       nU Vs/ s H   o"R                  S:X  d  M  UR                  PM"     nn/ SQnU Vs/ s H  oUU;  d  M
  UPM     nn[        U5      [        [        5      -
  n[        U5        [        U5      S:X  d   eg s  snf s  snf )Nr  )ConcatFromSequence
SequenceAtSequenceConstructSequenceEmptySequenceEraseSequenceInsertSequenceLengthSequenceMapSplitToSequenceOptionalOptionalGetElementOptionalHasElementStringSplitr   )	r  defsget_all_schemasdomainr  setr
   printlen)r   all_schemasschemaall_op_namesexcluded_opsr   untested_opss          r   test_ops_tested$TestAutomaticUpgrade.test_ops_tested  s    ii//12=U+RTAT+U
" &2L\r|5K\L<(3z?:l< A%%%- V$ Ms   B!B!	B&!B& )r   N)__name__
__module____qualname____firstlineno__r   r   r$   r*   r3   r8   r<   rC   rH   rL   rU   rY   rc   ri   rn   rq   ru   ry   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r  r  r  r  r  r  r  r#  r&  r*  r.  r2  r8  r;  r?  rC  rG  rK  rV  rZ  r^  rb  rf  rj  ro  ru  rx  r|  r  r  r  r  r  r  r  r  r  r  r  r  r  r  r  r  r  r  r  r  r  r  r  r  r  r"  r&  r+  r/  r4  r7  r;  r?  rB  rH  rK  rN  rR  rW  rZ  r`  rd  rh  rl  rp  rt  rx  r|  r  r  r  r  r  r  r  r  r  r  r  r  r  r  r  r  r  r  r  r  r  r  r  r  r  r  r  r  r  r  r  r  r  r  r  r  r!  r$  r'  r,  r1  r5  r:  r>  rB  rF  rJ  rV  rZ  r^  rb  rf  rj  rn  rr  rv  r|  r  r  r  r  r  r  r  r  r  r  r  r  r  r  r  r  r  r  r  r  r  r  r  r  r  r  r  r  r  r  r  r  strr  r  r  r  r"  r-  r1  r=  rC  rG  rM  rR  rV  r\  r`  rd  r  __static_attributes__r  r   r   r   r      s   GH*)	
**




RU





))
/	



I*IVU

SX
GI;








)(



=9







H
(H
@WJ

PI*JXTS

	



$P/,-
>





N

H/,
\&
P4
;




:GP









V
*EX

H






Q	

M













OGGKNIHHIHNI
L



$

+

	

	

<I

+L))(


???@@@--



IL	


I(
 5


+C





	

	
	
# 	
$ 	
514
&p
 p
 p
d8I
V
&






&r   r   __main__)unittestautomatic_conversion_test_basenumpyr   r  r   r   r
   TestAutomaticConversionr   r  mainr  r   r   <module>r     sK     %   $ 
Q&9QQ Q&h6 zMMO r   