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 )Nr   xygreaterinputsoutputs         test_greaterr   r   nametest_greater_int8test_greater_int16   )r   r   r   )sizedtypetest_greater_uint8test_greater_uint16test_greater_uint32test_greater_uint64)onnxhelper	make_nodenprandomrandnastypefloat32r
   r   int8int16randintuint8uint16uint32uint64noder   r	   z r/   W/home/ubuntu/.local/lib/python3.10/site-packages/onnx/backend/test/case/node/greater.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_bcastr   )
r   r   r   r    r!   r"   r#   r$   r
   r   r,   r/   r/   r0   export_greater_broadcast9   s   z Greater.export_greater_broadcastN)r   r   )__name__
__module____qualname__staticmethodr1   r3   r/   r/   r/   r0   r      s
    *r   )

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