o
    wiu                     @  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 )	ConvIntegerreturnNonec                  C  s   t g dt jd} t d}t g dt jd}t g dt jdddd}tjjdg d	d
gd}t	|| ||g|gdd d S )N	                     	   
      r   r
   r
   r   r   r   r   r   r   r   r	   r	   )            r	   r   xwx_zero_pointy)inputsoutputs test_convinteger_without_paddingr   r    name
nparrayastypeuint8reshapeint32onnxhelper	make_noder   )r   r   r   r   convinteger_node r/   d/home/ubuntu/sommelier/.venv/lib/python3.10/site-packages/onnx/backend/test/case/node/convinteger.pyexport_without_padding   s    
"
z"ConvInteger.export_without_paddingc                  C  s   t g dt jd} t d}t g dt jd}t g dt jd}tjjdg d	d
gg dd}t	|| ||g|gdd d S )Nr   r   r   r   r   )r   r
   r   r
   r   r   r   r      r   r      r   r3      r   )r   r   r   r   r   r   r   )r   r    padstest_convinteger_with_paddingr"   r$   )r   r   r   r   convinteger_node_with_paddingr/   r/   r0   export_with_padding&   s,   

zConvInteger.export_with_paddingN)r   r   )__name__
__module____qualname__staticmethodr1   r8   r/   r/   r/   r0   r      s
    r   )

__future__r   numpyr%   r+   onnx.backend.test.case.baser   onnx.backend.test.case.noder   r   r/   r/   r/   r0   <module>   s   