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 )ShrinkNone)returnc                  C  s\   t jjddgdgdd} tjddtjd}tjd	d
d
d
dgtjd}t| |g|gdd d S )Nr   xy      ?)inputsoutputslambd        @Zdtyper      Ztest_shrink_hardr   r   nameonnxhelperZ	make_nodenpZarangeZfloat32arrayr   nodeXY r   F/tmp/pip-unpacked-wheel-xnis5xre/onnx/backend/test/case/node/shrink.pyexport_hard_shrink   s    zShrink.export_hard_shrinkc                  C  s^   t jjddgdgddd} tjddtjd}tjd	d
d
d
dgtjd}t| |g|gdd d S )Nr   r   r	   r
   )r   r   r   Zbiasr   r   r   g      r   g      ?Ztest_shrink_softr   r   r   r   r   r   export_soft_shrink   s    zShrink.export_soft_shrinkN)__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
   