
    i                     l    S SK JrJrJr  S SKJrJr  S SKJr  S SK	J
r
  S SKJr  \ " S S\5      5       rg)	    )finalOptionalSequence)EthosUBackendEthosUCompileSpec)TOSAPartitioner)DelegationSpec)OperatorSupportBasec                   @    \ rS rSrSr S	S\S\\\      SS4S jjr	Sr
g)
EthosUPartitioner   zPartitions subgraphs supported by the Arm Ethos-U backend.

Args:
    compile_spec: EthosUCompileSpec object for configuring the lowering.
    additional_checks: Optional sequence of additional operator support checks.

Ncompile_specadditional_checksreturnc                     [        [        R                  UR                  5       5      U l        X l        UR                  U l        g N)r	   r   __name__to_listdelegation_specr   	tosa_spec)selfr   r   s      i/var/www/html/ai-image-ml/venv/lib/python3.13/site-packages/executorch/backends/arm/ethosu/partitioner.py__init__EthosUPartitioner.__init__   s:      .""L$8$8$: 
 "3%//    )r   r   r   r   )r   
__module____qualname____firstlineno____doc__r   r   r   r
   r   __static_attributes__ r   r   r   r      s?     FJ
0'
0 $H-@$AB
0 
	
0 
0r   r   N)typingr   r   r   executorch.backends.arm.ethosur   r   (executorch.backends.arm.tosa.partitionerr   #executorch.exir.backend.partitionerr	    torch.fx.passes.operator_supportr
   r   r!   r   r   <module>r'      s4    - , K D > @ 0 0 0r   