
    h                     N    S SK Jr  S SKrS SKrS SKJr  S SKJr   " S S\5      r	g)    )SequenceN)Base)expectc                   (    \ rS rSr\SS j5       rSrg)ExpandDynamicShape   Nc            
          S[         R                  R                  S[        [           S[        [           S[        [           S[         R                  R
                  4
S jn [         R                  R                  SSS	/S
/SS9n/ SQn[        R                  " U[        R                  S9n[        R                  " SS/[        R                  S9nU[        R                  " U[        R                  S9-  nU " XUR                  UR                  5      n[         R                  R                  US[         R                  R                  SS5      /S9n[        XsU/U/SS9  [        R                  " SS/[        R                  S9nU[        R                  " U[        R                  S9-  nU " XUR                  UR                  5      n[         R                  R                  US[         R                  R                  SS5      /S9n[        XsU/U/SS9  [        R                  " / SQ[        R                  S9nU[        R                  " U[        R                  S9-  nU " XUR                  UR                  5      n[         R                  R                  US[         R                  R                  SS5      /S9n[        XsU/U/SS9  [        R                  " / SQ[        R                  S9nU[        R                  " U[        R                  S9-  nU " XUR                  UR                  5      n[         R                  R                  US[         R                  R                  SS5      /S9n[        XsU/U/SS9  g )Nnodeinput_shapeshape_shapeoutput_shapereturnc           
         [         R                  R                  U /S[         R                  R                  S[         R                  R
                  U5      [         R                  R                  S[         R                  R                  U5      /[         R                  R                  S[         R                  R
                  U5      /S9nU$ )NExpandXshapeY)nodesnameinputsoutputs)onnxhelper
make_graphmake_tensor_value_infoTensorProtoFLOATINT64)r
   r   r   r   graphs        ]/var/www/fran/franai/venv/lib/python3.13/site-packages/onnx/backend/test/case/model/expand.pyr   -ExpandDynamicShape.export.<locals>.make_graph   s     KK**fKK66T--33[ KK66!1!1!7!7	 KK66T--33\ + E" L    r   r   r   r   test)r   )      r$   )dtyper%   r$   zbackend-test 	   )producer_nameopset_importstest_expand_shape_model1)r   r   r   test_expand_shape_model2)r%   r$   r%   test_expand_shape_model3)r%   r%   r$   r%   test_expand_shape_model4)r   r   	NodeProtor   int
GraphProto	make_nodenponesfloat32arrayint64r   make_model_gen_versionmake_opsetidr   )r   r
   r   xr   yr   models           r    exportExpandDynamicShape.export   s   	++''	!#	 "#	 #3-		
 [[##	2 {{$$XW~u6$RGGKrzz2 !Qrxx0RZZ004ekk177C22(;;33B:; 3 

 	uZ!;UV !Qrxx0RZZ004ekk177C22(;;33B:; 3 

 	uZ!;UV "((3RZZ004ekk177C22(;;33B:; 3 

 	uZ!;UV RXX6RZZ004ekk177C22(;;33B:; 3 

 	uZ!;UVr"    )r   N)__name__
__module____qualname____firstlineno__staticmethodr=   __static_attributes__r?   r"   r    r   r      s    HW HWr"   r   )
typingr   numpyr3   r   onnx.backend.test.case.baser   onnx.backend.test.case.modelr   r   r?   r"   r    <module>rJ      s&   
    , /JW JWr"   