U
    rh                     @  s\   d dl mZ d dlZd dlZd dlmZ d dlmZ dddZ	dddZ
G d	d
 d
eZdS )    )annotationsN)Base)expectc                 C  s    | j || }t|tjS )N)shapenparrayastypeZint64)xstartendZdims r   E/tmp/pip-unpacked-wheel-xnis5xre/onnx/backend/test/case/node/shape.pyshape_reference_impl   s    r   c                 C  sB   t jjddgdg||d}t|||}t||g|gd|  d d S )NShaper	   y)inputsoutputsr
   r   
test_shape)r   r   name)onnxhelperZ	make_noder   r   )ZtestnameZxvalr
   r   nodeZyvalr   r   r   r      s        r   c                   @  s   e Zd ZeddddZdS )r   None)returnc                  C  s   t dddgdddggt j} td|  t jdddt j} td|  td	| dd
 td| dd td| dd
 td| dd td| ddd td| ddd td| dd
 td| dd d S )N                  Z_example Z_start_1)r
   Z_end_1)r   Z_start_negative_1Z_end_negative_1Z_start_1_end_negative_1)r
   r   Z_start_1_end_2Z_clip_startiZ	_clip_end
   )r   r   r   Zfloat32r   randomZrandn)r	   r   r   r   export   s$    

zShape.exportN)__name__
__module____qualname__staticmethodr$   r   r   r   r   r      s   r   )NN)NN)
__future__r   Znumpyr   r   Zonnx.backend.test.case.baser   Zonnx.backend.test.case.noder   r   r   r   r   r   r   r   <module>   s   


