
    Si                     z    S SK Jr  S SKrS SKrS SKJr  S SKJr  S	S
S jjr	 S       S
S jjr
 " S S\5      rg)    )annotationsN)Base)expectc                    [         R                  " XS9nUS:X  a  [         R                  " X15      nUR                  [         R                  5      $ N)axis   )npargmaxexpand_dimsastypeint64datar   keepdimsresults       a/var/www/html/ai-image-ml/venv/lib/python3.13/site-packages/onnx/backend/test/case/node/argmax.pyargmax_use_numpyr      s8    YYt'F1}-==""    c                    [         R                  " X5      n [         R                  " XS9nU R                  U   U-
  S-
  nU(       a  [         R                  " X15      nUR                  [         R                  5      $ r   )r
   flipr   shaper   r   r   r   s       r   "argmax_use_numpy_select_last_indexr      s\     774DYYt'FZZ&*F-==""r   c                      \ rS rSr\SS j5       r\SS j5       r\SS j5       r\SS j5       r\SS j5       r	\SS j5       r
\SS j5       r\SS	 j5       rS
rg)ArgMax   c                    [         R                  " SS/SS//[         R                  S9n SnSn[        R                  R                  SS/S	/XS
9n[        XUS9n[        UU /U/SS9  [         R                  R                  SS/ SQ5      R                  [         R                  5      n [        XUS9n[        X0/U/SS9  g )N      
   dtyper	   r   r   r   r   inputsoutputsr   r   r   r   test_argmax_no_keepdims_exampler$   r%   namer   r      test_argmax_no_keepdims_randomr
   arrayfloat32onnxhelper	make_noder   r   randomuniformr   r   r   r   noder   s        r   export_no_keepdimsArgMax.export_no_keepdims    s    xx!Q!R)<{{$$fXz % 
 "$HE6H2		
 yy  b)4;;BJJG!$HE&8X	
r   c                 ~   [         R                  " SS/SS//[         R                  S9n SnSn[        R                  R                  SS/S/XS	9n[        XUS
9n[        X0/U/SS9  [         R                  R                  SS/ SQ5      R                  [         R                  5      n [        XUS
9n[        X0/U/SS9  g )Nr   r   r    r!   r	   r   r   r   r#   r&   test_argmax_keepdims_exampler(   r*   r+   test_argmax_keepdims_randomr.   r6   s        r   export_keepdimsArgMax.export_keepdims8   s    xx!Q!R)<{{$$fXz % 
 "$HE&8V	
 yy  b)4;;BJJG!$HE&8U	
r   c                 z   [         R                  " SS/SS//[         R                  S9n Sn[        R                  R                  SS/S/US	9n[        XS
9n[        UU /U/SS9  [         R                  R                  SS/ SQ5      R                  [         R                  5      n [        XS
9n[        UU /U/SS9  g )Nr   r   r    r!   r	   r   r   r   )r$   r%   r   r    test_argmax_default_axis_exampler(   r*   r+   test_argmax_default_axis_randomr.   r   r   r7   r   s       r   export_default_axes_keepdims#ArgMax.export_default_axes_keepdimsM   s    xx!Q!R)<{{$$fXzH % 

 "$:6H3		
 yy  b)4;;BJJG!$:6H2		
r   c                    [         R                  " SS/SS//[         R                  S9n SnSn[        R                  R                  SS/S	/XS
9n[        XUS9n[        UU /U/SS9  [         R                  R                  SS/ SQ5      R                  [         R                  5      n [        XUS9n[        UU /U/SS9  g )Nr   r   r    r!   r	   r   r   r   r#   r&   *test_argmax_negative_axis_keepdims_exampler(   r*   r+   )test_argmax_negative_axis_keepdims_randomr.   r6   s        r   export_negative_axis_keepdims$ArgMax.export_negative_axis_keepdimsh   s    xx!Q!R)<{{$$fXz % 
 "$HE6H=		
 yy  b)4;;BJJG!$HE6H<		
r   c            	        [         R                  " SS/SS//[         R                  S9n SnSn[        R                  R                  SS/S	/UUS
S9n[        XUS9n[        UU /U/SS9  [         R                  R                  SS/ SQ5      R                  [         R                  5      n [        XUS9n[        UU /U/SS9  g )Nr   r   r    r!   r	   r   r   r   r   Tr$   r%   r   r   select_last_indexr&   1test_argmax_no_keepdims_example_select_last_indexr(   r*   r+   0test_argmax_no_keepdims_random_select_last_indexr
   r/   r0   r1   r2   r3   r   r   r4   r5   r   r6   s        r   $export_no_keepdims_select_last_index+ArgMax.export_no_keepdims_select_last_index   s    xx!Q!R)<{{$$8J" % 
 4DhW6HD		
 yy  b)4;;BJJG3DhW6HC		
r   c            	        [         R                  " SS/SS//[         R                  S9n SnSn[        R                  R                  SS/S/UUS	S
9n[        XUS9n[        UU /U/SS9  [         R                  R                  SS/ SQ5      R                  [         R                  5      n [        XUS9n[        UU /U/SS9  g )Nr   r   r    r!   r	   r   r   r   TrM   r&   .test_argmax_keepdims_example_select_last_indexr(   r*   r+   -test_argmax_keepdims_random_select_last_indexrQ   r6   s        r   !export_keepdims_select_last_index(ArgMax.export_keepdims_select_last_index   s    xx!Q!R)<{{$$8J" % 
 4DhW6HA		
 yy  b)4;;BJJG3DhW6H@		
r   c                 |   [         R                  " SS/SS//[         R                  S9n Sn[        R                  R                  SS/S/US	S
9n[        XS9n[        UU /U/SS9  [         R                  R                  SS/ SQ5      R                  [         R                  5      n [        XS9n[        UU /U/SS9  g )Nr   r   r    r!   r	   r   r   r   T)r$   r%   r   rN   r@   2test_argmax_default_axis_example_select_last_indexr(   r*   r+   1test_argmax_default_axis_random_select_last_indexrQ   rC   s       r   .export_default_axes_keepdims_select_last_index5ArgMax.export_default_axes_keepdims_select_last_index   s    xx!Q!R)<{{$$8J" % 
 4DL6HE		
 yy  b)4;;BJJG3DL6HD		
r   c            	        [         R                  " SS/SS//[         R                  S9n SnSn[        R                  R                  SS/S	/UUS
S9n[        XUS9n[        UU /U/SS9  [         R                  R                  SS/ SQ5      R                  [         R                  5      n [        XUS9n[        UU /U/SS9  g )Nr   r   r    r!   rG   r	   r   r   r   TrM   r&   <test_argmax_negative_axis_keepdims_example_select_last_indexr(   r*   r+   ;test_argmax_negative_axis_keepdims_random_select_last_indexrQ   r6   s        r   /export_negative_axis_keepdims_select_last_index6ArgMax.export_negative_axis_keepdims_select_last_index   s    xx!Q!R)<{{$$8J" % 
 4DhW6HO		
 yy  b)4;;BJJG3DhW6HN		
r    N)returnNone)__name__
__module____qualname____firstlineno__staticmethodr8   r=   rD   rJ   rR   rW   r\   ra   __static_attributes__rc   r   r   r   r      s    
 
. 
 
( 
 
4 
 
4 
 
> 
 
> 
 
< 
 
r   r   )r   r	   )r   
np.ndarrayr   intr   rm   rd   rl   )r   T)
__future__r   numpyr
   r1   onnx.backend.test.case.baser   onnx.backend.test.case.noder   r   r   r   rc   r   r   <module>rr      sS    #   , .# 6:#
##/2##a
T a
r   