o
    i                     @   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   )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 }
|s"ddg|
 }|s)dg|
 }|s0d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   j/var/www/html/eduruby.in/lip-sync/lip-sync-env/lib/python3.10/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>   s0    