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edddZdS )		MaxUnpoolreturnNonec                  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g dg dg dg dgggtjd
}t| ||g|gdd d S )Nr   xTxIy   inputsoutputskernel_shapestrides         dtype            )r   r   r   r   )r   r   r   r   )r   r   r   r   *test_maxunpool_export_without_output_shaper   r   nameonnxhelper	make_nodenparrayfloat32int64r   )noder   r	   r
    r&   b/home/ubuntu/sommelier/.venv/lib/python3.10/site-packages/onnx/backend/test/case/node/maxunpool.pyexport_without_output_shape   s&     
z%MaxUnpool.export_without_output_shapec                  C  s   t jjdg d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g dg dg dg dg d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   r   r   r   r   )r   r   r   r*   r   )r   r   r   r+   r   '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)r   r   )__name__
__module____qualname__staticmethodr(   r-   r&   r&   r&   r'   r      s
    r   )

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