
    Ng                     V    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	dS )    )SequenceN)Base)expectc                   &    e Zd Zedd            ZdS )ExpandDynamicShapereturnNc            
         dt           j        j        dt          t                   dt          t                   dt          t                   dt           j        j        f
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input_shapeshape_shapeoutput_shaper   c           
      h   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   graphs        _/var/www/html/ai-engine/env/lib/python3.11/site-packages/onnx/backend/test/case/model/expand.pyr   z-ExpandDynamicShape.export.<locals>.make_graph   s     K**fK66T-3[  K66!1!7 	 K66T-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   exportzExpandDynamicShape.export   sM   	+'	!#	 "#	 #3-		
 [#	 	 	 	2 {$$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    )r   N)__name__
__module____qualname__staticmethodr;    r    r   r   r      s>        HW HW HW \HW HW HWr    r   )
typingr   numpyr1   r   onnx.backend.test.case.baser   onnx.backend.test.case.modelr   r   r@   r    r   <module>rE      s   
            , , , , , , / / / / / /JW JW JW JW JW JW JW JW JW JWr    