o
    soi
                     @  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-   [/home/ubuntu/.local/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 jddgt j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   )dtype)r   r   r   r   r   r   r   r   )r	   r   r	   r	   ) r   r
   r   r
   r   r   r   r      r   r      r   r2      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   w_zero_pointsr   r   )r   r   padstest_convinteger_with_paddingr    r"   )r   r   r4   r   r   convinteger_node_with_paddingr-   r-   r.   export_with_padding&   s2   
$*

zConvInteger.export_with_paddingN)r   r   )__name__
__module____qualname__staticmethodr/   r8   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   