
    /j                    r    d dl mZ d dlmZ d dlZd dlZd dlmZ d dl	m
Z
 erd dlmZ  G d de          ZdS )	    )annotations)TYPE_CHECKINGN)Base)expect)Sequencec                  &    e Zd Zedd            ZdS )ExpandDynamicShapereturnNonec                 B   dd	} t           j                            d
ddgdgd          }g d}t          j        |t          j                  }t          j        ddgt          j                  }|t          j        |t          j                  z  } | |||j        |j                  }t           j        	                    |dt           j        
                    dd          g          }t          |||g|gd           t          j        ddgt          j                  }|t          j        |t          j                  z  } | |||j        |j                  }t           j        	                    |dt           j        
                    dd          g          }t          |||g|gd           t          j        g dt          j                  }|t          j        |t          j                  z  } | |||j        |j                  }t           j        	                    |dt           j        
                    dd          g          }t          |||g|gd           t          j        g dt          j                  }|t          j        |t          j                  z  } | |||j        |j                  }t           j        	                    |dt           j        
                    dd          g          }t          |||g|gd           d S ) Nnodeonnx.helper.NodeProtoinput_shapeSequence[int]shape_shapeoutput_shaper
   onnx.helper.GraphProtoc           
     d   t           j                            | gdt           j                            dt           j        j        |          t           j                            dt           j        j        |          gt           j                            dt           j        j        |          g          S )NExpandXshapeY)nodesnameinputsoutputs)onnxhelper
make_graphmake_tensor_value_infoTensorProtoFLOATINT64)r   r   r   r   s       h/home/longshao/multi-rider-rag/.venv/lib/python3.11/site-packages/onnx/backend/test/case/model/expand.pyr   z-ExpandDynamicShape.export.<locals>.make_graph   s     ;))fK66T-3[  K66!1!7 	 K66T-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   r   r   r   r   r   r   r   r
   r   )r   r   	make_nodenponesfloat32arrayint64r   make_model_gen_versionmake_opsetidr   )r   r   r   xr   ygraphmodels           r$   exportzExpandDynamicShape.export   s	   	 	 	 	0 {$$XW~u6$RRiiGKrz222 !Qrx000RZ0000
4ek17CC22(;33B::; 3 
 

 	uaZ!;UVVVV !Qrx000RZ0000
4ek17CC22(;33B::; 3 
 

 	uaZ!;UVVVV "(333RZ0000
4ek17CC22(;33B::; 3 
 

 	uaZ!;UVVVV RX666RZ0000
4ek17CC22(;33B::; 3 
 

 	uaZ!;UVVVVVVr%   N)r
   r   )__name__
__module____qualname__staticmethodr>    r%   r$   r	   r	      s>        GW GW GW \GW GW GWr%   r	   )
__future__r   typingr   numpyr3   r   onnx.backend.test.case.baser   onnx.backend.test.case.modelr   collections.abcr   r	   rC   r%   r$   <module>rJ      s    # " " " " "                  , , , , , , / / / / / / )((((((IW IW IW IW IW IW IW IW IW IWr%   