o
    h                     @   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dejdejdejdee dee dee f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   )opsetzaten::col2imvisinputoutput_sizekernel_sizedilationpaddingstridec              	   C   sx   g }|D ]}t dD ]}	|| q
qt|d }
|s"ddg|
 }|s)dg|
 }|s0dg|
 }| jd||||||dS )N   r      Col2Im)dilations_ipads_i	strides_i)rangeappendr   _get_tensor_sizesop)gr   r   r   r   r   r   adjusted_paddingpad_num_dimensional_axis r!   Q/var/www/html/ai/venv/lib/python3.10/site-packages/torch/onnx/symbolic_opset18.pyr   "   s*   

)__doc__	functoolstypingr   torchr   
torch.onnxr   torch.onnx._internalr   r   __all__partialonnx_symbolic_onnx_symbolic
parse_argsbeartypeValueintr   r!   r!   r!   r"   <module>   s0    