
    Ng                     J    d dl Zd dlZd dlmZ d dlmZ  G d de          ZdS )    N)Base)expectc                   >    e Zd Zedd            Zedd            ZdS )ShrinkreturnNc                      t           j                            ddgdgd          } t          j        ddt          j                  }t          j        g d	t          j                  }t          | |g|g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Ys      ^/var/www/html/ai-engine/env/lib/python3.11/site-packages/onnx/backend/test/case/node/shrink.pyexport_hard_shrinkzShrink.export_hard_shrink   s    {$$5E	 % 
 
 IdCrz222H%%%RZ888tQC!3EFFFFFF    c                     t           j                            ddgdgdd          } t          j        ddt          j                  }t          j        g d	t          j                  }t          | |g|gd
           d S )Nr   r	   r
   r   )r   r   r   biasr   r   r   )g      r   r   r   g      ?test_shrink_softr   r   r    s      r$   export_soft_shrinkzShrink.export_soft_shrink   s    {$$5E % 
 
 IdCrz222H)))<<<tQC!3EFFFFFFr&   )r   N)__name__
__module____qualname__staticmethodr%   r*    r&   r$   r   r      s\        	G 	G 	G \	G 
G 
G 
G \
G 
G 
Gr&   r   )numpyr   r   onnx.backend.test.case.baser   onnx.backend.test.case.noder   r   r/   r&   r$   <module>r3      s~   
      , , , , , , . . . . . .G G G G GT G G G G Gr&   