U
    rh                     @  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                   @  s0   e Zd ZeddddZeddddZdS )SubNone)returnc                  C  s   t jjdddgdgd} tdddgtj}tdddgtj}|| }t| ||g|gd	d
 tj	dddtj}tj	dddtj}|| }t| ||g|gdd
 tjj
dddtjd}tjj
ddtjd}|| }t| ||g|gdd
 d S )Nr   xyzinputsoutputs         Ztest_sub_exampler   r   name      Ztest_sub      )r   r   r   )sizeZdtypeZtest_sub_uint8)onnxhelper	make_nodenparrayastypefloat32r   randomrandnrandintZuint8noder   r	   r
    r$   C/tmp/pip-unpacked-wheel-xnis5xre/onnx/backend/test/case/node/sub.pyexport   s"    z
Sub.exportc                  C  sf   t jjdddgdgd} tjdddtj}tjdtj}|| }t| ||g|gd	d
 d S )Nr   r   r	   r
   r   r   r   r   Ztest_sub_bcastr   )	r   r   r   r   r   r    r   r   r   r"   r$   r$   r%   export_sub_broadcast%   s    zSub.export_sub_broadcastN)__name__
__module____qualname__staticmethodr&   r'   r$   r$   r$   r%   r      s   r   )

__future__r   Znumpyr   r   Zonnx.backend.test.case.baser   Zonnx.backend.test.case.noder   r   r$   r$   r$   r%   <module>   s
   