U
    rh\                     @  sT   d dl mZ d dlZd dlZd dlmZ d dlmZ d dlm	Z	 G dd deZ
dS )    )annotationsN)helper)Base)expectc                   @  s   e Zd ZeddddZdS )UpsampleNone)returnc               
   C  s   t jjdddgdgdd} tjddgd	d
ggggtjd}tjddddgtjd}tjddddddgddddddgd	d	d	d
d
d
gd	d	d	d
d
d
ggggtjd}t| ||g|gdtddgd d S )Nr   XscalesYZnearest)inputsoutputsmode            )Zdtypeg      ?g       @g      @Ztest_upsample_nearest 	   )r   r   nameZopset_imports)onnxr   Z	make_nodenparrayZfloat32r   Zmake_opsetid)nodedatar
   output r   H/tmp/pip-unpacked-wheel-xnis5xre/onnx/backend/test/case/node/upsample.pyexport_nearest   s@    
zUpsample.export_nearestN)__name__
__module____qualname__staticmethodr   r   r   r   r   r      s   r   )
__future__r   Znumpyr   r   r   Zonnx.backend.test.case.baser   Zonnx.backend.test.case.noder   r   r   r   r   r   <module>   s   