
    h6                     f    S SK r  " S S\ R                  5      r\S:X  a  \ R                  " 5         gg)    Nc                       \ rS rSrSS jrSrg)TestONNXRuntime
   Nc                 R    SS K nASSKJnJn  SSK Jn  SSKJ	n  SSK
JnJnJnJn	  U" S5      n
U" U
5      nUR                  U5        UR                  USS	9  UR!                  USSSS
9nU	R#                  US5      nU" UR%                  5       S/S9nUR'                  5       S   R(                  nUR+                  5       S   R(                  nUR                  S5      nUR-                  U5      nUR/                  U/UU05        g ! [         a    [        R                  " S5      S ef = f)Nr   zonnxruntime not installed)float32random)InferenceSession)get_example)checkerloadshape_inferenceversion_converterzsigmoid.onnxT)
full_check)
check_typestrict_mode	data_propr   CPUExecutionProvider)	providers)         )onnxruntimeImportErrorunittestSkipTestnumpyr   r   r	   onnxruntime.datasetsr
   onnxr   r   r   r   check_modelinfer_shapesconvert_versionSerializeToString
get_inputsnameget_outputsastyperun)selfr   r   r   r	   r
   r   r   r   r   example1modelinferred_modelconverted_modelsess
input_nameoutput_namexs                     Q/var/www/fran/franai/venv/lib/python3.13/site-packages/onnx/test/test_with_ort.pytest_with_ort_example%TestONNXRuntime.test_with_ort_example   s0   	K 	*04JJ ~. XE"Ed3(55d 6 
 ,;;NBO  --/<R;S
 __&q)..
&&(+00MM)$HHW+Q0=  	K##$?@dJ	Ks   D "D& )returnN)__name__
__module____qualname____firstlineno__r2   __static_attributes__r4       r1   r   r   
   s    #1r;   r   __main__)r   TestCaser   r6   mainr4   r;   r1   <module>r?      s5    $1h'' $1N zMMO r;   