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dS )
ShrinkTestreturnNonec               	   C  s   t jjddgdgddd} t jj| gdt jdt jjdggt jdt jjdggd}t jj|dt jd	d
gd}t	j
g dt	jd}t	j
g dt	jd}t||g|gdd d S )NShrinkxyg      ?)lambdbias   )nodesnameinputsoutputszbackend-test 
   )producer_nameopset_imports)g       g              g      ?g       @)dtype)g      r   r   r   g      ?test_shrink)r   r   r   )onnxhelper	make_node
make_graphmake_tensor_value_infoTensorProtoFLOATmake_model_gen_versionmake_opsetidnparrayfloat32r   )nodegraphmodelr	   r
    r(   `/home/ubuntu/sommelier/.venv/lib/python3.10/site-packages/onnx/backend/test/case/model/shrink.pyexport   s.   
zShrinkTest.exportN)r   r   )__name__
__module____qualname__staticmethodr*   r(   r(   r(   r)   r      s    r   )

__future__r   numpyr"   r   onnx.backend.test.case.baser   onnx.backend.test.case.modelr   r   r(   r(   r(   r)   <module>   s   