o
    soi1
                     @  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 )	DivreturnNonec                  C  st  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 tjjddtjd}tjjddtjd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 tjjddtjd}tjjddtjd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 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            test_div_exampler   r   name         ?test_div   )r   r   r   )sizedtypetest_div_int8test_div_int16test_div_uint8test_div_uint16test_div_uint32test_div_uint64)onnxhelper	make_nodenparrayastypefloat32r   randomrandnrandrandintint8int16uint8uint16uint32uint64noder   r	   r
    r4   S/home/ubuntu/.local/lib/python3.10/site-packages/onnx/backend/test/case/node/div.pyexport   sJ   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   test_div_bcastr   )
r!   r"   r#   r$   r(   r)   r&   r'   r*   r   r2   r4   r4   r5   export_div_broadcast>   s   zDiv.export_div_broadcastN)r   r   )__name__
__module____qualname__staticmethodr6   r8   r4   r4   r4   r5   r      s
    /r   )

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