o
    soik                     @  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edddZedd	d
ZedddZedddZ	edddZ
edddZedddZedddZedddZedddZedddZdS )ModreturnNonec                  C  l   t jjdddgdgdd} tg dtj}tg dtj}t||}t| ||g|gd	d
 d S )Nr   xyz   inputsoutputsfmodg333333g@      @g333333@g       @g @g333333r   g g333333@r   test_mod_mixed_sign_float64r   r   name)	onnxhelper	make_nodenparrayastypefloat64r   r   noder	   r
   r    r!   S/home/ubuntu/.local/lib/python3.10/site-packages/onnx/backend/test/case/node/mod.pyexport_mod_mixed_sign_float64   
   z!Mod.export_mod_mixed_sign_float64c                  C  r   )Nr   r	   r
   r   r   r   r   r   test_mod_mixed_sign_float32r   )	r   r   r   r   r   r   float32r   r   r   r!   r!   r"   export_mod_mixed_sign_float32      z!Mod.export_mod_mixed_sign_float32c                  C  r   )Nr   r	   r
   r   r   r   r   r   test_mod_mixed_sign_float16r   )	r   r   r   r   r   r   float16r   r   r   r!   r!   r"   export_mod_mixed_sign_float16"   r(   z!Mod.export_mod_mixed_sign_float16c                  C  j   t jjdddgdgd} tg dtj}tg dtj}t||}t| ||g|gdd	 d S )
Nr   r	   r
   r   r   r            i      r3      r1   test_mod_mixed_sign_int64r   )	r   r   r   r   r   r   int64modr   r   r!   r!   r"   export_mod_mixed_sign_int64-      zMod.export_mod_mixed_sign_int64c                  C  r,   )
Nr   r	   r
   r   r-   r.   r4   test_mod_mixed_sign_int32r   )	r   r   r   r   r   r   int32r;   r   r   r!   r!   r"   export_mod_mixed_sign_int32:   r=   zMod.export_mod_mixed_sign_int32c                  C  r,   )
Nr   r	   r
   r   r-   r.   r4   test_mod_mixed_sign_int16r   )	r   r   r   r   r   r   int16r;   r   r   r!   r!   r"   export_mod_mixed_sign_int16G   r=   zMod.export_mod_mixed_sign_int16c                  C  r,   )
Nr   r	   r
   r   r-   r.   r4   test_mod_mixed_sign_int8r   )	r   r   r   r   r   r   int8r;   r   r   r!   r!   r"   export_mod_mixed_sign_int8T   r=   zMod.export_mod_mixed_sign_int8c                  C  r,   )
Nr   r	   r
   r   r-   r2   r0   r1   r5   r8   r3   test_mod_uint8r   )	r   r   r   r   r   r   uint8r;   r   r   r!   r!   r"   export_mod_uint8a   r=   zMod.export_mod_uint8c                  C  r,   )
Nr   r	   r
   r   r-   rG   rH   test_mod_uint16r   )	r   r   r   r   r   r   uint16r;   r   r   r!   r!   r"   export_mod_uint16n   r=   zMod.export_mod_uint16c                  C  r,   )
Nr   r	   r
   r   r-   rG   rH   test_mod_uint32r   )	r   r   r   r   r   r   uint32r;   r   r   r!   r!   r"   export_mod_uint32{   r=   zMod.export_mod_uint32c                  C  r,   )
Nr   r	   r
   r   r-   rG   rH   test_mod_uint64r   )	r   r   r   r   r   r   uint64r;   r   r   r!   r!   r"   export_mod_uint64   r=   zMod.export_mod_uint64c                  C  r   )Nr   r	   r
   r   r   r   r.   r4   test_mod_int64_fmodr   )	r   r   r   r   r   r   r:   r   r   r   r!   r!   r"   export_mod_int64_fmod   r$   zMod.export_mod_int64_fmodc                  C  sp   t jjdddgdgd} tddg dtj}td	gtj}t	||}t
| ||g|gd
d d S )Nr   r	   r
   r   r-   r      )r8   r5   r1   r0   test_mod_broadcastr   )r   r   r   r   arangereshaper   r?   r   r;   r   r   r!   r!   r"   export_mod_broadcast   s   	zMod.export_mod_broadcastN)r   r   )__name__
__module____qualname__staticmethodr#   r'   r+   r<   r@   rC   rF   rK   rN   rQ   rT   rV   r[   r!   r!   r!   r"   r      s6    

r   )

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