
    Sim
                    X    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 jr	 " S S\5      r
g)	    )annotationsN)Base)expectc                    [         R                  " U SU5      n[         R                  " USU5      n[         R                  " XCSS9n[         R                  " USU5      $ )Nr   wrap)mode)npswapaxeschoose)dataindicesaxisdata_swappedindex_swappedgathereds         i/var/www/html/ai-image-ml/venv/lib/python3.13/site-packages/onnx/backend/test/case/node/gatherelements.pygather_elementsr      sI    ;;tQ-LKKD1Myy6BH;;xD))    c                  P    \ rS rSr\SS j5       r\SS j5       r\SS j5       rSrg)GatherElements   c                 `   Sn [         R                  R                  SSS/S/U S9n[        R                  " SS/SS	//[        R
                  S
9n[        R                  " SS/SS//[        R                  S
9n[        X#U 5      n[        UX#R                  [        R                  5      /U/SS9  g )N   r   r   r   yinputsoutputsr            dtyper   test_gather_elements_0r   r   nameonnxhelper	make_noder	   arrayfloat32int32r   r   astypeint64r   noder   r   r   s        r   export_gather_elements_0'GatherElements.export_gather_elements_0   s    {{$$I&E	 % 
 xx!Q!Q(

;((QFQF+288<D40
 	..23C)		
r   c                 f   Sn [         R                  R                  SSS/S/U S9n[        R                  " / SQ/ SQ/ S	Q/[        R
                  S
9n[        R                  " / SQ/ SQ/[        R                  S
9n[        X#U 5      n[        UX#R                  [        R                  5      /U/SS9  g )Nr   r   r   r   r   r   r   r   r   r                	   r!   )r   r   r   )r   r   r   test_gather_elements_1r$   r&   r/   s        r   export_gather_elements_1'GatherElements.export_gather_elements_1.   s    {{$$I&E	 % 
 xxIy9L((Iy1BD40
 	..23C)		
r   c                 f   Sn [         R                  R                  SSS/S/U S9n[        R                  " / SQ/ SQ/ S	Q/[        R
                  S
9n[        R                  " / SQ/ SQ/[        R                  S
9n[        X#U 5      n[        UX#R                  [        R                  5      /U/SS9  g )Nr   r   r   r   r   r   r4   r5   r8   r!   )r   )rA   r   r   %test_gather_elements_negative_indicesr$   r&   r/   s        r   'export_gather_elements_negative_indices6GatherElements.export_gather_elements_negative_indicesF   s    {{$$I&E	 % 
 xxIy9L((K4BHHED40
 	..23C8		
r    N)returnNone)	__name__
__module____qualname____firstlineno__staticmethodr1   r=   rC   __static_attributes__rE   r   r   r   r      s<    
 
. 
 
. 
 
r   r   )r   )
__future__r   numpyr	   r'   onnx.backend.test.case.baser   onnx.backend.test.case.noder   r   r   rE   r   r   <module>rR      s)    #   , .*G
T G
r   