
    hz/                    l   S SK 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  S SK	r	S SK
Jr  S SKrS SKrS SKrS SKrS SK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
KJr   " S S\R8                  R:                  R<                  5      r " S S\R8                  R:                  R@                  5      r!\R8                  RD                  RG                  \!\$5      r%\RL                  " S5      (       a  \%RO                  S5        \RP                  " 5       S    S:X  a  \%RO                  S5        \RR                  " 5       S:X  a  \%RO                  S5        \%RO                  S5        \%RO                  S5        \%RO                  S5        \%RO                  S5        \%RO                  S5        \%RO                  S5        \b#  \" \5      \" S5      :  a  \%RO                  S5        \b#  \" \5      \" S5      :  a  \%RO                  S5        \RT                  SS  S!:  d  \" \	R,                  5      \" S"5      :  a  \%RO                  S#5        \b   \+" 5       RY                  \%RZ                  5        \$S$:X  a  \R\                  " S S%S&9r/\/R`                  Rb                  r2\3" \/R`                  Rh                  5      r4\3" \/R`                  Rj                  5      r5\3" \/R`                  Rl                  5      r7\3" \/R`                  Rp                  5      r9\:" S'5        \:" S(\2 S)\4 S*\5 S+\7 S,\9 3
5        gg! \ a    SrSrS r GNf = f)-    )annotationsN)AnyClassVar)Version)
ModelProto)Device
DeviceType)InferenceSession)__version__)get_available_providers)InvalidArgumentc                     / $ N r       \/var/www/fran/franai/venv/lib/python3.13/site-packages/onnx/test/test_backend_onnxruntime.pyr   r   !   s    	r   c                       \ rS rSrS rS rSrg)InferenceSessionBackendRep(   c                    Xl         g r   _session)selfsessions     r   __init__#InferenceSessionBackendRep.__init__)   s    r   c                   [        U[        R                  5      (       a  U/n[        U[        5      (       a  U R                  R                  5        Vs/ s H  o3R                  PM     nnU R                  R                  5        Vs/ s H  o3R                  PM     nn[        U5      [        U5      :X  a  [        [        XA5      5      nOz0 nSn[        XE5       H5  u  pXU   R                  :X  d  M  X   Xh'   US-  nU[        U5      :  d  M5    O4   O1[        U[        5      (       a  UnO[        S[        U5      < S35      eU R                  R                  S U5      n
U
$ s  snf s  snf )Nr      zUnexpected input type .)
isinstancenumpyndarraylistr   
get_inputsnameshapelendictzip	TypeErrortyperun)r   inputskwargsiinput_namesinput_shapesfeeds
pos_inputsinpr&   outss              r   r,   InferenceSessionBackendRep.run,   s2   femm,,XFfd##+/==+C+C+EF+Ea66+EKF-1]]-E-E-GH-GGG-GLH6{c+..S56
"%k"@JCz 2 8 88%+%7
"a
%V4! #A %%E4T&\4DAFGG}}  u-% GHs   E#E(r   N)__name__
__module____qualname____firstlineno__r   r,   __static_attributes__r   r   r   r   r   (   s     r   r   c                      \ rS rSr% \" \" 5       5      rS\S'   \S 5       r	\SS j5       r
\S 5       r\S 5       r\ S       SS jj5       r\SS
 j5       r\SS j5       rSrg	)InferenceSessionBackendE   zClassVar[set[str]]	providersc                    g)N)T r   )clsmodels     r   is_opset_supported*InferenceSessionBackend.is_opset_supportedH   s    r   c                    [        U5      nUR                  [        R                  :X  a  SU R                  ;   a  gUR                  [        R
                  :X  a  SU R                  ;   a  gg)NCPUExecutionProviderTCUDAExecutionProviderF)r   r+   r	   CPUr?   CUDA)rB   deviceds      r   supports_device'InferenceSessionBackend.supports_deviceL   sL    6N66Z^^#(>#--(O66Z__$)@CMM)Qr   c                   UR                    Vs0 s H  o"R                  UR                  _M     nnSU;  a  g  [        R                  R                  XS   S-
  5      $ s  snf ! [         a_    UR                   S S 2	 UR                  5        H7  u  pEUR                   R                  5       nXBl        US:w  a  UOUS-
  Ul        M9     Us $ f = f)NrA   r   )	opset_importdomainversiononnxversion_converterconvert_versionRuntimeErroritemsadd)rB   rC   rL   opsetskvs         r   convert_version_opset_before4InferenceSessionBackend.convert_version_opset_beforeU   s    /4/A/AB/A!((AII%/ABV		))99%aPP	 C
  	""1%&&**,!"bAa!e	 ' L	s    A$A# #A&CCc                   US:X  a  S/nOUS:X  a  S/nO[        SU< S35      e [        UR                  5       US9$ ! [         az  nS[	        U5      ;   a+  U=R
                  S	-  sl        U R                  X5      s S nA$ S
[	        U5      ;   a*  U R                  U5      nUb  U R                  XR5      s S nA$ UeS nAff = f)NrI   rG   rJ   rH   zUnexepcted device r   )r?   zUnsupported model IR versionr   z9Current official support for domain ai.onnx is till opset)
ValueErrorr
   SerializeToStringr   str
ir_versioncreate_inference_sessionr\   )rB   rC   rK   r?   e	new_models         r   rc   0InferenceSessionBackend.create_inference_sessione   s    U?/0Iv01I1&1=>>
	#E$;$;$=SS 	-Q7  A% 33EBBJcRSfT<<UC	(77	JJG	s-   > 
C4B=<C3B=5C;B==Cc                    [        U[        5      (       a  [        U5      $ [        U[        [        [
        45      (       a$  U R                  X5      nU R                  " XB40 UD6$ [        S[        U5       S35      e)NzUnexpected type z for model.)
r    r
   r   ra   bytesr   rc   preparer*   r+   )rB   rC   rK   r.   infs        r   ri   InferenceSessionBackend.preparey   sn     e-..-e44ec5*566..u=C;;s5f55*4;-{CDDr   Nc                N    U R                   " X40 UD6nUR                  " U40 UD6$ r   )ri   r,   )rB   rC   r-   rK   r.   reps         r   	run_model!InferenceSessionBackend.run_model   s)    kk%262wwv(((r   c                    [        S5      e)Nz%Unable to run the model node by node.)NotImplementedError)rB   noder-   rK   outputs_infor.   s         r   run_node InferenceSessionBackend.run_node   s    !"IJJr   r   )rK   ra   returnbool)rI   )rC   r   rK   ra   r.   r   rv   r   r   )NN)r7   r8   r9   r:   setr   r?   __annotations__classmethodrD   rM   r\   rc   ri   rn   rt   r;   r   r   r   r=   r=   E   s    $'(?(A$BI!B       & ',
E
E!$
E8;
E	#
E 
E ) ) K Kr   r=   APPVEYORz(test_vgg19|test_zfnet)32bitz)(test_vgg19|test_zfnet|test_bvlc_alexnet)Windowstest_sequence_modelz(test_bernoulli)a  (test_adagrad|test_adam|test_add_uint8|bitshift_left_uint16|bitshift_right_uint16|cast_BFLOAT16_to_FLOAT|cast_FLOAT_to_BFLOAT16|castlike_BFLOAT16_to_FLOAT|castlike_FLOAT_to_BFLOAT16|clip_default_int8_min_expanded|clip_default_int8_max_expanded|div_uint8|gru_batchwise|loop16_seq_none|lstm_batchwise|m(in|ax)_u?int(16|8)|momentum|mul_uint8|pow_types_float32_uint32|pow_types_float32_uint64|simple_rnn_batchwise|sub_uint8|gradient_of_add|test_batchnorm_epsilon_training_mode|test_batchnorm_example_training_mode|_to_FLOAT8E4M3FN|_to_FLOAT8E5M2|cast_FLOAT8E|castlike_FLOAT8E|test_dequantizelinear_axis|test_dequantizelinear|test_quantizelinear_axis|test_quantizelinear|test_affine_grid_)z8(cast_FLOAT_to_STRING|castlike_FLOAT_to_STRING|dft|stft)zm(resize_downsample_scales_cubic_align_corners|resize_downsample_scales_linear_align_corners|training_dropout)z(test_reduce_sum_empty_set|test_reduce_prod_empty_set|test_reduce_min_empty_set|test_reduce_max_empty_set|test_reduce_sum_square_empty_set|test_reduce_log_sum_empty_set|test_reduce_log_sum_exp_empty_set|test_reduce_l1_empty_set|test_reduce_l2_empty_set)aR  (maxunpool_export_with_output_shape|softplus_example_expanded|softplus_expanded|AvgPool[1-3]d|BatchNorm1d_3d_input_eval|BatchNorm[2-3]d_eval|GLU|Linear|PReLU|PoissonNLL|Softsign|operator_add_broadcast|operator_add_size1|operator_addconstant|operator_addmm|operator_basic|operator_mm|operator_non_float_params|operator_params|operator_pow)z1.16z(averagepool|_pad_|_resize_|_size_|cast|castlike|equal_string_broadcast|equal_string|equal|half_pixel_symmetric|identity|reshape)z1.17z(deform_conv|dft|gelu|gridsample|identity_opt|image_decoder|isinf_float16|label_encoder|optional_get_element_optional_sequence|reduce_max_bool_inputs|reduce_min_bool_inputs|regex_full_match|string_concat|string_split)   )      z1.23.5rA   __main__F)	verbosityexitz!---------------------------------z
tests_run=z errors=z	 skipped=z unexpected_successes=z expected_failures=);
__future__r   osplatformsysunittesttypingr   r   r!   packaging.versionr   onnx.backend.baserS   onnx.backend.testonnx.shape_inferenceonnx.version_converterr   r   r	   onnxruntimer
   r   ort_versionr   +onnxruntime.capi.onnxruntime_pybind11_stater   ImportErrorbackendbase
BackendRepr   Backendr=   testBackendTestr7   backend_testgetenvexcludearchitecturesystemversion_infoglobalsupdate
test_casesmainresresulttestsRun	tests_runr'   errorsskippedunexpectedSuccessesunexpected_successesexpectedFailuresexpected_failuresprintr   r   r   <module>r      s
  
 # 	  
     %      0,63K!2!2!=!= :HKdll//77 HKV ||  ,,-DhO99Z231(DE??	!./   ' (   #%P   O P      
   4 w{3gfoE	" w{3gfoE	& BQ& GE,=,=$>AR$R  I\,,- z
--!%
0C

##I""#F#**$$%Gszz==>CJJ778	
-.	
YKxxy	 B  45 6./	1 e  Ks   L" "L32L3