ó
    ÀíØhÉ  ã                   ó~   • S SK r S SKrS SKJrJr   " S S\ R                  5      r\S:X  a  \ R                  " 5         gg)é    N)ÚparserÚprinterc                   óD   • \ rS rSrS\R
                  SS4S jrSS jrSrg)	ÚTestBasicFunctionsé
   ÚgraphÚreturnNc                 óD  • U R                  [        UR                  5      S5        U R                  UR                  S   R                  S5        U R                  UR                  S   R                  S5        U R                  UR                  S   R                  S5        g )Né   r   ÚMatMulé   ÚAddé   ÚSoftmax)ÚassertEqualÚlenÚnodeÚop_type)Úselfr   s     ÚP/var/www/fran/franai/venv/lib/python3.13/site-packages/onnx/test/printer_test.pyÚcheck_graphÚTestBasicFunctions.check_graph   st   € Ø×Ñœ˜UŸZ™Z›¨!Ô,Ø×Ñ˜Ÿ™ A™×.Ñ.°Ô9Ø×Ñ˜Ÿ™ A™×.Ñ.°Ô6Ø×Ñ˜Ÿ™ A™×.Ñ.°	Õ:ó    c                 óü   • Sn[         R                  " U5      n[        R                  " U5      n[         R                  " U5      n[        R                  " U5      nU R	                  X55        U R                  U5        g )NzÐ
           agraph (float[N, 128] X, float[128,10] W, float[10] B) => (float[N] C)
           {
              T = MatMul(X, W)
              S = Add(T, B)
              C = Softmax(S)
           }
           )r   Úparse_graphr   Úto_textr   r   )r   Útext0Úgraph1Útext1Úgraph2Útext2s         r   Útest_parse_graphÚ#TestBasicFunctions.test_parse_graph   se   € ðˆô ×#Ò# EÓ*ˆÜ—’ Ó'ˆÜ×#Ò# EÓ*ˆÜ—’ Ó'ˆð 	×Ñ˜Ô&Ø×Ñ˜Õ r   © )r	   N)	Ú__name__Ú
__module__Ú__qualname__Ú__firstlineno__ÚonnxÚ
GraphProtor   r"   Ú__static_attributes__r$   r   r   r   r   
   s   † ð; §¡ð ;°Tô ;÷!r   r   Ú__main__)Úunittestr)   r   r   ÚTestCaser   r%   Úmainr$   r   r   Ú<module>r0      s:   ðó ã ß  ô!˜×*Ñ*ô !ð6 ˆzÓØ‡M‚M…Oð r   