
    Ng                     J    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 )    N)Base)expectc                   V    e Zd Zedd            Zedd            Zedd            ZdS )EyeLikereturnNc                  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       _/var/www/html/ai-engine/env/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   )r.   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$   )r   N)__name__
__module____qualname__staticmethodr#   r+   r3    r$   r"   r   r      sz        
Q 
Q 
Q \
Q N N N \N 
 
 
 \
 
 
r$   r   )numpyr   r   onnx.backend.test.case.baser   onnx.backend.test.case.noder   r   r8   r$   r"   <module>r<      st   
      , , , , , , . . . . . ./
 /
 /
 /
 /
d /
 /
 /
 /
 /
r$   