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                   @  s   e Zd ZeddddZdS )
ShrinkTestNone)returnc               	   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
dddddgt	jd}t	j
dddddgt	jd}t||g|gdd d S )NZShrinkxyg      ?)lambdZbias   )Znodesnameinputsoutputszbackend-test 
   )Zproducer_nameZopset_importsg       g      g        g      ?g       @)Zdtypeg      g      ?Ztest_shrink)r   r   r   )onnxhelperZ	make_nodeZ
make_graphZmake_tensor_value_infoZTensorProtoFLOATZmake_model_gen_versionZmake_opsetidnparrayZfloat32r   )nodegraphmodelr   r	    r   G/tmp/pip-unpacked-wheel-xnis5xre/onnx/backend/test/case/model/shrink.pyexport   s.    
zShrinkTest.exportN)__name__
__module____qualname__staticmethodr   r   r   r   r   r      s   r   )

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