o
    q::j                     @  sd   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dZ		ddddZ
G dd deZdS )    )annotationsN)Base)expectx
np.ndarraystart
int | Noneendreturnc                 C  s    | j || }t|tjS )N)shapenparrayastypeint64)r   r   r	   dims r   h/home/nk/hobo-godmode/plappi-mvp/.venv/lib/python3.10/site-packages/onnx/backend/test/case/node/shape.pyshape_reference_impl   s   r   testnamestrxvalNonec                 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   )r   r   r   r	   nodeyvalr   r   r   r      s
   r   c                   @  s   e Zd ZedddZdS )r   r
   r   c                  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 td| 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
   _start_greater_than_end)r   r   r   float32r   randomrandn)r   r   r   r   export"   s&   

zShape.exportN)r
   r   )__name__
__module____qualname__staticmethodr9   r   r   r   r   r   !   s    r   )NN)r   r   r   r   r	   r   r
   r   )
r   r   r   r   r   r   r	   r   r
   r   )
__future__r   numpyr   r   onnx.backend.test.case.baser   onnx.backend.test.case.noder   r   r   r   r   r   r   r   <module>   s   