
    sg!                        d dl mZmZ d dlmZ d dlmZmZmZm	Z	 d dl
mZ dZ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 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/ d0             Z!e G d1 d2             Z" ed       G d3 d4e             Z#e G d5 d6             Z$e G d7 d8             Z%e G d9 d:             Z&e G d; d<             Z'e G d= d>             Z(e G d? d@             Z) ed       G dA dBe             Z*e G dC dD             Z+e G dE dF             Z,e G dG dH             Z-e G dI dJ             Z.e G dK dL             Z/e G dM dN             Z0e G dO dP             Z1 ed       G dQ dRe             Z2e G dS dT             Z3e G dU dV             Z4e G dW dX             Z5e G dY dZ             Z6e G d[ d\             Z7e G d] d^             Z8e G d_ d`             Z9ya)b    )	dataclassfield)IntEnum)DictListOptionalTuple)_Union)         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            r      	   
            N)__name__
__module____qualname__UNKNOWNBYTECHARSHORTINTLONGHALFFLOATDOUBLECOMPLEXHALFCOMPLEXFLOATCOMPLEXDOUBLEBOOLBFLOAT16     M/var/www/html/venv/lib/python3.12/site-packages/torch/_export/serde/schema.pyr   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)r   r   r   Unknown	SparseCoo	SparseCsr	SparseCsc	SparseBsr	SparseBsc_mkldnnStridedr+   r,   r-   r/   r/       s*    GIIIIIGGr,   r/   c                        e Zd ZdZdZdZdZdZy)MemoryFormatr   r   r   r   r   N)r   r   r   r0   ContiguousFormatChannelsLastChannelsLast3dPreserveFormatr+   r,   r-   r9   r9   +   s    GLNNr,   r9   c                   ,    e Zd ZU eed<   dZee   ed<   y)DevicetypeNindex)r   r   r   str__annotations__rA   r   intr+   r,   r-   r?   r?   3   s    
IE8C=r,   r?   F)reprc                   ,    e Zd ZU eed<   eed<   eed<   y)SymExprHintas_intas_floatas_boolN)r   r   r   rD   rC   floatboolr+   r,   r-   rG   rG   9   s    KOMr,   rG   c                   ,    e Zd ZU eed<   dZee   ed<   y)SymExprexpr_strNhint)r   r   r   rB   rC   rP   r   rG   r+   r,   r-   rN   rN   D   s    M"&D(;
&r,   rN   c                   "    e Zd ZU eed<   eed<   y)SymIntas_exprrH   N)r   r   r   rN   rC   rD   r+   r,   r-   rR   rR   J   s    Kr,   rR   c                   "    e Zd ZU eed<   eed<   y)SymBoolrS   rJ   N)r   r   r   rN   rC   rL   r+   r,   r-   rU   rU   P   s    Mr,   rU   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)
r   r   r   r   rC   r   rR   rL   r?   r/   r+   r,   r-   rW   rW   V   s2    <N&\Nr,   rW   c                   "    e Zd ZU eed<   eed<   y)SymIntArgumentas_namerH   N)r   r   r   rB   rC   rD   r+   r,   r-   r`   r`   g   s    LKr,   r`   c                   "    e Zd ZU eed<   eed<   y)SymBoolArgumentra   rJ   N)r   r   r   rB   rC   rL   r+   r,   r-   rc   rc   s   s    LMr,   rc   c                       e Zd ZU eed<   y)TensorArgumentnameNr   r   r   rB   rC   r+   r,   r-   re   re   y       
Ir,   re   c                       e Zd ZU eed<   y)TokenArgumentrf   Nrg   r+   r,   r-   rj   rj   ~   rh   r,   rj   c                   (    e Zd ZU eed<   ed   ed<   y)OptionalTensorArgument	as_tensorr+   as_noneN)r   r   r   re   rC   r	   r+   r,   r-   rl   rl      s    2Yr,   rl   c                   "    e Zd ZU eed<   ded<   y)GraphArgumentrf   GraphgraphNrg   r+   r,   r-   rp   rp      s    
INr,   rp   c                   "    e Zd ZU eed<   eed<   y)CustomObjArgumentrf   	class_fqnNrg   r+   r,   r-   rt   rt      s    
INr,   rt   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<   e	ed<   y)Argumentr+   rn   rm   
as_tensorsrH   as_intsrI   	as_floats	as_string
as_strings
as_sym_intas_sym_intsas_scalar_typeas_memory_format	as_layout	as_devicerJ   as_boolsas_sym_boolas_sym_boolsas_graphas_optional_tensorsas_custom_objas_operatorN)r   r   r   r	   rC   re   r   rD   rK   rB   r`   r   r9   r/   r?   rL   rc   rp   rl   rt   r+   r,   r-   rw   rw      s    2Y^$$K#YOE{NS	n%%""M4j  ''455$$r,   rw   c                   "    e Zd ZU eed<   eed<   y)NamedArgumentrf   argN)r   r   r   rB   rC   rw   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)	r   r   r   rB   rC   r   r   rw   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<    ee	      Zeeef   ed
<   y)rq   r   r   nodestensor_valuessym_int_valuessym_bool_valuesFis_single_tensor_returndefault_factorycustom_obj_valuesN)r   r   r   r   rw   rC   r   r   rB   rW   rR   rU   r   rL   r   dictr   rt   r+   r,   r-   rq   rq      su    N(^:Z((f%%#w,''
 %*T)6;D6QtC!223Qr,   rq   c                       e Zd ZU eed<   y)UserInputSpecr   Nr   r   r   rw   rC   r+   r,   r-   r   r      s	     
Mr,   r   c                   F    e Zd ZU ed   ed<   eed<   eed<   eed<   eed<   y)ConstantValuer+   rn   rH   rI   r{   rJ   N)	r   r   r   r	   rC   rD   rK   rB   rL   r+   r,   r-   r   r      s     2YKONMr,   r   c                   "    e Zd ZU eed<   eed<   y)ConstantInputSpecrf   valueN)r   r   r   rB   rC   r   r+   r,   r-   r   r      s    
Ir,   r   c                   "    e Zd ZU eed<   eed<   y)InputToParameterSpecr   parameter_nameNr   r   r   re   rC   rB   r+   r,   r-   r   r          	r,   r   c                   ,    e Zd ZU eed<   eed<   eed<   y)InputToBufferSpecr   buffer_name
persistentN)r   r   r   re   rC   rB   rL   r+   r,   r-   r   r      s    	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                   "    e Zd ZU eed<   eed<   y)InputToCustomObjSpecr   custom_obj_nameN)r   r   r   rt   rC   rB   r+   r,   r-   r   r      s    	r,   r   c                       e Zd ZU eed<   y)InputTokenSpecr   Nr   r   r   rj   rC   r+   r,   r-   r   r         	r,   r   c                   T    e Zd ZU eed<   eed<   eed<   eed<   eed<   e	ed<   e
ed<   y)		InputSpec
user_input	parameterbuffertensor_constant
custom_objtokenconstant_inputN)r   r   r   r   rC   r   r   r   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)r   r   r   re   rC   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     s    	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   )      	r,   r   c                   "    e Zd ZU eed<   eed<   y)UserInputMutationSpecr   r   Nr   r+   r,   r-   r   r   /  r   r,   r   c                       e Zd ZU eed<   y)OutputTokenSpecr   Nr   r+   r,   r-   r   r   5  r   r,   r   c                   T    e Zd ZU eed<   eed<   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_inputuser_input_mutationr   N)r   r   r   r   rC   r   r   r   r   r   r   r+   r,   r-   r   r   :  s,    ''2233..r,   r   c                   .    e Zd ZU ee   ed<   ee   ed<   y)GraphSignatureinput_specsoutput_specsN)r   r   r   r   r   rC   r   r+   r,   r-   r   r   E  s    i z""r,   r   c                   "    e Zd ZU eed<   eed<   y)RangeConstraintmin_valmax_valNr   r   r   rD   rC   r+   r,   r-   r   r   K  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)r   r   r   r   rw   rC   rB   r+   r,   r-   r   r   Q  s"    N(^ LMr,   r   c                   ,    e Zd ZU eed<   dZee   ed<   y)ModuleCallEntryfqnN	signature)r   r   r   rB   rC   r   r   r   r+   r,   r-   r   r   \  s    	H/3Ix+,3r,   r   c                   X    e Zd ZU eed<   eed<   ee   ed<    ee	      Z
eeef   ed<   y)GraphModulerr   r   module_call_graphr   r   N)r   r   r   rq   rC   r   r   r   r   r   r   r   rB   r+   r,   r-   r   r   b  s1    L O,,$T:Hd38n:r,   r   c                   "    e Zd ZU eed<   eed<   y)SchemaVersionmajorminorNr   r+   r,   r-   r   r   o  s    JJr,   r   c                   z    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      Zee   ed<   dZeed<   y	)
ExportedProgramgraph_moduleopset_versionrange_constraintsschema_versionr   	verifiersz<=2.4torch_versionN)r   r   r   r   rC   r   rB   rD   r   r   r   listr   r   r   r+   r,   r-   r   r   u  sI    S>!C011!! 6ItCy6 M3 r,   r   N):dataclassesr   r   enumr   typingr   r   r   r	   torch._export.serde.unionr
   SCHEMA_VERSIONTREESPEC_VERSIONr   r/   r9   r?   rG   rN   rR   rU   rW   r`   rc   re   rj   rl   rp   rt   rw   r   r   rq   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   )  . . ,   "W 7       
 &   ' ' '
 V  
 f  
     V   f  
       V  
   
    v  4       R R R   
 F     
   
      
   
    & & &         
   
   
   
       # # #
   
    4 4 4
 ; ; ;   
 ! ! !r,   