
    sh8                        S SK JrJr  S SKJr  S SKJrJr  S SK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9 " S S\	5      5       r\ " S S5      5       r\" SS9 " S S\	5      5       r\" SS9 " S S\	5      5       r\" SS9 " S S\	5      5       r\ " S S5      5       r\" SS9 " S S\	5      5       r\" SS9 " S S \	5      5       r\" SS9 " S! S"\	5      5       r\ " S# S$5      5       r\ " S% S&5      5       r\" SS9 " S' S(\	5      5       r\ " S) S*5      5       r\ " S+ S,5      5       r\" SS9 " 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#\" SS9 " 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*\" SS9 " SG SH\	5      5       r+\ " SI SJ5      5       r,\ " SK SL5      5       r-\ " SM SN5      5       r.\ " SO SP5      5       r/\ " SQ SR5      5       r0\ " SS ST5      5       r1\ " SU SV5      5       r2\" SS9 " SW SX\	5      5       r3\ " SY SZ5      5       r4\ " S[ S\5      5       r5\ " S] S^5      5       r6\ " S_ S`5      5       r7\ " Sa Sb5      5       r8\ " Sc Sd5      5       r9\ " Se Sf5      5       r:\ " Sg Sh5      5       r;\ " Si Sj5      5       r<\ " Sk Sl5      5       r=\ " Sm Sn5      5       r>\ " So Sp5      5       r?\ " Sq Sr5      5       r@gs)t    )	dataclassfield)IntEnum)	AnnotatedOptional)_Union)         c                   X    \ 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
   r	   	   
                      N)__name__
__module____qualname____firstlineno__UNKNOWNBYTECHARSHORTINTLONGHALFFLOATDOUBLECOMPLEXHALFCOMPLEXFLOATCOMPLEXDOUBLEBOOLBFLOAT16UINT16FLOAT8E4M3FN
FLOAT8E5M2__static_attributes__r       n/Users/tiagomarins/Projetos/claudeai/copy_bank/venv/lib/python3.13/site-packages/torch/_export/serde/schema.pyr   r      sW    GDDE
CDDEFKLMDHFLJr3   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Stridedr2   r   r3   r4   r6   r6   %   s*    GIIIIIGGr3   r6   c                   (    \ rS rSrSrSrSrSrSrSr	g)	MemoryFormat0   r   r   r   r   r   r   N)
r   r   r   r    r8   ContiguousFormatChannelsLastChannelsLast3dPreserveFormatr2   r   r3   r4   rA   rA   0   s    GLNNr3   rA   c                   H    \ rS rSr% \\S4   \S'   Sr\\\	   S4   \S'   Sr
g)Device8   r   typeN   indexr   )r   r   r   r    r   str__annotations__rL   r   intr2   r   r3   r4   rH   rH   8   s)    
CG
*.E9Xc]B&'.r3   rH   F)reprc                   R    \ rS rSr% \\S4   \S'   \\S4   \S'   \\S4   \S'   Sr	g	)
SymExprHint>   r   as_intrK   as_boolr   as_floatr   N)
r   r   r   r    r   rO   rN   boolfloatr2   r   r3   r4   rR   rR   >   s/    c2gtRx  r	""r3   rR   c                   H    \ rS rSr% \\S4   \S'   Sr\\\	   S4   \S'   Sr
g)SymExprI   r   expr_strNrK   hintr   )r   r   r   r    r   rM   rN   r]   r   rR   r2   r   r3   r4   rZ   rZ   I   s*    R  15D)H[)2-
.5r3   rZ   c                   >    \ rS rSr% \\S4   \S'   \\S4   \S'   Srg)SymIntO   r   as_exprrK   rT   r   N)	r   r   r   r    r   rZ   rN   rO   r2   r   r3   r4   r_   r_   O   s     w{##c2gr3   r_   c                   >    \ rS rSr% \\S4   \S'   \\S4   \S'   Srg)SymFloatT   r   ra   rK   rV   r   N)	r   r   r   r    r   rZ   rN   rX   r2   r   r3   r4   rc   rc   T   s     w{##r	""r3   rc   c                   >    \ rS rSr% \\S4   \S'   \\S4   \S'   Srg)SymBoolZ   r   ra   rK   rU   r   N)	r   r   r   r    r   rZ   rN   rW   r2   r   r3   r4   rf   rf   Z   s     w{##tRx  r3   rf   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)
TensorMeta`   r   dtyperK   sizesr   requires_grad(   device2   strides<   storage_offsetF   layoutr   N)r   r   r   r    r   r   rN   listr_   rW   rH   r6   r2   r   r3   r4   ri   ri   `   su    Z^$$T&\2%&&T2X&&fbj!!tF|R'((fbj))fbj!!r3   ri   c                   >    \ rS rSr% \\S4   \S'   \\S4   \S'   Srg)SymIntArgumentq   r   as_namerK   rT   r   N)	r   r   r   r    r   rM   rN   rO   r2   r   r3   r4   rx   rx   q   s     sBwc2gr3   rx   c                   >    \ rS rSr% \\S4   \S'   \\S4   \S'   Srg)SymFloatArgument|   r   rz   rK   rV   r   N)	r   r   r   r    r   rM   rN   rX   r2   r   r3   r4   r|   r|   |   s     sBwr	""r3   r|   c                   >    \ rS rSr% \\S4   \S'   \\S4   \S'   Srg)SymBoolArgument   r   rz   rK   rU   r   N)	r   r   r   r    r   rM   rN   rW   r2   r   r3   r4   r   r      s     sBwtRx  r3   r   c                   *    \ rS rSr% \\S4   \S'   Srg)TensorArgument   r   namer   Nr   r   r   r    r   rM   rN   r2   r   r3   r4   r   r          
CG
r3   r   c                   *    \ rS rSr% \\S4   \S'   Srg)TokenArgument   r   r   r   Nr   r   r3   r4   r   r      r   r3   r   c                   >    \ rS rSr% \\S4   \S'   \\S4   \S'   Srg)OptionalTensorArgument   rK   	as_tensorr   as_noner   N)	r   r   r   r    r   r   rN   rW   r2   r   r3   r4   r   r      s!    +,,tRx  r3   r   c                   :    \ rS rSr% \\S4   \S'   \S   \S'   Srg)GraphArgument   r   r   )GraphrK   graphr   Nr   r   r3   r4   r   r      s    
CG
[!!r3   r   c                   >    \ rS rSr% \\S4   \S'   \\S4   \S'   Srg)CustomObjArgument   r   r   rK   	class_fqnr   Nr   r   r3   r4   r   r      s     
CG
b!!r3   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'   S4rg5)6Argument   r   r   rK   r   r   
as_tensorsrp   rT   rt   as_intsP   rV   rg   	as_floatsd   	as_stringe   
as_stringsn   
as_sym_intx   as_sym_ints   as_scalar_type   as_memory_format   	as_layout   	as_device   rU      as_bools   as_sym_bool   as_sym_bools   as_graph   as_optional_tensors   as_custom_obj   as_operator   as_sym_float   as_sym_floatsr   N)r   r   r   r    r   rW   rN   r   rv   rO   rX   rM   rx   r   rA   r6   rH   r   r   r   r   r|   r2   r   r3   r4   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9r3   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KEYWORDr2   r   r3   r4   r   r      s    GJGr3   r   c                   \    \ rS rSr% \\S4   \S'   \\S4   \S'   Sr\\	\
   S4   \S'   S	rg)
NamedArgument   r   r   rK   argNr   kindr   )r   r   r   r    r   rM   rN   r   r   r   r   r2   r   r3   r4   r   r      s;     CG
	8R<	  26D)H\*B.
/6r3   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g
)Node   r   targetrK   inputsr   outputsrn   metadataNrp   is_hop_single_tensor_returnr   )r   r   r   r    r   rM   rN   rv   r   r   dictr   r   rW   r2   r   r3   r4   r   r      sk    c2gd=)2-..tH~r)**S#X*++AE8D>2+=!>Er3   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   rK   r   r   nodesrn   tensor_valuesrp   sym_int_valuesrr   sym_bool_valuesFrt   is_single_tensor_returndefault_factoryr   custom_obj_valuesrg   sym_float_valuesr   N)r   r   r   r    r   rv   r   rN   r   r   rM   ri   r_   rf   r   rW   r   r   r   r   rc   r2   r   r3   r4   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[_E`yc+<&<!=r!AB`;@QU;ViS(] 3R 78Vr3   r   c                   *    \ rS rSr% \\S4   \S'   Srg)UserInputSpec   r   r   r   Nr   r   r   r    r   r   rN   r2   r   r3   r4   r   r      s     
8R<	  r3   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)ConstantValue   r   r   rK   rT   r   rV   rn   r   rp   rU   r   N)r   r   r   r    r   rW   rN   rO   rX   rM   r2   r   r3   r4   r   r      sM    tRx  c2gr	""b!!tRx  r3   r   c                   >    \ rS rSr% \\S4   \S'   \\S4   \S'   Srg)InputToConstantInputSpeci  r   r   rK   valuer   N)	r   r   r   r    r   rM   rN   r   r2   r   r3   r4   r   r     s!    
CG
]B&''r3   r   c                   >    \ rS rSr% \\S4   \S'   \\S4   \S'   Srg)InputToParameterSpeci  r   r   rK   parameter_namer   N	r   r   r   r    r   r   rN   rM   r2   r   r3   r4   r   r     !    	>2%	&&c2g&&r3   r   c                   R    \ rS rSr% \\S4   \S'   \\S4   \S'   \\S4   \S'   Sr	g	)
InputToBufferSpeci  r   r   rK   buffer_namer   
persistentr   N)
r   r   r   r    r   r   rN   rM   rW   r2   r   r3   r4   r   r     s0    	>2%	&&37##$(##r3   r   c                   >    \ rS rSr% \\S4   \S'   \\S4   \S'   Srg)InputToTensorConstantSpeci  r   r   rK   tensor_constant_namer   Nr   r   r3   r4   r   r     s!    	>2%	&&#CG,,r3   r   c                   >    \ rS rSr% \\S4   \S'   \\S4   \S'   Srg)InputToCustomObjSpeci  r   r   rK   custom_obj_namer   N)	r   r   r   r    r   r   rN   rM   r2   r   r3   r4   r   r     s"    	$b(	))sBw''r3   r   c                   *    \ rS rSr% \\S4   \S'   Srg)InputTokenSpeci!  r   r   r   Nr   r   r   r    r   r   rN   r2   r   r3   r4   r   r   !      	="$	%%r3   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_inputrK   	parameterr   bufferrn   tensor_constantrp   
custom_objrt   tokenrr   constant_inputr   N)r   r   r   r    r   r   rN   r   r   r   r   r   r   r2   r   r3   r4   r   r   &  sw    -+,,-r122'+,,8"<==.233^R'((6:;;r3   r   c                   *    \ rS rSr% \\S4   \S'   Srg)UserOutputSpeci1  r   r   r   Nr   r   r3   r4   r  r  1  s    	8R<	  r3   r  c                   *    \ rS rSr% \\S4   \S'   Srg)LossOutputSpeci6  r   r   r   N)r   r   r   r    r   r   rN   r2   r   r3   r4   r	  r	  6  s    	>2%	&&r3   r	  c                   >    \ rS rSr% \\S4   \S'   \\S4   \S'   Srg)BufferMutationSpeci;  r   r   rK   r   r   Nr   r   r3   r4   r  r  ;  s!    	>2%	&&37##r3   r  c                   >    \ rS rSr% \\S4   \S'   \\S4   \S'   Srg)GradientToParameterSpeciA  r   r   rK   r   r   Nr   r   r3   r4   r  r  A  r   r3   r  c                   >    \ rS rSr% \\S4   \S'   \\S4   \S'   Srg)GradientToUserInputSpeciG  r   r   rK   user_input_namer   Nr   r   r3   r4   r  r  G  !    	>2%	&&sBw''r3   r  c                   >    \ rS rSr% \\S4   \S'   \\S4   \S'   Srg)UserInputMutationSpeciM  r   r   rK   r  r   Nr   r   r3   r4   r  r  M  r  r3   r  c                   *    \ rS rSr% \\S4   \S'   Srg)OutputTokenSpeciS  r   r   r   Nr   r   r3   r4   r  r  S  r   r3   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)
OutputSpeciX  r   user_outputrK   loss_outputr   buffer_mutationrn   gradient_to_parameterrp   gradient_to_user_inputrr   user_input_mutationrt   r  r   N)r   r   r   r    r   r  rN   r	  r  r  r  r  r  r2   r   r3   r4   r  r  X  sv    >2-..>2-..12566$%<b%@AA%&=r&ABB"#8"#<==_b())r3   r  c                   J    \ rS rSr% \\\   S4   \S'   \\\   S4   \S'   Sr	g)GraphSignatureic  r   input_specsrK   output_specsr   N)
r   r   r   r    r   rv   r   rN   r  r2   r   r3   r4   r  r  c  s+    4	?B.//D,b011r3   r  c                   J    \ rS rSr% \\\   S4   \S'   \\\   S4   \S'   Srg)RangeConstraintii  r   min_valrK   max_valr   N)	r   r   r   r    r   r   rO   rN   r2   r   r3   r4   r#  r#  i  s*    x}b())x}b())r3   r#  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
)ModuleCallSignatureio  r   r   rK   r   r   in_specrn   out_specNrp   forward_arg_namesr   )r   r   r   r    r   rv   r   rN   rM   r*  r   r2   r   r3   r4   r'  r'  o  sk    d8nb())tH~r)** sBwR   =Ay$s)!4b!89@r3   r'  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   rM   rN   r.  r   r'  r2   r   r3   r4   r,  r,  ~  s+    	37	>BIy"56:;Br3   r,  c                   0    \ rS rSr% \\\   S4   \S'   Srg)NamedTupleDefi  r   field_namesr   N)	r   r   r   r    r   rv   rM   rN   r2   r   r3   r4   r0  r0    s    49b=))r3   r0  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   rp   r.  rr   module_call_graphr   rn   r   rt   treespec_namedtuple_fieldsr   N)r   r   r   r    r   r   rN   r  rv   r,  r   r   r   rM   r5  r0  r2   r   r3   r4   r3  r3    s~    UBY+,, !o!6!:;;.3D.IHiS#X*+IJO`dJe	$sM/A*BB*F Ger3   r3  c                   >    \ rS rSr% \\S4   \S'   \\S4   \S'   Srg)SchemaVersioni  r   majorrK   minorr   N)r   r   r   r    r   rO   rN   r2   r   r3   r4   r7  r7    s     S"WS"Wr3   r7  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rg)ExportedProgrami  r   graph_modulerK   opset_versionr   range_constraintsrr   schema_versionr   rt   	verifiersz<=2.4r   torch_versionr   N)r   r   r   r    r   r3  rN   r   rM   rO   r#  r7  r   rv   r@  rA  r2   r   r3   r4   r;  r;    s    KO,,T#s(^R/00 c?&:!;R!?@@mR/00*/*EIycB'E(/M9S"W%/r3   r;  c                   4    \ rS rSr% \\\\4   S4   \S'   Sr	g)Programi  r   methodsr   N)
r   r   r   r    r   r   rM   r;  rN   r2   r   r3   r4   rC  rC    s    tC013677r3   rC  c                       \ rS rSr% \\S4   \S'   \\\\4   S4   \S'   \\S4   \S'   \\\\4   S4   \S	'   \\\\4   S
4   \S'   \\\\4   S4   \S'   Sr	g)Modeli  r   r   rK   tensorPathsrn   programrp   	delegatesrr   deviceAllocationMaprt   constantPathsr   N)
r   r   r   r    r   rM   rN   r   rC  r2   r   r3   r4   rF  rF    s     CG
 4S>2-..w{## c7l+R/00"4S>2#566 T#s(^R/00r3   rF  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   rM   rN   rv   rQ  r   rO   rR  rS  rW   r2   r   r3   r4   rM  rM    s      Q'' 49a<(( DIqL))
 ?C	(3-*: ;B?C8C=!+;!<C ?Cy$):;Br3   rM  c                   >    \ rS rSr% \\S4   \S'   \\S4   \S'   Srg)ExternKernelNodei  r   r   rK   noder   N)	r   r   r   r    r   rM   rN   r   r2   r   r3   r4   rU  rU    s"     CG

D"H
r3   rU  c                   0    \ rS rSr% \\\   S4   \S'   Srg)ExternKernelNodesi  r   r   r   N)	r   r   r   r    r   rv   rU  rN   r2   r   r3   r4   rX  rX    s    T*+R/00r3   rX  N)Adataclassesr   r   enumr   typingr   r   torch._export.serde.unionr   SCHEMA_VERSIONTREESPEC_VERSIONr   r6   rA   rH   rR   rZ   r_   rc   rf   ri   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,  r0  r3  r7  r;  rC  rF  rM  rU  rX  r   r3   r4   <module>r_     s   )  & ,  
 (W 7  / / /
 #& # # 6 6 6
 V   #v # #
 !f ! !
 " " "  V   #v # # !f ! !
       !V ! !
 " " "
 " " " :v : :87  7 7 7 F F F W W W ! ! !
 !F ! ! ( ( (
 ' ' '
 $ $ $ - - -
 ( ( (
 & & & < < < ! ! ! ' ' ' $ $ $
 ' ' '
 ( ( (
 ( ( (
 & & & * * * 2 2 2
 * * *
 A A A C C C
 * * * 	f 	f 	f   
 0 0 0 8 8 8 1 1 1. C C C(   
 1 1 1r3   