
    bIg                    Z    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
  G d de      Zy)    )annotations)SequenceN)Base)expectc                      e Zd Zedd       Zy)ExpandDynamicShapec                    	 	 	 	 	 	 	 	 	 	 dd} t         j                  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                  j                  |dt         j                  j                  dd      g      }t        |||g|gd       t        j                  dd
gt        j                  	      }|t        j                  |t        j
                  	      z  } | |||j                  |j                        }t         j                  j                  |dt         j                  j                  dd      g      }t        |||g|gd       t        j                  g dt        j                  	      }|t        j                  |t        j
                  	      z  } | |||j                  |j                        }t         j                  j                  |dt         j                  j                  dd      g      }t        |||g|gd       t        j                  g dt        j                  	      }|t        j                  |t        j
                  	      z  } | |||j                  |j                        }t         j                  j                  |dt         j                  j                  dd      g      }t        |||g|gd       y )Nc           
        t         j                  j                  | gdt         j                  j                  dt         j                  j
                  |      t         j                  j                  dt         j                  j                  |      gt         j                  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)nodeinput_shapeshape_shapeoutput_shapegraphs        `/var/www/html/answerous/venv/lib/python3.12/site-packages/onnx/backend/test/case/model/expand.pyr   z-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   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   yr   models           r   exportzExpandDynamicShape.export   s   	'	&	 '	 (		
 $	2 {{$$XW~u6$RGGKrzz2 !Qrxx0RZZ004ekk177C22(;;33B:; 3 

 	uaZ!;UV !Qrxx0RZZ004ekk177C22(;;33B:; 3 

 	uaZ!;UV "((3RZZ004ekk177C22(;;33B:; 3 

 	uaZ!;UV RXX6RZZ004ekk177C22(;;33B:; 3 

 	uaZ!;UVr    N)r.   None)__name__
__module____qualname__staticmethodr:    r    r   r   r      s    HW HWr    r   )
__future__r   typingr   numpyr0   r   onnx.backend.test.case.baser   onnx.backend.test.case.modelr   r   r@   r    r   <module>rF      s)    #    , /JW JWr    