
    /jP                    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 )    )annotationsN)Base)expectc                  V    e Zd Zedd            Zedd            Zedd            ZdS )ConstantOfShapereturnNonec                    t          j        g d                              t           j                  } t          j                            dt          j        j        dgdg          }t          j        	                    ddgdg|          }t          j
        | t           j                  }t          || g|gd	
           d S )N)         value   r   xyinputsoutputsr   dtypetest_constantofshape_float_onesr   r   name)nparrayastypeint64onnxhelpermake_tensorTensorProtoFLOAT	make_nodeonesfloat32r   r   tensor_valuenoder   s       p/home/longshao/multi-rider-rag/.venv/lib/python3.11/site-packages/onnx/backend/test/case/node/constantofshape.pyexport_float_onesz!ConstantOfShape.export_float_ones   s    HYYY&&rx00{..T%+aS1#
 
 {$$5E	 % 
 
 GARZ(((tQC!3TUUUUUU    c                    t          j        ddg                              t           j                  } t          j                            dt          j        j        dgdg          }t          j        	                    ddgdg|	          }t          j
        | t           j        
          }t          || g|gd           d S )N
      r   r   r   r   r   r   r   r   test_constantofshape_int_zerosr   r   r   r   r   r   r   r    r!   INT32r#   zerosint32r   r&   s       r)   export_int32_zerosz"ConstantOfShape.export_int32_zeros   s    Hb!W$$RX..{..T%+aS1#
 
 {$$5E	 % 
 
 HQbh'''tQC!3STTTTTTr+   c                    t          j        dg                              t           j                  } t          j                            dt          j        j        dgdg          }t          j        	                    ddgdg|          }t          j
        | t           j                  }t          || g|gd	
           d S )Nr   r   r   r   r   r   r   r   #test_constantofshape_int_shape_zeror   r0   r&   s       r)   export_int32_shape_zeroz'ConstantOfShape.export_int32_shape_zero-   s    H
 
 &

	 	

 {..T%+aS1#
 
 {$$5E	 % 
 
 HQbh'''!qc0U	
 	
 	
 	
 	
 	
r+   N)r   r	   )__name__
__module____qualname__staticmethodr*   r4   r7    r+   r)   r   r      sz        V V V \V U U U \U 
 
 
 \
 
 
r+   r   )

__future__r   numpyr   r   onnx.backend.test.case.baser   onnx.backend.test.case.noder   r   r<   r+   r)   <module>rA      s    # " " " " "      , , , , , , . . . . . .3
 3
 3
 3
 3
d 3
 3
 3
 3
 3
r+   