a
    ¢º”h	  ã                   @   s^   d Z ddlZddlmZmZ ddl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ÚSequence)ÚOptionalÚ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 )ÚExportOptionsz,Arguments used by :func:`torch.onnx.export`.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   ÚsetÚtypeÚtorchÚnnÚModule© r!   r!   úF/var/www/auris/lib/python3.9/site-packages/torch/onnx/_experimental.pyr      s   
,r   )r   ZdataclassesÚcollections.abcr   r   Útypingr   r   r   Ztorch._C._onnxZ_CZ_onnxr   Z	dataclassr   r!   r!   r!   r"   Ú<module>   s   