
    ¯wg                     |    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c mZ ej                   G d d             Zy)z7Experimental classes and functions used by ONNX export.    N)MappingOptionalSequenceSetTypeUnionc                      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@                  jB                        f   ed<   y)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_onnxTrainingModeEVALr   r   r   r   strr   OperatorExportTypesONNXr   r   intr   r   r   r   r   r   r   r   r   torchnnModule     ]/home/mcse/projects/flask/flask-venv/lib/python3.12/site-packages/torch/onnx/_experimental.pyr
   r
   
   s   6 M4GT%,%9%9%>%>Hg"">+/K(3-(/,0L(8C=)08?8S8S8X8X'55X#'M8C=' $$TXL(73gc3h.?#.N(O#OPQX26$615M8GCH-.5KPtSehhoo1F-G'G!HPr(   r
   )r   dataclassestypingr   r   r   r   r   r   r$   torch._C._onnx_C_onnxr   	dataclassr
   r'   r(   r)   <module>r0      s<    =  @ @      Q Q Qr(   