o
    wiP                     @  sH   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 deZ	dS )    )annotationsN)Base)expectc                   @  s6   e Zd Zed
ddZed
ddZed
ddZd	S )ConstantOfShapereturnNonec                  C  sp   t g dt j} tjdtjjdgdg}tjj	ddgdg|d}t j
| t jd}t|| g|gd	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    r&   h/home/ubuntu/sommelier/.venv/lib/python3.10/site-packages/onnx/backend/test/case/node/constantofshape.pyexport_float_ones   s   z!ConstantOfShape.export_float_onesc                  C  sp   t ddgt j} tjdtjjdgdg}tjj	ddgdg|d	}t j
| t jd
}t|| g|gd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#   r&   r&   r'   export_int32_zeros   s   z"ConstantOfShape.export_int32_zerosc                  C  sn   t dgt j} tjdtjjdgdg}tjj	ddgdg|d}t j
| t jd}t|| g|gd	d
 d S )Nr   r   r   r   r   r   r   r   #test_constantofshape_int_shape_zeror   r,   r#   r&   r&   r'   export_int32_shape_zero-   s&   
z'ConstantOfShape.export_int32_shape_zeroN)r   r   )__name__
__module____qualname__staticmethodr(   r0   r2   r&   r&   r&   r'   r      s    r   )

__future__r   numpyr   r   onnx.backend.test.case.baser   onnx.backend.test.case.noder   r   r&   r&   r&   r'   <module>   s   