o
    wi                     @  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astypeint64)xstartenddims r   ^/home/ubuntu/sommelier/.venv/lib/python3.10/site-packages/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helper	make_noder   r   )testnamexvalr   r   nodeyvalr   r   r   r      s
   r   c                   @  s   e Zd ZedddZdS )r   returnNonec                  C  s   t g dg d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)         )         _exampler"   r#   r$    _start_1r    )r   _end_1)r   _start_negative_1_end_negative_1_start_1_end_negative_1)r   r   _start_1_end_2r!   _clip_starti	_clip_end
   )r   r   r   float32r   randomrandn)r
   r   r   r   export   s$   

zShape.exportN)r   r   )__name__
__module____qualname__staticmethodr5   r   r   r   r   r      s    r   )NN)
__future__r   numpyr   r   onnx.backend.test.case.baser   onnx.backend.test.case.noder   r   r   r   r   r   r   r   <module>   s   


