
    Ng-                     J    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 )    N)Base)expectc                   V    e Zd Zedd            Zedd            Zedd            ZdS )ConstantOfShapereturnNc                     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       g/var/www/html/ai-engine/env/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   r.   r$   s       r'   export_int32_shape_zeroz'ConstantOfShape.export_int32_shape_zero,   s    H
 
 &

	 	

 {..T%+aS1#
 
 {$$5E	 % 
 
 HQbh'''!qc0U	
 	
 	
 	
 	
 	
r)   )r   N)__name__
__module____qualname__staticmethodr(   r2   r5    r)   r'   r   r      sz        V V V \V U U U \U 
 
 
 \
 
 
r)   r   )numpyr   r   onnx.backend.test.case.baser   onnx.backend.test.case.noder   r   r:   r)   r'   <module>r>      st   
      , , , , , , . . . . . .3
 3
 3
 3
 3
d 3
 3
 3
 3
 3
r)   