o
    wi                     @  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 )	ShrinkreturnNonec                  C  sV   t jjddgdgdd} tjddtjd}tjg d	tjd}t| |g|gd
d d S )Nr   xy      ?)inputsoutputslambd        @dtype)r   r   r      test_shrink_hardr   r   nameonnxhelper	make_nodenparangefloat32arrayr   nodeXY r#   _/home/ubuntu/sommelier/.venv/lib/python3.10/site-packages/onnx/backend/test/case/node/shrink.pyexport_hard_shrink   s   zShrink.export_hard_shrinkc                  C  sX   t jjddgdgddd} tjddtjd}tjg d	tjd}t| |g|gd
d d S )Nr   r   r	   r
   )r   r   r   biasr   r   r   )g      r   r   r   g      ?test_shrink_softr   r   r   r#   r#   r$   export_soft_shrink   s   zShrink.export_soft_shrinkN)r   r   )__name__
__module____qualname__staticmethodr%   r(   r#   r#   r#   r$   r      s
    r   )

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