
    h                     B    S SK rS SKrS SKJr  S SKJr   " S S\5      rg)    N)Base)expectc                   (    \ rS rSr\SS j5       rSrg)
ShrinkTest   Nc            
         [         R                  R                  SS/S/SSS9n [         R                  R                  U /S[         R                  R	                  S[         R
                  R                  S/5      /[         R                  R	                  S[         R
                  R                  S/5      /S9n[         R                  R                  US[         R                  R                  S	S
5      /S9n[        R                  " / SQ[        R                  S9n[        R                  " / SQ[        R                  S9n[        X#/U/SS9  g )NShrinkxyg      ?)lambdbias   )nodesnameinputsoutputszbackend-test 
   )producer_nameopset_imports)g       g              g      ?g       @)dtype)g      r   r   r   g      ?test_shrink)r   r   r   )onnxhelper	make_node
make_graphmake_tensor_value_infoTensorProtoFLOATmake_model_gen_versionmake_opsetidnparrayfloat32r   )nodegraphmodelr
   r   s        ]/var/www/fran/franai/venv/lib/python3.13/site-packages/onnx/backend/test/case/model/shrink.pyexportShrinkTest.export   s   {{$$EE % 
 &&&2238H8H8N8NQRPST 2238H8H8N8NQRPST ' 	
 22(;;33B;< 3 
 HH0

CHH/rzzBuS1#MB     )returnN)__name__
__module____qualname____firstlineno__staticmethodr*   __static_attributes__r-   r,   r)   r   r      s    C Cr,   r   )numpyr#   r   onnx.backend.test.case.baser   onnx.backend.test.case.modelr   r   r-   r,   r)   <module>r8      s!   
   , /C Cr,   