o
    soi
                     @  sH   d dl mZ d dlZd dlZd dlmZ d dlmZ G dd deZ	dS )    )annotationsN)Base)expectc                   @  s(   e Zd ZedddZedddZdS )	GreaterreturnNonec                  C  s:  t jjdddgdgd} tjdddtj}tjdddtj}t||}t	| ||g|gd	d
 tjdddtj
}tjdddtj
}t||}t	| ||g|gdd
 tjdddtj}tjdddtj}t||}t	| ||g|gdd
 tjjddtjd}tjjddtjd}t||}t	| ||g|gdd
 tjjddtjd}tjjddtjd}t||}t	| ||g|gdd
 tjjddtjd}tjjddtjd}t||}t	| ||g|gdd
 tjjddtjd}tjjddtjd}t||}t	| ||g|gdd
 d S )NGreaterOrEqualxygreater_equalinputsoutputs         test_greater_equalr   r   nametest_greater_equal_int8test_greater_equal_int16   )r   r   r   )sizedtypetest_greater_equal_uint8test_greater_equal_uint16test_greater_equal_uint32test_greater_equal_uint64)onnxhelper	make_nodenprandomrandnastypefloat32r   r   int8int16randintuint8uint16uint32uint64noder	   r
   z r0   ]/home/ubuntu/.local/lib/python3.10/site-packages/onnx/backend/test/case/node/greater_equal.pyexport   sB   zGreater.exportc                  C  sj   t jjdddgdgd} tjdddtj}tjdtj}t||}t	| ||g|gd	d
 d S )Nr   r	   r
   r   r   r   r   r   test_greater_equal_bcastr   )
r   r   r    r!   r"   r#   r$   r%   r   r   r-   r0   r0   r1   export_greater_broadcast9   s   z Greater.export_greater_broadcastN)r   r   )__name__
__module____qualname__staticmethodr2   r4   r0   r0   r0   r1   r      s
    *r   )

__future__r   numpyr!   r   onnx.backend.test.case.baser   onnx.backend.test.case.noder   r   r0   r0   r0   r1   <module>   s   