U
    <c                  
   @   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e ejejeedddZdS )a  This file exports ONNX ops for opset 17.

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

~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~
https://github.com/onnx/onnx/blob/main/docs/Changelog.md#version-17-of-the-default-onnx-operator-set
New operators:
    BlackmanWindow
    DFT
    HammingWindow
    HannWindow
    LayerNormalization
    MelWeightMatrix
    STFT
    SequenceMap
    N)Sequence)_C)symbolic_helper)	jit_utilsregistration
layer_norm   )Zopsetzaten::layer_normvisfnone)ginputnormalized_shapeweightbiasepscudnn_enablec                 C   s    t | }| jd|||||dS )NZLayerNormalization)Z	epsilon_fZaxis_i)lenop)r   r   r   r   r   r   r   Zaxis r   ?/tmp/pip-unpacked-wheel-gikjz4vx/torch/onnx/symbolic_opset17.pyr   !   s    
)__doc__	functoolstypingr   Ztorchr   Z
torch.onnxr   Ztorch.onnx._internalr   r   __all__partialZonnx_symbolicZ_onnx_symbolic
parse_argsZGraphContextValueintfloatboolr   r   r   r   r   <module>   s"   