
    iY#                        S SK JrJr  S SKJr  S SKJrJrJrJ	r	  S SK
Js  Js  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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, S-5      5       r$\ " S. S/5      5       r%\ " S0 S15      5       r&\ " S2 S35      5       r'\" SS9 " S4 S5\5      5       r(\ " S6 S75      5       r)\ " S8 S95      5       r*\ " S: S;5      5       r+\ " S< S=5      5       r,\ " S> S?5      5       r-\ " S@ SA5      5       r.\" SS9 " SB SC\5      5       r/\ " SD SE5      5       r0\ " SF SG5      5       r1\ " SH SI5      5       r2\ " SJ SK5      5       r3\ " SL SM5      5       r4\ " SN SO5      5       r5\ " SP SQ5      5       r6\" SS9 " SR SS\5      5       r7\ " ST SU5      5       r8\ " SV SW5      5       r9\ " SX SY5      5       r:\ " SZ S[5      5       r;\ " S\ S]5      5       r<\ " S^ S_5      5       r=\ " S` Sa5      5       r>\ " Sb Sc5      5       r?\ " Sd Se5      5       r@g)f    )	dataclassfield)IntEnum)DictListOptionalTupleN)_Union)         c                   P    \ 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__static_attributes__r       [/var/www/html/ai-image-ml/venv/lib/python3.13/site-packages/executorch/exir/serde/schema.pyr   r      sM    GDDE
CDDEFKLMDHFr1   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Stridedr0   r   r1   r2   r4   r4   (   s*    GIIIIIGGr1   r4   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    r6   ContiguousFormatChannelsLastChannelsLast3dPreserveFormatr0   r   r1   r2   r?   r?   3   s    GLNNr1   r?   c                   4    \ rS rSr% \\S'   Sr\\   \S'   Sr	g)Device;   typeNindexr   )
r   r   r   r    str__annotations__rI   r   intr0   r   r1   r2   rF   rF   ;   s    
IE8C=r1   rF   F)reprc                   4    \ rS rSr% \\S'   \\S'   \\S'   Srg)SymExprHintA   as_intas_floatas_boolr   N)	r   r   r   r    rL   rK   floatboolr0   r   r1   r2   rO   rO   A   s    KOMr1   rO   c                   4    \ rS rSr% \\S'   Sr\\   \S'   Sr	g)SymExprL   expr_strNhintr   )
r   r   r   r    rJ   rK   rZ   r   rO   r0   r   r1   r2   rW   rW   L   s    M"&D(;
&r1   rW   c                   *    \ rS rSr% \\S'   \\S'   Srg)SymIntR   as_exprrQ   r   N)r   r   r   r    rW   rK   rL   r0   r   r1   r2   r\   r\   R   s    Kr1   r\   c                   *    \ rS rSr% \\S'   \\S'   Srg)SymBoolX   r^   rS   r   N)r   r   r   r    rW   rK   rU   r0   r   r1   r2   r`   r`   X   s    Mr1   r`   c                   h    \ rS rSr% \\S'   \\   \S'   \\S'   \	\S'   \\   \S'   \\S'   \
\S'   S	rg
)
TensorMeta^   dtypesizesrequires_graddevicestridesstorage_offsetlayoutr   N)r   r   r   r    r   rK   r   r\   rU   rF   r4   r0   r   r1   r2   rc   rc   ^   s2    <N&\Nr1   rc   c                   *    \ rS rSr% \\S'   \\S'   Srg)SymIntArgumento   as_namerQ   r   N)r   r   r   r    rJ   rK   rL   r0   r   r1   r2   rm   rm   o   s    LKr1   rm   c                   *    \ rS rSr% \\S'   \\S'   Srg)SymBoolArgument{   ro   rS   r   N)r   r   r   r    rJ   rK   rU   r0   r   r1   r2   rq   rq   {   s    LMr1   rq   c                        \ rS rSr% \\S'   Srg)TensorArgument   namer   Nr   r   r   r    rJ   rK   r0   r   r1   r2   rt   rt          
Ir1   rt   c                        \ rS rSr% \\S'   Srg)TokenArgument   rv   r   Nrw   r   r1   r2   rz   rz      rx   r1   rz   c                   0    \ rS rSr% \\S'   \S   \S'   Srg)OptionalTensorArgument   	as_tensorr   as_noneN)r   r   r   r    rt   rK   r	   r0   r   r1   r2   r}   r}      s    2Yr1   r}   c                   *    \ rS rSr% \\S'   S\S'   Srg)GraphArgument   rv   Graphgraphr   Nrw   r   r1   r2   r   r      s    
INr1   r   c                   *    \ rS rSr% \\S'   \\S'   Srg)CustomObjArgument   rv   	class_fqnr   Nrw   r   r1   r2   r   r      s    
INr1   r   c                   2   \ rS rSr% \S   \S'   \\S'   \\   \S'   \\S'   \\   \S'   \	\S'   \\	   \S	'   \
\S
'   \\
   \S'   \\S'   \\   \S'   \\S'   \\S'   \\S'   \\S'   \\S'   \\   \S'   \\S'   \\   \S'   \\S'   \\   \S'   \\S'   \
\S'   Srg)Argument   r   r   r   
as_tensorsrQ   as_intsrR   	as_floats	as_string
as_strings
as_sym_intas_sym_intsas_scalar_typeas_memory_format	as_layout	as_devicerS   as_boolsas_sym_boolas_sym_boolsas_graphas_optional_tensorsas_custom_objas_operatorN)r   r   r   r    r	   rK   rt   r   rL   rT   rJ   rm   r   r?   r4   rF   rU   rq   r   r}   r   r0   r   r1   r2   r   r      s    2Y^$$K#YOE{NS	n%%""M4j  ''455$$r1   r   c                   *    \ rS rSr% \\S'   \\S'   Srg)NamedArgument   rv   argr   N)r   r   r   r    rJ   rK   r   r0   r   r1   r2   r   r      s     I	Mr1   r   c                   ^    \ rS rSr% \\S'   \\S'   \\   \S'   \\   \S'   \	\\4   \S'   Sr
g)	Node   rv   targetinputsoutputsmetadatar   N)r   r   r   r    rJ   rK   r   r   r   r   r0   r   r1   r2   r   r      s/    
IK(^38nr1   r   c                       \ rS rSr% \\   \S'   \\   \S'   \\   \S'   \\	\
4   \S'   \\	\4   \S'   \\	\4   \S'   Sr\\S	'   \" \S
9r\\	\4   \S'   Srg)r      r   r   nodestensor_valuessym_int_valuessym_bool_valuesFis_single_tensor_returndefault_factorycustom_obj_valuesr   N)r   r   r   r    r   r   rK   r   r   rJ   rc   r\   r`   r   rU   r   dictr   r   r0   r   r1   r2   r   r      su    N(^:Z((f%%#w,''
 %*T)6;D6QtC!223Qr1   r   c                        \ rS rSr% \\S'   Srg)UserInputSpec   r   r   Nr   r   r   r    r   rK   r0   r   r1   r2   r   r      s	     
Mr1   r   c                   N    \ rS rSr% \S   \S'   \\S'   \\S'   \\S'   \	\S'   Sr
g)	ConstantValue   r   r   rQ   rR   r   rS   N)r   r   r   r    r	   rK   rL   rT   rJ   rU   r0   r   r1   r2   r   r      s     2YKONMr1   r   c                   *    \ rS rSr% \\S'   \\S'   Srg)ConstantInputSpec   rv   valuer   N)r   r   r   r    rJ   rK   r   r0   r   r1   r2   r   r      s    
Ir1   r   c                   *    \ rS rSr% \\S'   \\S'   Srg)InputToParameterSpec   r   parameter_namer   Nr   r   r   r    rt   rK   rJ   r0   r   r1   r2   r   r          	r1   r   c                   4    \ rS rSr% \\S'   \\S'   \\S'   Srg)InputToBufferSpec   r   buffer_name
persistentr   N)	r   r   r   r    rt   rK   rJ   rU   r0   r   r1   r2   r   r      s    	r1   r   c                   *    \ rS rSr% \\S'   \\S'   Srg)InputToTensorConstantSpec   r   tensor_constant_namer   Nr   r   r1   r2   r   r      s    	r1   r   c                   *    \ rS rSr% \\S'   \\S'   Srg)InputToCustomObjSpeci  r   custom_obj_namer   N)r   r   r   r    r   rK   rJ   r0   r   r1   r2   r   r     s    	r1   r   c                        \ rS rSr% \\S'   Srg)InputTokenSpeci  r   r   Nr   r   r   r    rz   rK   r0   r   r1   r2   r   r         	r1   r   c                   \    \ rS rSr% \\S'   \\S'   \\S'   \\S'   \	\S'   \
\S'   \\S'   S	rg
)	InputSpeci  
user_input	parameterbuffertensor_constant
custom_objtokenconstant_inputr   N)r   r   r   r    r   rK   r   r   r   r   r   r   r0   r   r1   r2   r   r     s,    ##..$$%%r1   r   c                        \ rS rSr% \\S'   Srg)UserOutputSpeci  r   r   Nr   r   r1   r2   r   r     s    	Mr1   r   c                        \ rS rSr% \\S'   Srg)LossOutputSpeci   r   r   N)r   r   r   r    rt   rK   r0   r   r1   r2   r   r      s    	r1   r   c                   *    \ rS rSr% \\S'   \\S'   Srg)BufferMutationSpeci%  r   r   r   Nr   r   r1   r2   r   r   %  s    	r1   r   c                   *    \ rS rSr% \\S'   \\S'   Srg)GradientToParameterSpeci+  r   r   r   Nr   r   r1   r2   r   r   +  r   r1   r   c                   *    \ rS rSr% \\S'   \\S'   Srg)GradientToUserInputSpeci1  r   user_input_namer   Nr   r   r1   r2   r   r   1      	r1   r   c                   *    \ rS rSr% \\S'   \\S'   Srg)UserInputMutationSpeci7  r   r   r   Nr   r   r1   r2   r   r   7  r   r1   r   c                        \ rS rSr% \\S'   Srg)OutputTokenSpeci=  r   r   Nr   r   r1   r2   r   r   =  r   r1   r   c                   \    \ rS rSr% \\S'   \\S'   \\S'   \\S'   \	\S'   \
\S'   \\S'   S	rg
)
OutputSpeciB  user_outputloss_outputbuffer_mutationgradient_to_parametergradient_to_user_inputuser_input_mutationr   r   N)r   r   r   r    r   rK   r   r   r   r   r   r   r0   r   r1   r2   r   r   B  s,    ''2233..r1   r   c                   6    \ rS rSr% \\   \S'   \\   \S'   Srg)GraphSignatureiM  input_specsoutput_specsr   N)	r   r   r   r    r   r   rK   r   r0   r   r1   r2   r   r   M  s    i z""r1   r   c                   *    \ rS rSr% \\S'   \\S'   Srg)RangeConstraintiS  min_valmax_valr   Nr   r   r   r    rL   rK   r0   r   r1   r2   r   r   S  s    LLr1   r   c                   J    \ rS rSr% \\   \S'   \\   \S'   \\S'   \\S'   Srg)ModuleCallSignatureiY  r   r   in_specout_specr   N)	r   r   r   r    r   r   rK   rJ   r0   r   r1   r2   r  r  Y  s"    N(^ LMr1   r  c                   4    \ rS rSr% \\S'   Sr\\   \S'   Sr	g)ModuleCallEntryid  fqnN	signaturer   )
r   r   r   r    rJ   rK   r  r   r  r0   r   r1   r2   r  r  d  s    	H/3Ix+,3r1   r  c                   :    \ rS rSr% \\S'   \\S'   \\   \S'   Sr	g)GraphModuleij  r   r  module_call_graphr   N)
r   r   r   r    r   rK   r   r   r  r0   r   r1   r2   r	  r	  j  s    L O,,r1   r	  c                   *    \ rS rSr% \\S'   \\S'   Srg)SchemaVersioniv  majorminorr   Nr   r   r1   r2   r  r  v  s    JJr1   r  c                       \ rS rSr% \\S'   \\\4   \S'   \\\	4   \S'   \
\S'   \\S'   \" \S9r\\   \S'   S	r\\S'   S
rg)ExportedProgrami|  graph_moduleopset_versionrange_constraintsschema_versiondialectr   	verifiers r   N)r   r   r   r    r	  rK   r   rJ   rL   r   r  r   listr  r   r  r0   r   r1   r2   r  r  |  sN    S>!C011!!L 6ItCy6GSr1   r  c                   *    \ rS rSr% \\S'   \\S'   Srg)CompileSpeci  keyr   r   Nrw   r   r1   r2   r  r    s    	HJr1   r  c                       \ rS rSr% \\S'   \\S'   \\   \S'   \R                  \S'   \\S'   \\S'   Sr
\\   \S	'   S
rg)LoweredBackendModulei  
backend_idprocessed_bytescompile_specsoriginal_moduleoriginal_state_dictoriginal_constantsNnamed_data_storer   )r   r   r   r    rJ   rK   r   r  export_schemar  r$  r   bytesr0   r   r1   r2   r  r    s?    O$$"222(,huo,r1   r  )Adataclassesr   r   enumr   typingr   r   r   r	   executorch.exir.serde.schemaexirserdeschemar%  executorch.exir.serde.unionr
   SCHEMA_VERSIONTREESPEC_VERSIONr   r4   r?   rF   rO   rW   r\   r`   rc   rm   rq   rt   rz   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   r1   r2   <module>r1     s   )  . . 4 4 .   "W 7       
 &   ' ' '
 V  
 f  
     V   f  
       V  
   
    v  4       R R R   
 F     
   
      
   
    & & &         
   
   
   
       # # #
   
    4 4 4
 - - -   
      
 - - -r1   