U
    Mh                     @   s^   d Z ddlZddlmZmZmZmZmZmZ ddl	Z	ddl
m  mZ ejG dd dZdS )z7Experimental classes and functions used by ONNX export.    N)MappingOptionalSequenceSetTypeUnionc                   @   s  e Zd ZU dZdZeed< dZeed< ej	j
Zej	ed< dZeee  ed< dZeee  ed	< ejjZejed
< dZee ed< dZeed< dZeeeeeeef ee f f  ed< dZee ed< dZeeeef  ed< dZeeeeej j!  f ed< dS )ExportOptionszArguments used by :func:`torch.onnx.export`.

    TODO: Adopt this in `torch.onnx.export` api to replace keyword arguments.
    Texport_paramsFverbosetrainingNinput_namesoutput_namesoperator_export_typeopset_versiondo_constant_foldingdynamic_axeskeep_initializers_as_inputscustom_opsetsexport_modules_as_functions)"__name__
__module____qualname____doc__r	   bool__annotations__r
   _C_onnxZTrainingModeZEVALr   r   r   r   strr   ZOperatorExportTypesZONNXr   r   intr   r   r   r   r   r   r   r   r   torchnnModule r!   r!   J/var/www/html/venv/lib/python3.8/site-packages/torch/onnx/_experimental.pyr   
   s   
,r   )r   dataclassestypingr   r   r   r   r   r   r   Ztorch._C._onnxZ_CZ_onnxr   	dataclassr   r!   r!   r!   r"   <module>   s    