U
    9%e                  
   @   s   d Z ddlZddlmZ ddlmZ ddlmZ ddlm	Z	m
Z
 dgZeje
jdd	Zed
edddddde	jejejejee ee ee dddZdS )a  This file exports ONNX ops for opset 18.

Note [ONNX Operators that are added/updated in opset 18]

~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~
https://github.com/onnx/onnx/blob/main/docs/Changelog.md#version-18-of-the-default-onnx-operator-set
New operators:
    CenterCropPad
    Col2Im
    Mish
    OptionalGetElement
    OptionalHasElement
    Pad
    Resize
    ScatterElements
    ScatterND
    N)Sequence)_C)symbolic_helper)	_beartyperegistrationcol2im   )Zopsetzaten::col2imvis)inputoutput_sizekernel_sizedilationpaddingstridec              	   C   sx   g }|D ]}t dD ]}	|| qqt|d }
|sDddg|
 }|sRdg|
 }|s`dg|
 }| jd||||||dS )N   r      ZCol2Im)Zdilations_iZpads_iZ	strides_i)rangeappendr   Z_get_tensor_sizesop)gr   r   r   r   r   r   Zadjusted_paddingpad_Znum_dimensional_axis r   Z/var/www/html/Darija-Ai-API/env/lib/python3.8/site-packages/torch/onnx/symbolic_opset18.pyr   "   s(    

)__doc__	functoolstypingr   Ztorchr   Z
torch.onnxr   Ztorch.onnx._internalr   r   __all__partialZonnx_symbolicZ_onnx_symbolic
parse_argsZbeartypeValueintr   r   r   r   r   <module>   s"   