o
    wiR	                     @  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 )	Less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lessinputsoutputs         	test_lessr   r   nametest_less_int8test_less_int16   )r   r   r   )sizedtypetest_less_uint8test_less_uint16test_less_uint32test_less_uint64)onnxhelper	make_nodenprandomrandnastypefloat32r
   r   int8int16randintuint8uint16uint32uint64noder   r	   z r/   ]/home/ubuntu/sommelier/.venv/lib/python3.10/site-packages/onnx/backend/test/case/node/less.pyexport   sB   zLess.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_less_bcastr   )
r   r   r   r    r!   r"   r#   r$   r
   r   r,   r/   r/   r0   export_less_broadcast9   s   zLess.export_less_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   