
    Si                    t    S SK Jr  S SKJr  S SKrS SKrS SKJr  S SK	J
r
  \(       a  S SKJr   " S S\5      rg)	    )annotations)TYPE_CHECKINGN)Base)expect)Sequencec                  (    \ rS rSr\SS j5       rSrg)ExpandDynamicShape   c                 ^             S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 )Nc           
        [         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ExpandXshapeY)nodesnameinputsoutputs)onnxhelper
make_graphmake_tensor_value_infoTensorProtoFLOATINT64)nodeinput_shapeshape_shapeoutput_shapes       b/var/www/html/ai-image-ml/venv/lib/python3.13/site-packages/onnx/backend/test/case/model/expand.pyr   -ExpandDynamicShape.export.<locals>.make_graph   s     ;;))fKK66T--33[ KK66!1!1!7!7	 KK66T--33\ *      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   zonnx.helper.NodeProtor   Sequence[int]r   r/   r   r/   returnzonnx.helper.GraphProto)r   r   	make_nodenponesfloat32arrayint64r   make_model_gen_versionmake_opsetidr   )r   r   r   xr   ygraphmodels           r    exportExpandDynamicShape.export   s   	'	&	 '	 (		
 $	0 {{$$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"    N)r0   None)__name__
__module____qualname____firstlineno__staticmethodr=   __static_attributes__r?   r"   r    r	   r	      s    GW GWr"   r	   )
__future__r   typingr   numpyr2   r   onnx.backend.test.case.baser   onnx.backend.test.case.modelr   collections.abcr   r	   r?   r"   r    <module>rM      s/    #     , /(IW IWr"   