
    bIgP                    N    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)    )annotationsN)Base)expectc                  B    e Zd Zedd       Zedd       Zedd       Zy)ConstantOfShapec                    t        j                  g d      j                  t         j                        } t        j
                  j                  dt        j                  j                  dgdg      }t        j
                  j                  ddgdg|      }t        j                  | t         j                        }t        || g|gd	
       y )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       h/var/www/html/answerous/venv/lib/python3.12/site-packages/onnx/backend/test/case/node/constantofshape.pyexport_float_onesz!ConstantOfShape.export_float_ones   s    HHY&&rxx0{{..T%%++aS1#
 {{$$5E	 % 
 GGARZZ(tQC!3TU    c                    t        j                  ddg      j                  t         j                        } t        j
                  j                  dt        j                  j                  dgdg      }t        j
                  j                  ddgdg|	      }t        j                  | t         j                  
      }t        || g|gd       y )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    HHb!W$$RXX.{{..T%%++aS1#
 {{$$5E	 % 
 HHQbhh'tQC!3STr)   c                    t        j                  dg      j                  t         j                        } t        j
                  j                  dt        j                  j                  dgdg      }t        j
                  j                  ddgdg|      }t        j                  | t         j                        }t        || g|gd	
       y )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    HH
 &
	 	

 {{..T%%++aS1#
 {{$$5E	 % 
 HHQbhh'!qc0U	
r)   N)returnNone)__name__
__module____qualname__staticmethodr(   r2   r5    r)   r'   r   r      s@    V V 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
d 3
r)   