
    9jfA                     
   U d dl mZmZ d dlmZ d dlmZ d dlmZm	Z	 dZ
dZ G d de      Z G d	 d
e      Z G d de      Zi ej                  dej                   dej"                  dej$                  dej&                  dej(                  dej*                  dej,                  dej.                  dej0                  dej2                  dej4                  dej6                  dej8                  dej:                  dej<                  dej>                  dej@                  dejB                  dejD                  d ejF                  d!iZ$e%e&e'f   e(d"<   ejR                  d#ejT                  d$ejV                  d%ejX                  d&ejZ                  d'ej\                  d(ej^                  d)iZ0e%e&e'f   e(d*<   ejb                  d+ejd                  d,ejf                  d-ejh                  d.iZ5e%e&e'f   e(d/<   e G d0 d1             Z6e	 G d2 d3e             Z7e G d4 d5             Z8e	 G d6 d7e             Z9e	 G d8 d9e             Z:e	 G d: d;e             Z;e G d< d=             Z<e	 G d> d?e             Z=e	 G d@ dAe             Z>e	 G dB dCe             Z?e G dD dE             Z@e G dF dG             ZAe	 G dH dIe             ZBe G dJ dK             ZCe G dL dM             ZDe G dN dO             ZEe	 G dP dQe             ZF G dR dSe      ZGe G dT dU             ZHe G dV dW             ZIe G dX dY             ZJe G dZ d[             ZKe	 G d\ d]e             ZLe G d^ d_             ZMe G d` da             ZNe G db dc             ZOe G dd de             ZPe G df dg             ZQe G dh di             ZRe	 G dj dke             ZSe G dl dm             ZTe G dn do             ZUe G dp dq             ZVe G dr ds             ZWe G dt du             ZXe G dv dw             ZYe G dx dy             ZZe G dz d{             Z[e	 G d| d}e             Z\e G d~ d             Z]e G d d             Z^e G d d             Z_e G d d             Z`e G d d             Zae G d d             Zbe G d d             Zce G d d             Zde G d d             Zee G d d             Zfe G d d             Zge G d d             Zhe G d d             Ziy)    )	dataclassfield)IntEnum)	Annotated)_Union_union_dataclass)         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dZdZdZdZdZdZdZdZy)
ScalarTyper   r                     r	   	   
                            !   "   #   N)__name__
__module____qualname__UNKNOWNBYTECHARSHORTINTLONGHALFFLOATDOUBLECOMPLEXHALFCOMPLEXFLOATCOMPLEXDOUBLEBOOLBFLOAT16UINT16FLOAT8E4M3FN
FLOAT8E5M2FLOAT8E4M3FNUZFLOAT8E5M2FNUZFLOAT8E8M0FNUUINT32UINT64     Z/media/conek/DATA/Code/OCR/venv/lib/python3.12/site-packages/torch/_export/serde/schema.pyr   r      sp    GDDE
CDDEFKLMDHFLJNNMFFr;   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>   0   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#   r?   ContiguousFormatChannelsLastChannelsLast3dPreserveFormatr:   r;   r<   rH   rH   ;   s    GLNNr;   rH   ByteCharShortIntLongHalfFloatDoubleComplexHalfComplexFloatComplexDoubleBoolBFloat16UInt16Float8_e4m3fnFloat8_e5m2Float8_e4m3fnuzFloat8_e5m2fnuzFloat8_e8m0fnuUInt32UInt64SCALAR_TYPE_TO_C10SparserA   rB   rC   rD   MkldnnrF   LAYOUT_TO_C10
ContiguousrJ   rK   PreserveMEMORY_FORMAT_TO_C10c                   @    e Zd ZU eedf   ed<   dZeedz  df   ed<   y)Devicer   typeNr
   index)r!   r"   r#   r   str__annotations__rl   intr:   r;   r<   rj   rj   m   s(    
CG
'+E9S4Z^$+r;   rj   c                   J    e Zd ZU eedf   ed<   eedf   ed<   eedf   ed<   y)SymExprHintr   as_intr
   as_boolr   as_floatN)r!   r"   r#   r   ro   rn   boolfloatr:   r;   r<   rq   rq   s   s/    c2gtRx  r	""r;   rq   c                   @    e Zd ZU eedf   ed<   dZeedz  df   ed<   y)SymExprr   expr_strNr
   hint)r!   r"   r#   r   rm   rn   rz   rq   r:   r;   r<   rx   rx   ~   s*    R  .2D)K$&*
+2r;   rx   c                   6    e Zd ZU eedf   ed<   eedf   ed<   y)SymIntr   as_exprr
   rr   N)r!   r"   r#   r   rx   rn   ro   r:   r;   r<   r|   r|      s     w{##c2gr;   r|   c                   6    e Zd ZU eedf   ed<   eedf   ed<   y)SymFloatr   r}   r
   rt   N)r!   r"   r#   r   rx   rn   rv   r:   r;   r<   r   r      s     w{##r	""r;   r   c                   6    e Zd ZU eedf   ed<   eedf   ed<   y)SymBoolr   r}   r
   rs   N)r!   r"   r#   r   rx   rn   ru   r:   r;   r<   r   r      s     w{##tRx  r;   r   c                       e Zd ZU eedf   ed<   eee   df   ed<   eedf   ed<   ee	df   ed<   eee   d	f   ed
<   eedf   ed<   ee
df   ed<   y)
TensorMetar   dtyper
   sizesr   requires_grad(   device2   strides<   storage_offsetF   layoutN)r!   r"   r#   r   r   rn   listr|   ru   rj   r>   r:   r;   r<   r   r      su    Z^$$T&\2%&&T2X&&fbj!!tF|R'((fbj))fbj!!r;   r   c                   6    e Zd ZU eedf   ed<   eedf   ed<   y)SymIntArgumentr   as_namer
   rr   N)r!   r"   r#   r   rm   rn   ro   r:   r;   r<   r   r      s     sBwc2gr;   r   c                   6    e Zd ZU eedf   ed<   eedf   ed<   y)SymFloatArgumentr   r   r
   rt   N)r!   r"   r#   r   rm   rn   rv   r:   r;   r<   r   r      s     sBwr	""r;   r   c                   6    e Zd ZU eedf   ed<   eedf   ed<   y)SymBoolArgumentr   r   r
   rs   N)r!   r"   r#   r   rm   rn   ru   r:   r;   r<   r   r      s     sBwtRx  r;   r   c                   "    e Zd ZU eedf   ed<   y)TensorArgumentr   nameNr!   r"   r#   r   rm   rn   r:   r;   r<   r   r          
CG
r;   r   c                   "    e Zd ZU eedf   ed<   y)TokenArgumentr   r   Nr   r:   r;   r<   r   r      r   r;   r   c                   6    e Zd ZU eedf   ed<   eedf   ed<   y)OptionalTensorArgumentr
   	as_tensorr   as_noneN)r!   r"   r#   r   r   rn   ru   r:   r;   r<   r   r      s!    +,,tRx  r;   r   c                   2    e Zd ZU eedf   ed<   ed   ed<   y)GraphArgumentr   r   )Graphr
   graphNr   r:   r;   r<   r   r      s    
CG
[!!r;   r   c                   6    e Zd ZU eedf   ed<   eedf   ed<   y)CustomObjArgumentr   r   r
   	class_fqnNr   r:   r;   r<   r   r      s     
CG
b!!r;   r   c                   6    e Zd ZU eedf   ed<   eedf   ed<   y)ComplexValuer   realr
   imagN)r!   r"   r#   r   rv   rn   r:   r;   r<   r   r      s     
E2I

E2I
r;   r   c                      e Zd ZU eedf   ed<   eedf   ed<   eee   df   ed<   eedf   ed<   eee   d	f   ed
<   ee	df   ed<   eee	   df   ed<   ee
df   ed<   eee
   df   ed<   eedf   ed<   eee   df   ed<   eedf   ed<   eedf   ed<   eedf   ed<   eedf   ed<   eedf   ed <   eee   d!f   ed"<   eed#f   ed$<   eee   d%f   ed&<   eed'f   ed(<   eee   d)f   ed*<   eed+f   ed,<   ee
d-f   ed.<   eed/f   ed0<   eee   d1f   ed2<   eed3f   ed4<   eed5f   ed6<   eeee      d7f   ed8<   eeee      d9f   ed:<   eee
d f   d;f   ed<<   eeee	      d=f   ed><   y?)@Argumentr   r   r
   r   r   
as_tensorsr   rr   r   as_intsP   rt   Z   	as_floatsd   	as_stringe   
as_stringsn   
as_sym_intx   as_sym_ints   as_scalar_type   as_memory_format   	as_layout   	as_device   rs      as_bools   as_sym_bool   as_sym_bools   as_graph   as_optional_tensors   as_custom_obj   as_operator   as_sym_float   as_sym_floats   as_optional_tensori  
as_complexi  as_nested_tensorsi  as_int_listsi"  as_string_to_argumenti,  as_float_listsN)r!   r"   r#   r   ru   rn   r   r   ro   rv   rm   r   r   rH   r>   rj   r   r   r   r   r   r   dictr:   r;   r<   r   r      s7   tRx  +,,$~.233c2gtCy"}%%r	""eb))c""$s)S.)).#-..4/455j#o..c 122%%%%tSy!!T
C((?C/00D13677s*++"4(>#?#DEE.34438$$,c122T"23S899!"8#"=>>,+,, d>&:!;S!@AADcOS011$T#z/%:C%?@@d4;/455r;   r   c                       e Zd ZdZdZdZy)ArgumentKindr   r   r   N)r!   r"   r#   r$   
POSITIONALKEYWORDr:   r;   r<   r   r     s    GJGr;   r   c                   T    e Zd ZU eedf   ed<   eedf   ed<   dZeedz  df   ed<   y)NamedArgumentr   r   r
   argNr   kind)	r!   r"   r#   r   rm   rn   r   r   r   r:   r;   r<   r   r     s;     CG
	8R<	  /3D)L4'+
,3r;   r   c                       e Zd ZU eedf   ed<   eee   df   ed<   eee   df   ed<   ee	eef   df   ed<   d	Z
eed	z  d
f   ed<   d	Zeed	z  df   ed<   y	)Noder   targetr
   inputsr   outputsr   metadataNr   is_hop_single_tensor_returnr   r   )r!   r"   r#   r   rm   rn   r   r   r   r   r   ru   r   r:   r;   r<   r   r     s    c2gd=)2-..tH~r)**S#X*++>B4$;?!;B&*D)C$JN
#*r;   r   c                   .   e Zd ZU eee   df   ed<   eee   df   ed<   eee   df   ed<   eee	e
f   df   ed<   eee	ef   d	f   ed
<   eee	ef   df   ed<   dZeedf   ed<    ee      Zeee	ef   df   ed<    ee      Zeee	ef   df   ed<   y)r   r   r   r
   r   r   nodesr   tensor_valuesr   sym_int_valuesr   sym_bool_valuesFr   is_single_tensor_returndefault_factoryr   custom_obj_valuesr   sym_float_valuesN)r!   r"   r#   r   r   r   rn   r   r   rm   r   r|   r   r   ru   r   r   r   r   r   r:   r;   r<   r   r   (  s    d8nb())tH~r)**T$Z^$$T#z/2B677d3;/344tCL12566
 49YtRx08EJFyc+<&<!=r!AB  <AQU;ViS(] 3R 78Vr;   r   c                   "    e Zd ZU eedf   ed<   y)UserInputSpecr   r   Nr!   r"   r#   r   r   rn   r:   r;   r<   r   r   ;  s     
8R<	  r;   r   c                   r    e Zd ZU eedf   ed<   eedf   ed<   eedf   ed<   eedf   ed<   eed	f   ed
<   y)ConstantValuer   r   r
   rr   r   rt   r   r   r   rs   N)	r!   r"   r#   r   ru   rn   ro   rv   rm   r:   r;   r<   r   r   A  sM    tRx  c2gr	""b!!tRx  r;   r   c                   6    e Zd ZU eedf   ed<   eedf   ed<   y)InputToConstantInputSpecr   r   r
   valueN)r!   r"   r#   r   rm   rn   r   r:   r;   r<   r   r   J  s!    
CG
]B&''r;   r   c                   6    e Zd ZU eedf   ed<   eedf   ed<   y)InputToParameterSpecr   r   r
   parameter_nameNr!   r"   r#   r   r   rn   rm   r:   r;   r<   r  r  P  !    	>2%	&&c2g&&r;   r  c                   J    e Zd ZU eedf   ed<   eedf   ed<   eedf   ed<   y)InputToBufferSpecr   r   r
   buffer_namer   
persistentN)r!   r"   r#   r   r   rn   rm   ru   r:   r;   r<   r  r  V  s0    	>2%	&&37##$(##r;   r  c                   6    e Zd ZU eedf   ed<   eedf   ed<   y)InputToTensorConstantSpecr   r   r
   tensor_constant_nameNr  r:   r;   r<   r
  r
  ]  s!    	>2%	&&#CG,,r;   r
  c                   6    e Zd ZU eedf   ed<   eedf   ed<   y)InputToCustomObjSpecr   r   r
   custom_obj_nameN)r!   r"   r#   r   r   rn   rm   r:   r;   r<   r  r  c  s"    	$b(	))sBw''r;   r  c                   "    e Zd ZU eedf   ed<   y)InputTokenSpecr   r   Nr!   r"   r#   r   r   rn   r:   r;   r<   r  r  i      	="$	%%r;   r  c                       e Zd ZU eedf   ed<   eedf   ed<   eedf   ed<   eedf   ed<   ee	d	f   ed
<   ee
df   ed<   eedf   ed<   y)	InputSpecr   
user_inputr
   	parameterr   bufferr   tensor_constantr   
custom_objr   tokenr   constant_inputN)r!   r"   r#   r   r   rn   r  r  r
  r  r  r   r:   r;   r<   r  r  n  sw    -+,,-r122'+,,8"<==.233^R'((6:;;r;   r  c                   "    e Zd ZU eedf   ed<   y)UserOutputSpecr   r   Nr   r:   r;   r<   r  r  y  s    	8R<	  r;   r  c                   "    e Zd ZU eedf   ed<   y)LossOutputSpecr   r   N)r!   r"   r#   r   r   rn   r:   r;   r<   r  r  ~  s    	>2%	&&r;   r  c                   6    e Zd ZU eedf   ed<   eedf   ed<   y)BufferMutationSpecr   r   r
   r  Nr  r:   r;   r<   r!  r!    s!    	>2%	&&37##r;   r!  c                   6    e Zd ZU eedf   ed<   eedf   ed<   y)ParameterMutationSpecr   r   r
   r  Nr  r:   r;   r<   r#  r#    r  r;   r#  c                   6    e Zd ZU eedf   ed<   eedf   ed<   y)GradientToParameterSpecr   r   r
   r  Nr  r:   r;   r<   r%  r%    r  r;   r%  c                   6    e Zd ZU eedf   ed<   eedf   ed<   y)GradientToUserInputSpecr   r   r
   user_input_nameNr  r:   r;   r<   r'  r'    !    	>2%	&&sBw''r;   r'  c                   6    e Zd ZU eedf   ed<   eedf   ed<   y)UserInputMutationSpecr   r   r
   r(  Nr  r:   r;   r<   r+  r+    r)  r;   r+  c                   "    e Zd ZU eedf   ed<   y)OutputTokenSpecr   r   Nr  r:   r;   r<   r-  r-    r  r;   r-  c                       e Zd ZU eedf   ed<   eedf   ed<   eedf   ed<   eedf   ed<   ee	d	f   ed
<   ee
df   ed<   eedf   ed<   eedf   ed<   y)
OutputSpecr   user_outputr
   loss_outputr   buffer_mutationr   gradient_to_parameterr   gradient_to_user_inputr   user_input_mutationr   r  r   parameter_mutationN)r!   r"   r#   r   r  rn   r  r!  r%  r'  r+  r-  r#  r:   r;   r<   r/  r/    s    >2-..>2-..12566$%<b%@AA%&=r&ABB"#8"#<==_b())!"7";<<r;   r/  c                   B    e Zd ZU eee   df   ed<   eee   df   ed<   y)GraphSignaturer   input_specsr
   output_specsN)r!   r"   r#   r   r   r  rn   r/  r:   r;   r<   r8  r8    s+    4	?B.//D,b011r;   r8  c                   B    e Zd ZU eedz  df   ed<   eedz  df   ed<   y)RangeConstraintNr   min_valr
   max_valr!   r"   r#   r   ro   rn   r:   r;   r<   r<  r<    s(    sTz2~&&sTz2~&&r;   r<  c                       e Zd ZU eee   df   ed<   eee   df   ed<   eedf   ed<   eedf   ed<   d	Zeee   d	z  d
f   ed<   y	)ModuleCallSignaturer   r   r
   r   r   in_specr   out_specNr   forward_arg_names)	r!   r"   r#   r   r   r   rn   rm   rD  r:   r;   r<   rA  rA    sj    d8nb())tH~r)** sBwR   :>ycT!12!56=r;   rA  c                   @    e Zd ZU eedf   ed<   dZeedz  df   ed<   y)ModuleCallEntryr   fqnNr   	signature)r!   r"   r#   r   rm   rn   rH  rA  r:   r;   r<   rF  rF    s+    	37	;?Iy,t3R78?r;   rF  c                   (    e Zd ZU eee   df   ed<   y)NamedTupleDefr   field_namesN)r!   r"   r#   r   r   rm   rn   r:   r;   r<   rJ  rJ    s    49b=))r;   rJ  c                       e Zd ZU eedf   ed<   eedf   ed<   eee   df   ed<    e	e
      Zee
eef   df   ed	<    e	e
      Zee
eef   d
f   ed<   y)GraphModuler   r   r   rH  r   module_call_graphr   r   r   r   treespec_namedtuple_fieldsN)r!   r"   r#   r   r   rn   r8  r   rF  r   r   r   rm   rO  rJ  r:   r;   r<   rM  rM    s    UBY+,, !o!6!:;;.3D.IHiS#X*+IJOK	$sM/A*BB*F G r;   rM  c                   6    e Zd ZU eedf   ed<   eedf   ed<   y)SchemaVersionr   majorr
   minorNr?  r:   r;   r<   rQ  rQ    s,    R  R r;   rQ  c                       e Zd ZU eedf   ed<   eeeef   df   ed<   eeee	f   df   ed<   ee
df   ed<    ee	      Zeee   d
f   ed<   dZeedf   ed<    ee	      Zeee   df   ed<   y)ExportedProgramr   graph_moduler
   opset_versionr   range_constraintsr   schema_versionr   r   	verifiersz<=2.4r   torch_versionr   guards_codeN)r!   r"   r#   r   rM  rn   r   rm   ro   r<  rQ  r   r   rZ  r[  r\  r:   r;   r<   rU  rU    s    KO,,T#s(^R/00 c?&:!;R!?@@mR/00*/*EIycB'E(/M9S"W%/,1$,GK49b=)Gr;   rU  c                   d    e Zd ZU eedf   ed<   eedf   ed<   eedf   ed<   eedz  df   ed	<   y)
PayloadMetar   	path_namer
   is_paramr   
use_pickleNr   tensor_meta)r!   r"   r#   r   rm   rn   ru   r   r:   r;   r<   r^  r^    sH     b!!b!! $(##:,b011r;   r^  c                   ,    e Zd ZU eeeef   df   ed<   y)PayloadConfigr   configN)r!   r"   r#   r   r   rm   r^  rn   r:   r;   r<   rd  rd    s    d3+,b011r;   rd  c                       e Zd ZU eedf   ed<   eee   df   ed<   eee   df   ed<   dZeedz  df   ed	<   dZ	eedz  d
f   ed<   dZ
eedz  df   ed<   y)AOTInductorModelPickleDatar   library_basenamer   input_namesr   output_namesNr   floating_point_input_dtyper   floating_point_output_dtyper   aot_inductor_model_is_cpu)r!   r"   r#   r   rm   rn   r   rk  ro   rl  rm  ru   r:   r;   r<   rg  rg    s      Q'' 49a<(( DIqL))
 <@	#*a- 8?<@3:q=!9@ <@ya8?r;   rg  c                   6    e Zd ZU eedf   ed<   eedf   ed<   y)ExternKernelNoder   r   r
   nodeN)r!   r"   r#   r   rm   rn   r   r:   r;   r<   ro  ro  5  s"     CG

D"H
r;   ro  c                   (    e Zd ZU eee   df   ed<   y)ExternKernelNodesr   r   N)r!   r"   r#   r   r   ro  rn   r:   r;   r<   rr  rr  <  s    T*+R/00r;   rr  N)jdataclassesr   r   enumr   typingr   torch._export.serde.unionr   r   SCHEMA_VERSIONTREESPEC_VERSIONr   r>   rH   r%   r&   r'   r(   r)   r*   r+   r,   r-   r.   r/   r0   r1   r2   r3   r4   r5   r6   r7   r8   r9   rb   r   ro   rm   rn   r@   rA   rB   rC   rD   rE   rF   re   rI   rJ   rK   rL   rh   rj   rq   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%  r'  r+  r-  r/  r8  r<  rA  rF  rJ  rM  rQ  rU  r^  rd  rg  ro  rr  r:   r;   r<   <module>ry     s   ) (   >   2W 7 &OOV&OOV& g& NNE	&
 OOV& OOV& g& x& M& ^& o& OOV& & x& _&  =!&" 0#&$ 0.xx+& DcN 2 h
k
k
k
k
NNH
NNI!tCH~  !!<~!1	( d38n  , , ,
 #& # # 3 3 3
 V  
 #v # #
 !f ! !
 " " "  V   #v # # !f ! !
       !V ! !
 " " "
 " " "
    6v 6 6D7  4 4 4 + + + W W W$ ! ! !
 !F ! ! ( ( (
 ' ' '
 $ $ $ - - -
 ( ( (
 & & & < < < ! ! ! ' ' ' $ $ $
 ' ' '
 ' ' '
 ( ( (
 ( ( (
 & & & = = = 2 2 2
 ' ' '
 > > > @ @ @
 * * *        H H H$ 	2 	2 	2 2 2 2 @ @ @*    1 1 1r;   