
    bIg                    N    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	y)    )annotationsN)Base)expectc                  B    e Zd Zedd       Zedd       Zedd       Zy)EyeLikec                 ,   d} t         j                  j                  ddgdg      }t        j                  j                  dd| t        j                        }t        j                  | d   | d	   t        j                  
      }t        ||g|gd       y )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/answerous/venv/lib/python3.12/site-packages/onnx/backend/test/case/node/eyelike.pyexport_without_dtypezEyeLike.export_without_dtype   s    {{$$5E % 
 IIa5AFF58U1XRXX6tQC!3OP    c                 ^   d} t         j                  j                  ddgdgt         j                  j                        }t
        j                  j                  dd| t
        j                        }t        j                  | d   | d	   t
        j                  
      }t        ||g|gd       y )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""))	 % 
 IIa5AFF58U1XRZZ8tQC!3LMr$   c                 f   d} d}t         j                  j                  ddgdg|t         j                  j                        }t
        j                  j                  dd| t
        j                  	      }t        j                  | d   | d   |t
        j                  
      }t        ||g|gd       y )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!""(( % 
 IIa5AFF58U1X)<BJJO3C:		
r$   N)returnNone)__name__
__module____qualname__staticmethodr#   r+   r3    r$   r"   r   r      s@    
Q 
Q 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$   