
    ȅi:                        S SK JrJr  S SKJr  S SKJrJr  S SKJ	r	J
r
  SrSr " S S\5      r " S	 S
\5      r " S S\5      r\ " S S5      5       r\
 " S S\	5      5       r\ " S S5      5       r\
 " S S\	5      5       r\
 " S S\	5      5       r\
 " S S\	5      5       r\ " S S5      5       r\
 " S S\	5      5       r\
 " S S\	5      5       r\
 " S S \	5      5       r\ " S! S"5      5       r\ " S# S$5      5       r\
 " S% S&\	5      5       r\ " S' S(5      5       r\ " S) S*5      5       r\ " S+ S,5      5       r\
 " S- S.\	5      5       r  " S/ S0\5      r!\ " S1 S25      5       r"\ " S3 S45      5       r#\ " S5 S65      5       r$\ " S7 S85      5       r%\
 " S9 S:\	5      5       r&\ " S; S<5      5       r'\ " S= S>5      5       r(\ " S? S@5      5       r)\ " SA SB5      5       r*\ " SC SD5      5       r+\ " SE SF5      5       r,\
 " SG SH\	5      5       r-\ " SI SJ5      5       r.\ " SK SL5      5       r/\ " SM SN5      5       r0\ " SO SP5      5       r1\ " SQ SR5      5       r2\ " SS ST5      5       r3\ " SU SV5      5       r4\ " SW SX5      5       r5\
 " SY SZ\	5      5       r6\ " S[ S\5      5       r7\ " S] S^5      5       r8\ " S_ S`5      5       r9\ " Sa Sb5      5       r:\ " Sc Sd5      5       r;\ " Se Sf5      5       r<\ " Sg Sh5      5       r=\ " Si Sj5      5       r>\ " Sk Sl5      5       r?\ " Sm Sn5      5       r@\ " So Sp5      5       rA\ " Sq Sr5      5       rB\ " Ss St5      5       rCgu)v    )	dataclassfield)IntEnum)	AnnotatedOptional)_Union_union_dataclass)         c                   `    \ rS rSrSrSrSrSrSrSr	Sr
S	rS
rSrSrSrSrSrSrSrSrSrSrSrg)
ScalarType   r   r                     r
   	   
                             N)__name__
__module____qualname____firstlineno__UNKNOWNBYTECHARSHORTINTLONGHALFFLOATDOUBLECOMPLEXHALFCOMPLEXFLOATCOMPLEXDOUBLEBOOLBFLOAT16UINT16FLOAT8E4M3FN
FLOAT8E5M2FLOAT8E4M3FNUZFLOAT8E5M2FNUZ__static_attributes__r        T/home/james-whalen/.local/lib/python3.13/site-packages/torch/_export/serde/schema.pyr   r      sa    GDDE
CDDEFKLMDHFLJNNr9   r   c                   4    \ rS rSrSrSrSrSrSrSr	Sr
S	rS
rg)Layout(   r   r   r   r   r   r   r   r   r    N)r!   r"   r#   r$   Unknown	SparseCoo	SparseCsr	SparseCsc	SparseBsr	SparseBsc_mkldnnStridedr8   r    r9   r:   r<   r<   (   s*    GIIIIIGGr9   r<   c                   (    \ rS rSrSrSrSrSrSrSr	g)	MemoryFormat3   r   r   r   r   r   r    N)
r!   r"   r#   r$   r>   ContiguousFormatChannelsLastChannelsLast3dPreserveFormatr8   r    r9   r:   rG   rG   3   s    GLNNr9   rG   c                   H    \ rS rSr% \\S4   \S'   Sr\\\	   S4   \S'   Sr
g)Device;   r   typeN   indexr    )r!   r"   r#   r$   r   str__annotations__rR   r   intr8   r    r9   r:   rN   rN   ;   s)    
CG
*.E9Xc]B&'.r9   rN   c                   R    \ rS rSr% \\S4   \S'   \\S4   \S'   \\S4   \S'   Sr	g	)
SymExprHintA   r   as_intrQ   as_boolr   as_floatr    N)
r!   r"   r#   r$   r   rU   rT   boolfloatr8   r    r9   r:   rW   rW   A   s/    c2gtRx  r	""r9   rW   c                   H    \ rS rSr% \\S4   \S'   Sr\\\	   S4   \S'   Sr
g)SymExprL   r   expr_strNrQ   hintr    )r!   r"   r#   r$   r   rS   rT   rb   r   rW   r8   r    r9   r:   r_   r_   L   s*    R  15D)H[)2-
.5r9   r_   c                   >    \ rS rSr% \\S4   \S'   \\S4   \S'   Srg)SymIntR   r   as_exprrQ   rY   r    N)	r!   r"   r#   r$   r   r_   rT   rU   r8   r    r9   r:   rd   rd   R   s     w{##c2gr9   rd   c                   >    \ rS rSr% \\S4   \S'   \\S4   \S'   Srg)SymFloatX   r   rf   rQ   r[   r    N)	r!   r"   r#   r$   r   r_   rT   r]   r8   r    r9   r:   rh   rh   X   s     w{##r	""r9   rh   c                   >    \ rS rSr% \\S4   \S'   \\S4   \S'   Srg)SymBool^   r   rf   rQ   rZ   r    N)	r!   r"   r#   r$   r   r_   rT   r\   r8   r    r9   r:   rk   rk   ^   s     w{##tRx  r9   rk   c                       \ rS rSr% \\S4   \S'   \\\   S4   \S'   \\	S4   \S'   \\
S4   \S	'   \\\   S
4   \S'   \\S4   \S'   \\S4   \S'   Srg)
TensorMetad   r   dtyperQ   sizesr   requires_gradr=   device2   strides<   storage_offsetF   layoutr    N)r!   r"   r#   r$   r   r   rT   listrd   r\   rN   r<   r8   r    r9   r:   rn   rn   d   su    Z^$$T&\2%&&T2X&&fbj!!tF|R'((fbj))fbj!!r9   rn   c                   >    \ rS rSr% \\S4   \S'   \\S4   \S'   Srg)SymIntArgumentu   r   as_namerQ   rY   r    N)	r!   r"   r#   r$   r   rS   rT   rU   r8   r    r9   r:   r|   r|   u   s     sBwc2gr9   r|   c                   >    \ rS rSr% \\S4   \S'   \\S4   \S'   Srg)SymFloatArgument   r   r~   rQ   r[   r    N)	r!   r"   r#   r$   r   rS   rT   r]   r8   r    r9   r:   r   r      s     sBwr	""r9   r   c                   >    \ rS rSr% \\S4   \S'   \\S4   \S'   Srg)SymBoolArgument   r   r~   rQ   rZ   r    N)	r!   r"   r#   r$   r   rS   rT   r\   r8   r    r9   r:   r   r      s     sBwtRx  r9   r   c                   *    \ rS rSr% \\S4   \S'   Srg)TensorArgument   r   namer    Nr!   r"   r#   r$   r   rS   rT   r8   r    r9   r:   r   r          
CG
r9   r   c                   *    \ rS rSr% \\S4   \S'   Srg)TokenArgument   r   r   r    Nr   r    r9   r:   r   r      r   r9   r   c                   >    \ rS rSr% \\S4   \S'   \\S4   \S'   Srg)OptionalTensorArgument   rQ   	as_tensorr   as_noner    N)	r!   r"   r#   r$   r   r   rT   r\   r8   r    r9   r:   r   r      s!    +,,tRx  r9   r   c                   :    \ rS rSr% \\S4   \S'   \S   \S'   Srg)GraphArgument   r   r   )GraphrQ   graphr    Nr   r    r9   r:   r   r      s    
CG
[!!r9   r   c                   >    \ rS rSr% \\S4   \S'   \\S4   \S'   Srg)CustomObjArgument   r   r   rQ   	class_fqnr    Nr   r    r9   r:   r   r      s     
CG
b!!r9   r   c                   >    \ rS rSr% \\S4   \S'   \\S4   \S'   Srg)ComplexValue   r   realrQ   imagr    N)r!   r"   r#   r$   r   r]   rT   r8   r    r9   r:   r   r      s     
E2I

E2I
r9   r   c                      \ rS rSr% \\S4   \S'   \\S4   \S'   \\\   S4   \S'   \\	S4   \S	'   \\\	   S
4   \S'   \\
S4   \S'   \\\
   S4   \S'   \\S4   \S'   \\\   S4   \S'   \\S4   \S'   \\\   S4   \S'   \\S4   \S'   \\S4   \S'   \\S4   \S'   \\S4   \S'   \\S 4   \S!'   \\\   S"4   \S#'   \\S$4   \S%'   \\\   S&4   \S''   \\S(4   \S)'   \\\   S*4   \S+'   \\S,4   \S-'   \\S.4   \S/'   \\S04   \S1'   \\\   S24   \S3'   \\S44   \S5'   \\S64   \S7'   \\\\	      S84   \S9'   \\\S 4   S:4   \S;'   S<rg=)>Argument   r   r   rQ   r   r   
as_tensorsrt   rY   rx   as_intsP   r[   Z   	as_floatsro   	as_stringe   
as_stringsn   
as_sym_intx   as_sym_ints   as_scalar_type   as_memory_format   	as_layout   	as_device   rZ      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_int_lists"  as_string_to_argumentr    N)r!   r"   r#   r$   r   r\   rT   r   rz   rU   r]   rS   r|   r   rG   r<   rN   r   r   r   r   r   r   dictr8   r    r9   r:   r   r      s   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cOS011$T#z/%:C%?@@r9   r   c                        \ rS rSrSrSrSrSrg)ArgumentKind   r   r   r   r    N)r!   r"   r#   r$   r%   
POSITIONALKEYWORDr8   r    r9   r:   r   r      s    GJGr9   r   c                   \    \ rS rSr% \\S4   \S'   \\S4   \S'   Sr\\	\
   S4   \S'   S	rg)
NamedArgument   r   r   rQ   argNr   kindr    )r!   r"   r#   r$   r   rS   rT   r   r   r   r   r8   r    r9   r:   r   r      s;     CG
	8R<	  26D)H\*B.
/6r9   r   c                       \ rS rSr% \\S4   \S'   \\\   S4   \S'   \\\	   S4   \S'   \\
\\4   S4   \S	'   S
r\\\   S4   \S'   S
r\\\   S4   \S'   Srg
)Node   r   targetrQ   inputsr   outputsr=   metadataNrt   is_hop_single_tensor_returnrv   r   r    )r!   r"   r#   r$   r   rS   rT   rz   r   r   r   r   r   r\   r   r8   r    r9   r:   r   r      s    c2gd=)2-..tH~r)**S#X*++AE8D>2+=!>E)-D)HSM2%
&-r9   r   c                   *   \ rS rSr% \\\   S4   \S'   \\\   S4   \S'   \\\   S4   \S'   \\	\
\4   S4   \S	'   \\	\
\4   S
4   \S'   \\	\
\4   S4   \S'   Sr\\S4   \S'   \" \	S9r\\	\
\4   S4   \S'   \" \	S9r\\	\
\4   S4   \S'   Srg)r      r   r   rQ   r   r   nodesr=   tensor_valuesrt   sym_int_valuesrv   sym_bool_valuesFrx   is_single_tensor_returndefault_factoryr   custom_obj_valuesr   sym_float_valuesr    N)r!   r"   r#   r$   r   rz   r   rT   r   r   rS   rn   rd   rk   r   r\   r   r   r   r   rh   r8   r    r9   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Vr9   r   c                   *    \ rS rSr% \\S4   \S'   Srg)UserInputSpeci  r   r   r    Nr!   r"   r#   r$   r   r   rT   r8   r    r9   r:   r   r     s     
8R<	  r9   r   c                   z    \ rS rSr% \\S4   \S'   \\S4   \S'   \\S4   \S'   \\	S4   \S	'   \\S
4   \S'   Sr
g)ConstantValuei  r   r   rQ   rY   r   r[   r=   r   rt   rZ   r    N)r!   r"   r#   r$   r   r\   rT   rU   r]   rS   r8   r    r9   r:   r   r     sM    tRx  c2gr	""b!!tRx  r9   r   c                   >    \ rS rSr% \\S4   \S'   \\S4   \S'   Srg)InputToConstantInputSpeci  r   r   rQ   valuer    N)	r!   r"   r#   r$   r   rS   rT   r   r8   r    r9   r:   r   r     s!    
CG
]B&''r9   r   c                   >    \ rS rSr% \\S4   \S'   \\S4   \S'   Srg)InputToParameterSpeci  r   r   rQ   parameter_namer    N	r!   r"   r#   r$   r   r   rT   rS   r8   r    r9   r:   r   r     !    	>2%	&&c2g&&r9   r   c                   R    \ rS rSr% \\S4   \S'   \\S4   \S'   \\S4   \S'   Sr	g	)
InputToBufferSpecr   r   r   rQ   buffer_namer   
persistentr    N)
r!   r"   r#   r$   r   r   rT   rS   r\   r8   r    r9   r:   r   r   "  s0    	>2%	&&37##$(##r9   r   c                   >    \ rS rSr% \\S4   \S'   \\S4   \S'   Srg)InputToTensorConstantSpeci)  r   r   rQ   tensor_constant_namer    Nr   r    r9   r:   r  r  )  s!    	>2%	&&#CG,,r9   r  c                   >    \ rS rSr% \\S4   \S'   \\S4   \S'   Srg)InputToCustomObjSpeci/  r   r   rQ   custom_obj_namer    N)	r!   r"   r#   r$   r   r   rT   rS   r8   r    r9   r:   r  r  /  s"    	$b(	))sBw''r9   r  c                   *    \ rS rSr% \\S4   \S'   Srg)InputTokenSpeci5  r   r   r    Nr!   r"   r#   r$   r   r   rT   r8   r    r9   r:   r  r  5      	="$	%%r9   r  c                       \ rS rSr% \\S4   \S'   \\S4   \S'   \\S4   \S'   \\	S4   \S	'   \\
S
4   \S'   \\S4   \S'   \\S4   \S'   Srg)	InputSpeci:  r   
user_inputrQ   	parameterr   bufferr=   tensor_constantrt   
custom_objrx   tokenrv   constant_inputr    N)r!   r"   r#   r$   r   r   rT   r   r   r  r  r  r   r8   r    r9   r:   r  r  :  sw    -+,,-r122'+,,8"<==.233^R'((6:;;r9   r  c                   *    \ rS rSr% \\S4   \S'   Srg)UserOutputSpeciE  r   r   r    Nr   r    r9   r:   r  r  E  s    	8R<	  r9   r  c                   *    \ rS rSr% \\S4   \S'   Srg)LossOutputSpeciJ  r   r   r    N)r!   r"   r#   r$   r   r   rT   r8   r    r9   r:   r  r  J  s    	>2%	&&r9   r  c                   >    \ rS rSr% \\S4   \S'   \\S4   \S'   Srg)BufferMutationSpeciO  r   r   rQ   r   r    Nr   r    r9   r:   r  r  O  s!    	>2%	&&37##r9   r  c                   >    \ rS rSr% \\S4   \S'   \\S4   \S'   Srg)ParameterMutationSpeciU  r   r   rQ   r   r    Nr   r    r9   r:   r  r  U  r   r9   r  c                   >    \ rS rSr% \\S4   \S'   \\S4   \S'   Srg)GradientToParameterSpeci[  r   r   rQ   r   r    Nr   r    r9   r:   r  r  [  r   r9   r  c                   >    \ rS rSr% \\S4   \S'   \\S4   \S'   Srg)GradientToUserInputSpecia  r   r   rQ   user_input_namer    Nr   r    r9   r:   r  r  a  !    	>2%	&&sBw''r9   r  c                   >    \ rS rSr% \\S4   \S'   \\S4   \S'   Srg)UserInputMutationSpecig  r   r   rQ   r  r    Nr   r    r9   r:   r"  r"  g  r   r9   r"  c                   *    \ rS rSr% \\S4   \S'   Srg)OutputTokenSpecim  r   r   r    Nr  r    r9   r:   r$  r$  m  r	  r9   r$  c                       \ rS rSr% \\S4   \S'   \\S4   \S'   \\S4   \S'   \\	S4   \S	'   \\
S
4   \S'   \\S4   \S'   \\S4   \S'   \\S4   \S'   Srg)
OutputSpecir  r   user_outputrQ   loss_outputr   buffer_mutationr=   gradient_to_parameterrt   gradient_to_user_inputrv   user_input_mutationrx   r  r   parameter_mutationr    N)r!   r"   r#   r$   r   r  rT   r  r  r  r  r"  r$  r  r8   r    r9   r:   r&  r&  r  s    >2-..>2-..12566$%<b%@AA%&=r&ABB"#8"#<==_b())!"7";<<r9   r&  c                   J    \ rS rSr% \\\   S4   \S'   \\\   S4   \S'   Sr	g)GraphSignaturei~  r   input_specsrQ   output_specsr    N)
r!   r"   r#   r$   r   rz   r  rT   r&  r8   r    r9   r:   r/  r/  ~  s+    4	?B.//D,b011r9   r/  c                   J    \ rS rSr% \\\   S4   \S'   \\\   S4   \S'   Srg)RangeConstrainti  r   min_valrQ   max_valr    N)	r!   r"   r#   r$   r   r   rU   rT   r8   r    r9   r:   r3  r3    s*    x}b())x}b())r9   r3  c                       \ rS rSr% \\\   S4   \S'   \\\   S4   \S'   \\S4   \S'   \\S4   \S	'   S
r	\\
\\      S4   \S'   Srg
)ModuleCallSignaturei  r   r   rQ   r   r   in_specr=   out_specNrt   forward_arg_namesr    )r!   r"   r#   r$   r   rz   r   rT   rS   r:  r   r8   r    r9   r:   r7  r7    sk    d8nb())tH~r)** sBwR   =Ay$s)!4b!89@r9   r7  c                   H    \ rS rSr% \\S4   \S'   Sr\\\	   S4   \S'   Sr
g)ModuleCallEntryi  r   fqnNr   	signaturer    )r!   r"   r#   r$   r   rS   rT   r>  r   r7  r8   r    r9   r:   r<  r<    s+    	37	>BIy"56:;Br9   r<  c                   0    \ rS rSr% \\\   S4   \S'   Srg)NamedTupleDefi  r   field_namesr    N)	r!   r"   r#   r$   r   rz   rS   rT   r8   r    r9   r:   r@  r@    s    49b=))r9   r@  c                       \ rS rSr% \\S4   \S'   \\S4   \S'   \\\	   S4   \S'   \
" \S9r\\\\4   S	4   \S
'   \
" \S9r\\\\4   S4   \S'   Srg)GraphModulei  r   r   rt   r>  rv   module_call_graphr   r=   r   rx   treespec_namedtuple_fieldsr    N)r!   r"   r#   r$   r   r   rT   r/  rz   r<  r   r   r   rS   rE  r@  r8   r    r9   r:   rC  rC    s    UBY+,, !o!6!:;;.3D.IHiS#X*+IJOK	$sM/A*BB*F G r9   rC  c                   >    \ rS rSr% \\S4   \S'   \\S4   \S'   Srg)SchemaVersioni  r   majorrQ   minorr    N)r!   r"   r#   r$   r   rU   rT   r8   r    r9   r:   rG  rG    s,    R  R r9   rG  c                       \ rS rSr% \\S4   \S'   \\\\	4   S4   \S'   \\\\
4   S4   \S'   \\S4   \S	'   \" \S
9r\\\   S4   \S'   Sr\\S4   \S'   \" \S
9r\\\   S4   \S'   Srg)ExportedProgrami  r   graph_modulerQ   opset_versionr   range_constraintsrv   schema_versionr   rx   	verifiersz<=2.4r   torch_versionr   guards_coder    N)r!   r"   r#   r$   r   rC  rT   r   rS   rU   r3  rG  r   rz   rP  rQ  rR  r8   r    r9   r:   rK  rK    s    KO,,T#s(^R/00 c?&:!;R!?@@mR/00*/*EIycB'E(/M9S"W%/,1$,GK49b=)Gr9   rK  c                   l    \ rS rSr% \\S4   \S'   \\S4   \S'   \\S4   \S'   \\\	   S4   \S	'   S
r
g)PayloadMetai  r   	path_namerQ   is_paramr   
use_pickler=   tensor_metar    N)r!   r"   r#   r$   r   rS   rT   r\   r   rn   r8   r    r9   r:   rT  rT    sH     b!!b!! $(##8J/344r9   rT  c                   4    \ rS rSr% \\\\4   S4   \S'   Sr	g)PayloadConfigi  r   configr    N)
r!   r"   r#   r$   r   r   rS   rT  rT   r8   r    r9   r:   rZ  rZ    s    d3+,b011r9   rZ  c                       \ rS rSr% \\S4   \S'   \\\   S4   \S'   \\\   S4   \S'   Sr\\	\
   S	4   \S
'   Sr\\	\
   S4   \S'   Sr\\	\   S4   \S'   Srg)AOTInductorModelPickleDatai  r   library_basenamer   input_namesr   output_namesNr   floating_point_input_dtyper   floating_point_output_dtyper   aot_inductor_model_is_cpur    )r!   r"   r#   r$   r   rS   rT   rz   ra  r   rU   rb  rc  r\   r8   r    r9   r:   r]  r]    s      Q'' 49a<(( DIqL))
 ?C	(3-*: ;B?C8C=!+;!<C ?Cy$):;Br9   r]  c                   >    \ rS rSr% \\S4   \S'   \\S4   \S'   Srg)ExternKernelNodei  r   r   rQ   noder    N)	r!   r"   r#   r$   r   rS   rT   r   r8   r    r9   r:   re  re    s"     CG

D"H
r9   re  c                   0    \ rS rSr% \\\   S4   \S'   Srg)ExternKernelNodesi  r   r   r    N)	r!   r"   r#   r$   r   rz   re  rT   r8   r    r9   r:   rh  rh    s    T*+R/00r9   rh  N)Ddataclassesr   r   enumr   typingr   r   torch._export.serde.unionr   r	   SCHEMA_VERSIONTREESPEC_VERSIONr   r<   rG   rN   rW   r_   rd   rh   rk   rn   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/  r3  r7  r<  r@  rC  rG  rK  rT  rZ  r]  re  rh  r    r9   r:   <module>ro     sv   )  & >  
 ,W 7  / / /
 #& # # 6 6 6
 V  
 #v # #
 !f ! !
 " " "  V   #v # # !f ! !
       !V ! !
 " " "
 " " "
    Av A A@7  7 7 7 . . . W W W$ ! ! !
 !F ! ! ( ( (
 ' ' '
 $ $ $ - - -
 ( ( (
 & & & < < < ! ! ! ' ' ' $ $ $
 ' ' '
 ' ' '
 ( ( (
 ( ( (
 & & & = = = 2 2 2
 * * *
 A A A C C C
 * * *        H H H$ 	5 	5 	5 2 2 2 C C C*    1 1 1r9   