U
    rhP                     @  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                   @  sB   e Zd ZeddddZeddddZeddddZd	S )
ConstantOfShapeNone)returnc                  C  sr   t d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         value   r   xyinputsoutputsr   ZdtypeZtest_constantofshape_float_onesr   r   name)nparrayastypeint64onnxhelpermake_tensorTensorProtoFLOAT	make_nodeZonesZfloat32r   r   Ztensor_valuenoder    r!   O/tmp/pip-unpacked-wheel-xnis5xre/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   Ztest_constantofshape_int_zerosr   r   r   r   r   r   r   r   r   ZINT32r   zerosZ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   Z#test_constantofshape_int_shape_zeror   r&   r   r!   r!   r"   export_int32_shape_zero-   s2          z'ConstantOfShape.export_int32_shape_zeroN)__name__
__module____qualname__staticmethodr#   r(   r)   r!   r!   r!   r"   r      s   r   )

__future__r   Znumpyr   r   Zonnx.backend.test.case.baser   Zonnx.backend.test.case.noder   r   r!   r!   r!   r"   <module>   s
   