
    /j                    V    d dl mZ d dlZd dlZd dlmZ d dlmZ  G d de          Z	dS )    )annotationsN)Base)expectc                  V    e Zd Zedd            Zedd            Zedd            ZdS )EyeLikereturnNonec                 0   d} t           j                            ddgdg          }t          j                            dd| t          j                  }t          j        | d         | d	         t          j        
          }t          ||g|gd           d S )N)   r   r   xy)inputsoutputsr   d   sizedtype   r   test_eyelike_without_dtyper   r   name)	onnxhelper	make_nodenprandomrandintint32eyer   shapenoder   r   s       h/home/longshao/multi-rider-rag/.venv/lib/python3.11/site-packages/onnx/backend/test/case/node/eyelike.pyexport_without_dtypezEyeLike.export_without_dtype   s    {$$5E % 
 
 Ia5AAF58U1XRX666tQC!3OPPPPPP    c                 P   d} t           j                            ddgdgt           j        j                  }t
          j                            dd| t
          j                  }t          j	        | d         | d	         t
          j
        
          }t          ||g|gd           d S )N)   r   r   r   r   )r   r   r   r   r   r   r   r   test_eyelike_with_dtyper   )r   r   r   TensorProtoDOUBLEr   r   r   r   r    float64r   r!   s       r$   export_with_dtypezEyeLike.export_with_dtype   s    {$$5E")	 % 
 
 Ia5AAF58U1XRZ888tQC!3LMMMMMMr&   c                 X   d} d}t           j                            ddgdg|t           j        j                  }t
          j                            dd| t
          j        	          }t          j	        | d         | d         |t
          j
        
          }t          ||g|gd           d S )N)r      r   r   r   r   )r   r   kr   r   r   r   )r0   r   'test_eyelike_populate_off_main_diagonalr   )r   r   r   r*   FLOATr   r   r   r   r    float32r   )r"   off_diagonal_offsetr#   r   r   s        r$   !export_populate_off_main_diagonalz)EyeLike.export_populate_off_main_diagonal)   s    {$$5E!"( % 
 
 Ia5AAF58U1X)<BJOOO3C:		
 	
 	
 	
 	
 	
r&   N)r   r	   )__name__
__module____qualname__staticmethodr%   r-   r5    r&   r$   r   r      sz        
Q 
Q 
Q \
Q N N N \N 
 
 
 \
 
 
r&   r   )

__future__r   numpyr   r   onnx.backend.test.case.baser   onnx.backend.test.case.noder   r   r:   r&   r$   <module>r?      s    # " " " " "      , , , , , , . . . . . ./
 /
 /
 /
 /
d /
 /
 /
 /
 /
r&   