U
    rhu                     @  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 )ConvIntegerNone)returnc                  C  s   t ddddddddd	g	t jd
} t d}t ddddgt jd}t ddddgt jdddd}tjjddddgdgd}t	|| ||g|gdd d S )N                     	   
      r   r	   r	   r   r   r   r   r               r   xwx_zero_pointy)inputsoutputsZ test_convinteger_without_paddingr   r   name
nparrayZastypeZuint8ZreshapeZint32onnxhelperZ	make_noder   )r   r   r   r   Zconvinteger_node r%   K/tmp/pip-unpacked-wheel-xnis5xre/onnx/backend/test/case/node/convinteger.pyexport_without_padding   s(    
 &  z"ConvInteger.export_without_paddingc                  C  s   t ddddddddd	g	t jd
} t d}t ddddgt jd}t ddddddddddddddddgt jd}tjjddddgdgddddgd}t	|| ||g|gdd d S )Nr   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   ZpadsZtest_convinteger_with_paddingr   r    )r   r   r   r   Zconvinteger_node_with_paddingr%   r%   r&   export_with_padding&   s4    
 *
zConvInteger.export_with_paddingN)__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
   