
    bIg                    N    d dl mZ d dlZd dlZd dlmZ d dlmZ  G d de      Z	y)    )annotationsN)Base)expectc                      e Zd Zedd       Zy)
ShrinkTestc            
        t         j                  j                  ddgdgdd      } t         j                  j                  | gdt         j                  j	                  dt         j
                  j                  dg      gt         j                  j	                  dt         j
                  j                  dg      g      }t         j                  j                  |dt         j                  j                  d	d
      g      }t        j                  g dt        j                        }t        j                  g dt        j                        }t        ||g|gd       y )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/html/answerous/venv/lib/python3.12/site-packages/onnx/backend/test/case/model/shrink.pyexportzShrinkTest.export   s   {{$$EE % 
 &&&2238H8H8N8NQRPST 2238H8H8N8NQRPST ' 	
 22(;;33B;< 3 
 HH0

CHH/rzzBuaS1#MB    N)returnNone)__name__
__module____qualname__staticmethodr*    r+   r)   r   r      s    C Cr+   r   )

__future__r   numpyr#   r   onnx.backend.test.case.baser   onnx.backend.test.case.modelr   r   r2   r+   r)   <module>r7      s$    #   , /C Cr+   