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 )DivNone)returnc                  C  s  t jjdddgdgd} tddgtj}tdd	gtj}|| }t| ||g|gd
d tj	dddtj}tj
dddtjd }|| }t| ||g|gdd tjjddtjd}tjjddtjdd }|| }t| ||g|gdd d S )Nr   xyzinputsoutputs            Ztest_div_exampler   r   name         ?Ztest_div   )r   r   r   )sizeZdtypeZtest_div_uint8)onnxhelper	make_nodenparrayastypefloat32r   randomrandnrandrandintZuint8noder   r	   r
    r%   C/tmp/pip-unpacked-wheel-xnis5xre/onnx/backend/test/case/node/div.pyexport   s"    z
Div.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d	 }|| }t	| ||g|gd
d d S )Nr   r   r	   r
   r   r   r   r   r   Ztest_div_bcastr   )
r   r   r   r   r   r    r   r   r!   r   r#   r%   r%   r&   export_div_broadcast%   s    zDiv.export_div_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
   