o
    wi\                     @  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ZdS )UpsamplereturnNonec               	   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g dtjd}tjg dg dg dg dgggtjd}t| ||g|gdtddgd d S )Nr   XscalesYnearest)inputsoutputsmode            )dtype)      ?r   g       @g      @)r   r   r   r   r   r   )r   r   r   r   r   r   test_upsample_nearest 	   )r   r   nameopset_imports)onnxr   	make_nodenparrayfloat32r   make_opsetid)nodedatar
   output r$   a/home/ubuntu/sommelier/.venv/lib/python3.10/site-packages/onnx/backend/test/case/node/upsample.pyexport_nearest   s@   

zUpsample.export_nearestN)r   r   )__name__
__module____qualname__staticmethodr&   r$   r$   r$   r%   r      s    r   )
__future__r   numpyr   r   r   onnx.backend.test.case.baser   onnx.backend.test.case.noder   r   r$   r$   r$   r%   <module>   s   