
    i                        S SK Jr  S SKJr  S SKJrJrJr  S SKJ	r	  S SK
Jr  \ " S S5      5       r\ " S S	5      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\\\\\\\\\\\\4   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!\ " S0 S15      5       r"\ " S2 S35      5       r#\ " S4 S55      5       r$\\ \!\"\#\$4   r%\ " S6 S75      5       r&\ " S8 S95      5       r'\ " S: S;5      5       r( " S< S=\5      r)\ " S> S?5      5       r*\ " S@ SA5      5       r+\ " SB SC5      5       r,\ " SD SE5      5       r-\ " SF SG5      5       r.\ " SH SI5      5       r/\ " SJ SK5      5       r0\ " SL SM5      5       r1\ " SN SO5      5       r2gP)Q    )	dataclass)IntEnum)ListOptionalUnion)CompileSpec)
ScalarTypec                   L    \ rS rSr% \\S'   \\S'   \\S'   \S\4S j5       rSrg)	AllocationDetails   	memory_idmemory_offset_lowmemory_offset_highreturnc                 :    U R                   U R                  S-  -  $ )N    )r   r   selfs    U/var/www/html/ai-image-ml/venv/lib/python3.13/site-packages/executorch/exir/schema.pymemory_offsetAllocationDetails.memory_offset   s    %%)@)@B)FGG     N)	__name__
__module____qualname____firstlineno__int__annotations__propertyr   __static_attributes__r   r   r   r   r      s/    NHs H Hr   r   c                   &    \ rS rSr% \\   \S'   Srg)OptionalTensorList   itemsr   Nr   r   r   r   r   r   r   r!   r   r   r   r#   r#          9r   r#   c                   $    \ rS rSrSrSrSrSrSrg)TensorShapeDynamism$   2
Check program.fbs for explanations of this enum.
r         r   N)	r   r   r   r   __doc__STATICDYNAMIC_BOUNDDYNAMIC_UNBOUNDr!   r   r   r   r)   r)   $   s     FMOr   r)   c                       \ rS rSrSrSrSrg)TensorDataLocation.   r   r,   r   N)r   r   r   r   SEGMENTEXTERNALr!   r   r   r   r3   r3   .   s    GHr   r3   c                   ^    \ rS rSr% SrSr\\S'   Sr\	\
   \S'   \R                  r\\S'   Srg)	ExtraTensorInfo3   r+   r   mutable_data_segments_idxNfully_qualified_namelocationr   )r   r   r   r   r.   r:   r   r   r;   r   strr3   r5   r<   r!   r   r   r   r8   r8   3   s4     &'s&*.(3-.#5#=#=H =r   r8   c                       \ rS rSr% \\S'   \\S'   \\   \S'   \\   \S'   \\S'   \\S'   \\S'   \	\
   \S	'   \\S
'   Sr\	\   \S'   Srg)Tensor>   scalar_typestorage_offsetsizes	dim_orderrequires_gradlayoutdata_buffer_idxallocation_infoshape_dynamismNextra_tensor_infor   )r   r   r   r   r	   r   r   r   boolr   r   r)   rJ   r8   r!   r   r   r   r?   r?   >   sW    9CyK/00 ('37x07r   r?   c                       \ rS rSrSrg)NullN   r   N)r   r   r   r   r!   r   r   r   rM   rM   N   s    r   rM   c                        \ rS rSr% \\S'   Srg)IntS   int_valr   Nr   r   r   r   r   r   r!   r   r   r   rP   rP   S   s    Lr   rP   c                        \ rS rSr% \\S'   Srg)BoolX   bool_valr   N)r   r   r   r   rK   r   r!   r   r   r   rU   rU   X   s    Nr   rU   c                   F    \ rS rSr% \\\4   \S'   S\SS4S jrSS jr	Sr
g)	Double]   
double_valr   Nc                 l    U[        S5      :X  a  SU l        g U[        S5      :X  a  SU l        g Xl        g )Ninf-inf)floatr[   )r   r[   s     r   __init__Double.__init__a   s-    u%#DO5=($DO(Or   c                    [        U R                  [        5      (       a  U R                  S;   d   eg [        U R                  [        5      (       d   eU R                  [        S5      :X  a   eU R                  [        S5      :X  a   eg )N)r]   r^   r]   r^   )
isinstancer[   r=   r_   r   s    r   __post_init__Double.__post_init__i   sk    doos++??o555doou5555%,666%-7777r   )r[   )r   N)r   r   r   r   r   r_   r=   r   r`   rd   r!   r   r   r   rY   rY   ]   s)    eSj!!)5 )T )8r   rY   c                        \ rS rSr% \\S'   Srg)Stringr   
string_valr   Nr   r   r   r   r=   r   r!   r   r   r   rg   rg   r   s    Or   rg   c                   *    \ rS rSr% \\S'   \\S'   Srg)ContainerMetadataw   encoded_inp_strencoded_out_strr   Nrj   r   r   r   rl   rl   w   s    r   rl   c                   &    \ rS rSr% \\   \S'   Srg)IntList}   r%   r   Nr&   r   r   r   rq   rq   }   r'   r   rq   c                   &    \ rS rSr% \\   \S'   Srg)
DoubleList   r%   r   N)r   r   r   r   r   r_   r   r!   r   r   r   rt   rt          ;r   rt   c                   &    \ rS rSr% \\   \S'   Srg)BoolList   r%   r   N)r   r   r   r   r   rK   r   r!   r   r   r   rx   rx      s    :r   rx   c                   &    \ rS rSr% \\   \S'   Srg)
TensorList   r%   r   Nr&   r   r   r   r{   r{      r'   r   r{   c                        \ rS rSr% S\S'   Srg)EValue   KernelTypesvalr   Nr   r   r   r   r   r!   r   r   r   r~   r~      s
     
r   r~   c                        \ rS rSr% \\S'   Srg)Buffer   storager   Nr   r   r   r   bytesr   r!   r   r   r   r   r      s    Nr   r   c                        \ rS rSr% \\S'   Srg)BackendDelegateInlineData   datar   Nr   r   r   r   r   r      s    
Kr   r   c                   0    \ rS rSr% \\S'   \\   \S'   Srg)
KernelCall   op_indexargsr   Nr   r   r   r   r   r   r   r!   r   r   r   r   r      s    M
s)Or   r   c                   0    \ rS rSr% \\S'   \\   \S'   Srg)DelegateCall   delegate_indexr   r   Nr   r   r   r   r   r      s    
s)Or   r   c                   *    \ rS rSr% \\S'   \\S'   Srg)MoveCall   	move_frommove_tor   NrS   r   r   r   r   r      s    NLr   r   c                   *    \ rS rSr% \\S'   \\S'   Srg)JumpFalseCall   cond_value_indexdestination_instructionr   NrS   r   r   r   r   r      s      r   r   c                        \ rS rSr% \\S'   Srg)FreeCall   value_indexr   NrS   r   r   r   r   r      s    r   r   c                        \ rS rSr% S\S'   Srg)Instruction   InstructionArguments
instr_argsr   Nr   r   r   r   r   r      s    &&r   r   c                   >    \ rS rSr% \\S'   \\S'   \\S'   \\S'   Srg)Frame   filenamelinenonamecontextr   Nr   r   r   r   r=   r   r   r!   r   r   r   r   r      s    MK
ILr   r   c                   &    \ rS rSr% \\   \S'   Srg)	FrameList   r%   r   N)r   r   r   r   r   r   r   r!   r   r   r   r   r      rv   r   r   c                       \ rS rSrSrSrSrg)DataLocation   r   r,   r   N)r   r   r   r   INLINEr5   r!   r   r   r   r   r      s    FGr   r   c                   *    \ rS rSr% \\S'   \\S'   Srg)BackendDelegateDataReference   r<   indexr   N)r   r   r   r   r   r   r   r!   r   r   r   r   r      s    Jr   r   c                   :    \ rS rSr% \\S'   \\S'   \\   \S'   Sr	g)BackendDelegate   id	processedcompile_specsr   N)
r   r   r   r   r=   r   r   r   r   r!   r   r   r   r   r      s    G++$$r   r   c                   \    \ rS rSr% \\   \S'   \\   \S'   \\   \S'   \\\	      \S'   Sr
g)Chain   inputsoutputsinstructions
stacktracer   N)r   r   r   r   r   r   r   r   r   r   r!   r   r   r   r   r      s0    I#Y{##i))r   r   c                   *    \ rS rSr% \\S'   \\S'   Srg)Operatori  r   overloadr   Nrj   r   r   r   r   r     s    
IMr   r   c                       \ rS rSr% \\S'   \\S'   \\   \S'   \\	   \S'   \\	   \S'   \\
   \S'   \\   \S'   \\   \S	'   \\	   \S
'   Srg)ExecutionPlani  r   container_meta_typevaluesr   r   chains	operators	delegatesnon_const_buffer_sizesr   N)r   r   r   r   r=   r   rl   r   r~   r   r   r   r   r!   r   r   r   r   r     sV    
I**LI#YKH~O$$
 !I%r   r   c                   *    \ rS rSr% \\S'   \\S'   Srg)DataSegmenti  offsetsizer   NrS   r   r   r   r   r     s    K
Ir   r   c                   0    \ rS rSr% \\S'   \\   \S'   Srg)SubsegmentOffsetsi  segment_indexoffsetsr   Nr   r   r   r   r   r     s    #Yr   r   c                   *    \ rS rSr% \\S'   \\S'   Srg)	NamedDatai%  keyr   r   Nr   r   r   r   r   r   %  s    	Hr   r   c                       \ rS rSr% \\S'   \\   \S'   \\   \S'   \\	   \S'   \\
   \S'   \\S'   Sr\\\      \S	'   Sr\\\      \S
'   Srg)Programi+  versionexecution_planconstant_bufferbackend_delegate_datasegmentsconstant_segmentNmutable_data_segments
named_datar   )r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r!   r   r   r   r   r   +  sb    L''&\! 9::;''?C8D):$;<C,0Ji)0r   r   N)3dataclassesr   enumr   typingr   r   r   +executorch.exir.backend.compile_spec_schemar   executorch.exir.scalar_typer	   r   r#   r)   r3   r8   r?   rM   rP   rU   rY   rg   rl   rq   rt   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   <module>r      s   "  ( ( C 2 	H 	H 	H   '  
 > > > 8 8 8 	 	 	       8 8 8(      
             



   
         
   
   
 ! ! !
    	  ' ' '      7 
   
 % % % * * *   
 & & &    
   
   
 1 1 1r   