o
    wim                     @  s@   d dl mZ d dlZd dlmZ d dlmZ G dd deZdS )    )annotationsN)OpRun)_conv_implementationc                   @  s&   e Zd Z								dddZdS )ConvIntegerNc              
   C  s   t |jdk rtd|j d|p| j}|p| j}|p| j}|p#| j}|	p(| j}	|
p-| j}
|	t
j}|r:||8 }|	t
j}|rF||8 }t||d |||||	|
		t
jfS )N   z3X must have at least 3 dimensions but its shape is .)lenshape
ValueErrorauto_pad	dilationsgroupkernel_shapepadsstridesastypenpint32r   )selfXWx_zero_pointw_zero_pointr   r   r   r   r   r    r   _/home/ubuntu/sommelier/.venv/lib/python3.10/site-packages/onnx/reference/ops/op_conv_integer.py_run   s*   





zConvInteger._run)NNNNNNNN)__name__
__module____qualname__r   r   r   r   r   r      s    r   )	
__future__r   numpyr   onnx.reference.op_runr   onnx.reference.ops.op_convr   r   r   r   r   r   <module>   s
   