
    Vh	                     |    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
c mZ ej                   G d d             Zy)z7Experimental classes and functions used by ONNX export.    N)MappingSequence)Optional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   settypetorchnnModule     H/home/dcms/DCMS/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collections.abcr   r   typingr   r   r$   torch._C._onnx_C_onnxr   	dataclassr   r'   r(   r)   <module>r1      s<    =  - "      Q Q Qr(   