
    Ng                     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 )	    N)Base)expectc                     | j         ||         }t          j        |                              t          j                  S )N)shapenparrayastypeint64)xstartenddimss       ]/var/www/html/ai-engine/env/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   returnNc                  T   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           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   s    r   exportzShape.export   s9   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******r   )r!   N)__name__
__module____qualname__staticmethodr8    r   r   r   r      s2        + + + \+ + +r   r   )NN)
numpyr   r   onnx.backend.test.case.baser   onnx.backend.test.case.noder   r   r   r   r=   r   r   <module>rA      s   
      , , , , , , . . . . . .+ + + +
N N N N+ + + + +D + + + + +r   