
    h\                         S SK r S SKrS SKrS SKJrJrJrJr   " S S\ R                  5      r	\
S:X  a  \ R                  " 5         gg)    N)TensorProtohelpernumpy_helpershape_inferencec                       \ rS rSrSS jrSrg)TestTrainingTool   Nc                 ~   SS/nSn[         R                  R                  " U6 R                  [         R                  5      n[
        R                  " X2S9n[        R                  " U[        R                  U5      nSS/nSn[         R                  R                  " U6 R                  [         R                  5      n[
        R                  " XS9n	[        R                  " U[        R                  U5      n
SS/nSn[        R                  " U[        R                  U5      n[        R                  " SX'/U/S9n[        R                  " U/SXZ/U/XI/5      nSS/nS	n[         R                  R                  " U6 R                  [         R                  5      n[
        R                  " UUS9n[        R                  " U[        R                  U5      nSS/nS
n[        R                  " U[        R                  U5      n[        R                  " SUU/U/S9n[        R                  " U/SU/U/U/5      n[        R                  " UUU4/S S 5      n[        R                  " U5      n[        R                   R#                  U5        UR$                  R'                  5       nUR)                  U5        [        R                  " [+        UR,                  R.                  5      [+        UR$                  S   R0                  R.                  5      -   S[+        UR,                  R2                  5      [+        UR$                  S   R0                  R2                  5      -   [+        UR,                  R4                  5      [+        UR$                  S   R0                  R4                  5      -   [+        UR,                  R6                  5      [+        UR$                  S   R0                  R6                  5      -   5      n[        R                  " U5      n[8        R:                  " U5      n[        R                   R#                  U5        g )N   A)nameBCMatMul)inputsoutputssimple_inferenceXYsimple_trainingr   full_training_graph)nprandomrandastypefloat32r   
from_arrayr   make_tensor_value_infor   FLOAT	make_node
make_graphmake_training_info
make_modelonnxcheckercheck_modeltraining_infoaddCopyFromlistgraphnode	algorithminputoutputinitializerr   infer_shapes) selfA_shapeA_namer   A_initializerA_value_infoB_shapeB_namer   B_initializerB_value_infoC_shapeC_nameC_value_infoinference_nodeinference_graphX_shapeX_namer   X_initializerX_value_infoY_shapeY_nameY_value_infor,   training_graphr'   modelnew_training_infor   full_training_modelfull_training_model_with_shapess                                    V/var/www/fran/franai/venv/lib/python3.13/site-packages/onnx/test/training_tool_test.pytest_training_info_proto)TestTrainingTool.test_training_info_proto   s   a&IINNG$++BJJ7$//?44V[=N=NPWXa&IINNG$++BJJ7$//?44V[=N=NPWXa&44V[=N=NPWX))f-x
 !++(N*
 a&IINNG$++BJJ7$//?44V[=N=NPWXa&44V[=N=NPWXF#H
  **F%~~

 11ff-.d

 !!/2  '!//335""=1 %//!!"T%*=*=a*@*J*J*O*O%PP!""#d5+>+>q+A+K+K+Q+Q&RR##$tE,?,?,B,L,L,S,S'TT(()5&&q)33??@A
 %//0CD*9*F*F+
' 	  !@A     )returnN)__name__
__module____qualname____firstlineno__rM   __static_attributes__rP   rO   rL   r   r      s	    RBrO   r   __main__)unittestnumpyr   r$   r   r   r   r   TestCaser   rR   mainrP   rO   rL   <module>r\      sE   
    C CSBx(( SBl zMMO rO   