
    Si                    &   S SK Jr  S SKrS SKrS SKrS SKJrJr  S SKr	S SK
r	S SKr	S SKr	S SK	JrJrJr  S SKJrJr  S SKJr  \(       a
  S SKJr  S SKr " S S	\	R.                  R0                  R2                  5      r1 S
krSS jrSSS00r\	R.                  R<                  R?                  \\ \S9r!\RD                  " 5       S    S:X  a  \!RG                  S5        \$" 5       RK                  \!RL                  5        \ S:X  a  \RN                  " 5         gg)    )annotationsN)TYPE_CHECKINGAny)
ModelProto	NodeProtoTensorProto)Device
DeviceType)!BackendIsNotSupposedToImplementIt)Sequencec                     ^  \ rS rSr\ S       SU 4S jjj5       r\  S           S	U 4S jjj5       r\S
S j5       rSrU =r	$ )DummyBackend$   c                4  > [         TU ]  " X40 UD6  [        R                  R	                  U5        SSS.UEn[        R
                  R                  " U40 UD6n[        R                  " UR                  R                  UR                  R                  5       Vs0 s H  nUR                  U_M     nn[        U5      (       a  UR                  R                   H  n[        UR                  5       H  u  pxUR                   S:X  a  US:w  a  M  X;   d   eXX   R"                  R$                  n	U	R&                  [(        R*                  :w  d   eU	R,                  R.                   H  n
U
R1                  S5      S:X  a  M   e   M     M     [3        S5      es  snf )NT)
check_typestrict_modeDropoutr   value	dim_valueWThis is the dummy backend test that doesn't verify the results but does run the checker)superprepareonnxcheckercheck_modelshape_inferenceinfer_shapes	itertoolschaingraph
value_infooutputname!do_enforce_test_coverage_safelistnode	enumerateop_typetypetensor_type	elem_typer   	UNDEFINEDshapedim
WhichOneofr   )clsmodeldevicekwargsvivalue_infosr%   ir"   ttr-   	__class__s              Z/var/www/html/ai-image-ml/venv/lib/python3.13/site-packages/onnx/test/test_backend_test.pyr   DummyBackend.prepare%   s`    	00  ' !%TDVD$$11%B6B  ooekk&<&<ekk>P>PQ
Q GGRKQ 	 

 -U33((!*4;;!7IA||y0Q!V !000$,11==B<<;+@+@@@@!xx||"~~g6+EEE  , "8 ) 0e
 	
!
s   Fc                6   > [         TU ]  XX4S9  [        S5      e)N)r1   outputs_infor   )r   run_noder   )r/   r%   inputsr1   r;   r2   r7   s         r8   r<   DummyBackend.run_nodeE   s'     	fP/e
 	
    c                R    [        U5      nUR                  [        R                  :H  $ )N)r	   r(   r
   CPU)r/   r1   ds      r8   supports_deviceDummyBackend.supports_deviceS   s    6Nvv''r?    )rA   )r0   r   r1   strr2   r   returnz#onnx.backend.base.BackendRep | None)rA   N)r%   r   r=   r   r1   rF   r;   z4Sequence[tuple[numpy.dtype, tuple[int, ...]]] | Noner2   r   rG   ztuple[Any, ...] | None)r1   rF   rG   bool)
__name__
__module____qualname____firstlineno__classmethodr   r<   rC   __static_attributes____classcell__)r7   s   @r8   r   r   $   s    .3

(+
?B
	,
 
> 
 MQ

 
 	

 K
 
 
 
 
 ( (r?   r   >
   vgg19zfnetresnet50
SingleRelu
shufflenetdensenet121bvlc_alexnetinception_v1inception_v2squeezenet_oldc                    U R                   R                  [        ;  a  g[        S U R                   R                   5       5      $ )NFc              3  >   #    U  H  oR                   S ;  v   M     g7f)>   GRURNNLSTMN)r'   ).0r%   s     r8   	<genexpr>4do_enforce_test_coverage_safelist.<locals>.<genexpr>j   s     WFVd||#99FVs   )r    r#   test_coverage_safelistallr%   )r0   s    r8   r$   r$   g   s3    {{55WekkFVFVWWWr?   test_mvnr   F)test_kwargs32bitz)(test_vgg19|test_zfnet|test_bvlc_alexnet)__main__)r0   r   rG   rH   )(
__future__r   r   platformunittesttypingr   r   onnx.backend.baser   onnx.backend.testonnx.shape_inferenceonnx.version_converterr   r   r   r	   r
   onnx.backend.test.runnerr   collections.abcr   numpybackendbaseBackendr   rb   r$   re   testBackendTestrI   backend_testarchitectureexcludeglobalsupdate
test_casesmainrE   r?   r8   <module>r      s    #    %     3 3 0 F(2(4<<$$,, 2(j X &
 ||  ,,( -  1(EF 	  (( )zMMO r?   