
    /jA                    f    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Z	d	dZ
 G d de          ZdS )
    )annotationsN)Base)expectc                    | j         ||         }t          j        |                              t          j                  S )N)shapenparrayastypeint64)xstartenddimss       f/home/longshao/multi-rider-rag/.venv/lib/python3.11/site-packages/onnx/backend/test/case/node/shape.pyshape_reference_implr      s0    759D8D>>  ***    c                    t           j                            ddgdg||          }t          |||          }t	          ||g|gd| z              d S )NShaper   y)inputsoutputsr   r   
test_shape)r   r   name)onnxhelper	make_noder   r   )testnamexvalr   r   nodeyvals         r   r   r      sf    ;  uEs !  D  eS11D
4\H5LMMMMMMr   c                  &    e Zd Zedd            ZdS )r   returnNonec                 z   t          j        g dg dg                              t           j                  } t	          d|            t           j                            ddd                              t           j                  } t	          d|            t	          d| d	
           t	          d| d	           t	          d| d
           t	          d| d           t	          d| d	d           t	          d| d	d           t	          d| d
           t	          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
   _start_greater_than_end)r   r	   r
   float32r   randomrandn)r   s    r   exportzShape.export   sQ   H				
 

 &

 	
 	:q!!!IOOAq!$$++BJ772q:q****8QA&&&&&4444$aR0000,aqbAAAA#QaQ7777=!3////;r****,aqa@@@@@@r   N)r"   r#   )__name__
__module____qualname__staticmethodr;    r   r   r   r      s8        A A A \A A Ar   r   )NN)
__future__r   numpyr   r   onnx.backend.test.case.baser   onnx.backend.test.case.noder   r   r   r   r@   r   r   <module>rE      s    # " " " " "      , , , , , , . . . . . .+ + + +
N N N NA A A A AD A A A A Ar   