
    Ph                         d dl mZmZ d dlmZ d dlmZmZmZm	Z	 dZ
dZ G d d      Z G d d	e      Z G d
 de      Z G d de      Ze G d d             Z ed       G d de             Ze G d d             Z ed       G d de             Z ed       G d de             Ze G d d             Z ed       G d de             Z ed       G d de             Ze G d  d!             Z ed       G d" d#e             Ze G d$ d%             Ze G d& d'             Z ed       G d( d)e             Ze G d* d+             Ze G d, d-             Ze G d. d/             Ze G d0 d1             Z e G d2 d3             Z!e G d4 d5             Z"e G d6 d7             Z#e G d8 d9e             Z$e G d: d;             Z%e G d< d=             Z&e G d> d?             Z'e G d@ dA             Z(e G dB dC             Z)e G dD dEe             Z*e G dF dG             Z+e G dH dI             Z,e G dJ dK             Z-e G dL dM             Z.e G dN dO             Z/e G dP dQ             Z0yR)S    )	dataclassfields)IntEnum)DictListOptionalTuple      c                   N    e Zd Zed        Zd Zed        Zed        Zd Z	d Z
y)_Unionc           
          t        |      dk(  sJ  | di i t        |       D ci c]  }|j                  d  c}|S c c}w )Nr    )lenr   name)clskwargsfs      eC:\Users\daisl\Desktop\realtime-object-detection\venv\Lib\site-packages\torch/_export/serde/schema.pycreatez_Union.create   sK    6{aGFvc{;{!{;FvFGG;s   ?c                 J     t         fdt               D              dk(  sJ y )Nc              3   P   K   | ]  }t        |j                        d  y w)Nr   getattrr   .0r   selfs     r   	<genexpr>z'_Union.__post_init__.<locals>.<genexpr>   s"     PlgdAFF.C.O1ls   &&r   )sumr   r   s   `r   __post_init__z_Union.__post_init__   s!    PfTlPPTUUUU    c                 L     t         fdt               D        d       }|J |S )Nc              3   x   K   | ]1  }t        |j                        t        |j                         3 y wNr   r   s     r   r   z_Union.value.<locals>.<genexpr>   s0     e<a74QRQWQWCXCdGD!&&)<s   ::nextr   )r   vals   ` r   valuez_Union.value   s)    e6$<egkl
r"   c                 L     t         fdt               D        d       }|J |S )Nc              3   d   K   | ]'  }t        |j                        |j                   ) y wr%   r   r   s     r   r   z_Union.type.<locals>.<genexpr>   s&     [Aqvv9N9Zs   00r&   )r   val_types   ` r   typez_Union.type   s+    [[]ab###r"   c                 "    | j                         S r%   )__repr__r    s    r   __str__z_Union.__str__#   s    }}r"   c                 f    t        |       j                   d| j                    d| j                   dS )N(=))r-   __name__r)   r    s    r   r/   z_Union.__repr__&   s.    t*%%&a		{!DJJ<qAAr"   N)r5   
__module____qualname__classmethodr   r!   propertyr)   r-   r0   r/   r   r"   r   r   r      sO    H HV  
  
Br"   r   c                   D    e Zd ZdZdZdZdZdZdZdZ	dZ
d	Zd
ZdZdZdZdZy)
ScalarTyper   r   r
                     	   
            N)r5   r6   r7   UNKNOWNBYTECHARSHORTINTLONGHALFFLOATDOUBLECOMPLEXHALFCOMPLEXFLOATCOMPLEXDOUBLEBOOLBFLOAT16r   r"   r   r;   r;   *   sH    GDDE
CDDEFKLMDHr"   r;   c                   ,    e Zd ZdZdZdZdZdZdZdZ	dZ
y	)
Layoutr   r   r
   r<   r=   r>   r?   r@   N)r5   r6   r7   Unknown	SparseCoo	SparseCsr	SparseCsc	SparseBsr	SparseBsc_mkldnnStridedr   r"   r   rV   rV   ;   s*    GIIIIIGGr"   rV   c                        e Zd ZdZdZdZdZdZy)MemoryFormatr   r   r
   r<   r=   N)r5   r6   r7   rW   ContiguousFormatChannelsLastChannelsLast3dPreserveFormatr   r"   r   r`   r`   F   s    GLNNr"   r`   c                   (    e Zd ZU eed<   ee   ed<   y)Devicer-   indexN)r5   r6   r7   str__annotations__r   intr   r"   r   rf   rf   N   s    
IC=r"   rf   F)reprc                   ,    e Zd ZU eed<   eed<   eed<   y)SymExprHintas_intas_floatas_boolN)r5   r6   r7   rj   ri   floatboolr   r"   r   rm   rm   T   s    KOMr"   rm   c                   ,    e Zd ZU eed<   dZee   ed<   y)SymExprexpr_strNhint)r5   r6   r7   rh   ri   rv   r   rm   r   r"   r   rt   rt   _   s    M"&D(;
&r"   rt   c                   "    e Zd ZU eed<   eed<   y)SymIntas_exprrn   N)r5   r6   r7   rt   ri   rj   r   r"   r   rx   rx   e   s    Kr"   rx   c                   "    e Zd ZU eed<   eed<   y)SymBoolry   rp   N)r5   r6   r7   rt   ri   rr   r   r"   r   r{   r{   k   s    Mr"   r{   c                   `    e Zd ZU eed<   ee   ed<   eed<   eed<   ee   ed<   e	ed<   e
ed<   y)	
TensorMetadtypesizesrequires_graddevicestridesstorage_offsetlayoutN)r5   r6   r7   r;   ri   r   rx   rr   rf   rj   rV   r   r"   r   r}   r}   q   s2    <N&\Nr"   r}   c                   "    e Zd ZU eed<   eed<   y)SymIntArgumentas_namern   N)r5   r6   r7   rh   ri   rj   r   r"   r   r   r      s    LKr"   r   c                   "    e Zd ZU eed<   eed<   y)SymBoolArgumentr   rp   N)r5   r6   r7   rh   ri   rr   r   r"   r   r   r      s    LMr"   r   c                       e Zd ZU eed<   y)TensorArgumentr   Nr5   r6   r7   rh   ri   r   r"   r   r   r          
Ir"   r   c                   (    e Zd ZU eed<   ed   ed<   y)OptionalTensorArgument	as_tensorr   as_noneN)r5   r6   r7   rh   ri   r	   r   r"   r   r   r      s    N2Yr"   r   c                   "    e Zd ZU eed<   ded<   y)GraphArgumentr   GraphgraphNr   r   r"   r   r   r      s    
INr"   r   c                       e Zd ZU eed<   y)CustomObjArgumentr   Nr   r   r"   r   r   r      r   r"   r   c                       e Zd ZU ed   ed<   eed<   ee   ed<   eed<   ee   ed<   eed<   ee   ed<   e	ed	<   ee	   ed
<   e
ed<   ee
   ed<   eed<   eed<   eed<   eed<   eed<   ee   ed<   eed<   ee   ed<   eed<   ee   ed<   eed<   y)Argumentr   r   r   
as_tensorsrn   as_intsro   	as_floats	as_string
as_strings
as_sym_intas_sym_intsas_scalar_typeas_memory_format	as_layout	as_devicerp   as_boolsas_sym_boolas_sym_boolsas_graphas_optional_tensorsas_custom_objN)r5   r6   r7   r	   ri   r   r   rj   rq   rh   r   r;   r`   rV   rf   rr   r   r   r   r   r   r"   r   r   r      s    2Y^$$K#YOE{NS	n%%""M4j  ''455$$r"   r   c                   "    e Zd ZU eed<   eed<   y)NamedArgumentr   argN)r5   r6   r7   rh   ri   r   r   r"   r   r   r      s     I	Mr"   r   c                   L    e Zd ZU eed<   ee   ed<   ee   ed<   eeef   ed<   y)NodetargetinputsoutputsmetadataN)	r5   r6   r7   rh   ri   r   r   r   r   r   r"   r   r   r      s*    K(^38nr"   r   c                       e Zd ZU ee   ed<   ee   ed<   ee   ed<   eee	f   ed<   eee
f   ed<   eeef   ed<   dZeed<   y	)
r   r   r   nodestensor_valuessym_int_valuessym_bool_valuesFis_single_tensor_returnN)r5   r6   r7   r   r   ri   r   r   rh   r}   rx   r{   r   rr   r   r"   r   r   r      sZ    N(^:Z((f%%#w,''
 %*T)r"   r   c                       e Zd ZU eed<   y)UserInputSpecr   Nr5   r6   r7   r   ri   r   r"   r   r   r      s	     
Mr"   r   c                   "    e Zd ZU eed<   eed<   y)InputToParameterSpecr   parameter_nameNr5   r6   r7   r   ri   rh   r   r"   r   r   r          	r"   r   c                   "    e Zd ZU eed<   eed<   y)InputToBufferSpecr   buffer_nameNr   r   r"   r   r   r          	r"   r   c                   "    e Zd ZU eed<   eed<   y)InputToTensorConstantSpecr   tensor_constant_nameNr   r   r"   r   r   r      s    	r"   r   c                   6    e Zd ZU eed<   eed<   eed<   eed<   y)	InputSpec
user_input	parameterbuffertensor_constantN)r5   r6   r7   r   ri   r   r   r   r   r"   r   r   r      s    ##..r"   r   c                       e Zd ZU eed<   y)UserOutputSpecr   Nr   r   r"   r   r   r     s    	Mr"   r   c                       e Zd ZU eed<   y)LossOutputSpecr   N)r5   r6   r7   r   ri   r   r"   r   r   r     s    	r"   r   c                   "    e Zd ZU eed<   eed<   y)BufferMutationSpecr   r   Nr   r   r"   r   r   r     r   r"   r   c                   "    e Zd ZU eed<   eed<   y)GradientToParameterSpecr   r   Nr   r   r"   r   r   r     r   r"   r   c                   "    e Zd ZU eed<   eed<   y)GradientToUserInputSpecr   user_input_nameNr   r   r"   r   r   r     s    	r"   r   c                   @    e Zd ZU eed<   eed<   eed<   eed<   eed<   y)
OutputSpecuser_outputloss_outputbuffer_mutationgradient_to_parametergradient_to_user_inputN)	r5   r6   r7   r   ri   r   r   r   r   r   r"   r   r   r   #  s     ''2233r"   r   c                   .    e Zd ZU ee   ed<   ee   ed<   y)GraphSignatureinput_specsoutput_specsN)r5   r6   r7   r   r   ri   r   r   r"   r   r   r   ,  s    i z""r"   r   c                   "    e Zd ZU eed<   eed<   y)RangeConstraintmin_valmax_valN)r5   r6   r7   rj   ri   r   r"   r   r   r   2  s    LLr"   r   c                   B    e Zd ZU ee   ed<   ee   ed<   eed<   eed<   y)ModuleCallSignaturer   r   in_specout_specN)r5   r6   r7   r   r   ri   rh   r   r"   r   r   r   8  s"    N(^ LMr"   r   c                   ,    e Zd ZU eed<   dZee   ed<   y)ModuleCallEntryfqnN	signature)r5   r6   r7   rh   ri   r   r   r   r   r"   r   r   r   C  s    	H/3Ix+,3r"   r   c                   2    e Zd ZU eed<   eed<   ee   ed<   y)GraphModuler   r   module_call_graphN)r5   r6   r7   r   ri   r   r   r   r   r"   r   r   r   I  s    L O,,r"   r   c                   T    e Zd ZU eed<   eeef   ed<   eeef   ed<   eed<   eed<   y)ExportedProgramgraph_moduleopset_versionrange_constraintsschema_versiondialectN)	r5   r6   r7   r   ri   r   rh   rj   r   r   r"   r   r   r   S  s1    S>!C011Lr"   r   N)1dataclassesr   r   enumr   typingr   r   r   r	   SCHEMA_VERSIONTREESPEC_VERSIONr   r;   rV   r`   rf   rm   rt   rx   r{   r}   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r"   r   <module>r      s   *  . .  B B8 "W 7    
 &   ' ' '
 V  
 f  
     V   f  
    V  
   
   
 %v % %2       * * *   
   
   
   
 / / /         
   
   
 4 4 4 # # #
   
    4 4 4
 - - -   r"   