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                   @  s0   e Zd ZeddddZeddddZdS )	MaxUnpoolNone)returnc                  C  s   t jjdddgdgddgddgd} tjddgdd	ggggtjd
}tjddgddggggtjd
}tjddddgddddgddddgdddd	ggggtjd
}t| ||g|gdd d S )Nr   xTxIy   inputsoutputsZkernel_shapestrides         Zdtype            r   Z*test_maxunpool_export_without_output_shaper   r   nameonnxhelperZ	make_nodenparrayZfloat32Zint64r   )noder   r	   r
    r    I/tmp/pip-unpacked-wheel-xnis5xre/onnx/backend/test/case/node/maxunpool.pyexport_without_output_shape   s&      .z%MaxUnpool.export_without_output_shapec               
   C  s   t jjddddgdgddgddgd} tjdd	gd
dggggtjd}tjdd
gddggggtjd}tjdtjd}tjdddddgdddd	dgdddddgdd
dddgdddddggggtjd}t| |||g|gdd d S )Nr   r   r	   output_shaper
   r   r   r      r      r   r   r   )r   r   r   r   r   Z'test_maxunpool_export_with_output_shaper   r   )r   r   r	   r#   r
   r    r    r!   export_with_output_shape$   s6      z"MaxUnpool.export_with_output_shapeN)__name__
__module____qualname__staticmethodr"   r&   r    r    r    r!   r      s   r   )

__future__r   Znumpyr   r   Zonnx.backend.test.case.baser   Zonnx.backend.test.case.noder   r   r    r    r    r!   <module>   s
   