
    h                     B    S SK rS SKrS SKJr  S SKJr   " S S\5      rg)    N)Base)expectc                   P    \ rS rSr\SS j5       r\SS j5       r\SS j5       rSrg)EyeLike   Nc                     Sn [         R                  R                  SS/S/S9n[        R                  R                  SSU [        R                  S9n[        R                  " U S   U S	   [        R                  S
9n[        X/U/SS9  g )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/fran/franai/venv/lib/python3.13/site-packages/onnx/backend/test/case/node/eyelike.pyexport_without_dtypeEyeLike.export_without_dtype   s    {{$$5E % 
 IIa5AFF58U1XRXX6tC!3OP    c                  F   Sn [         R                  R                  SS/S/[         R                  R                  S9n[
        R                  R                  SSU [
        R                  S9n[
        R                  " U S   U S	   [
        R                  S
9n[        X/U/SS9  g )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_dtypeEyeLike.export_with_dtype   s    {{$$5E""))	 % 
 IIa5AFF58U1XRZZ8tC!3LMr%   c                  P   Sn Sn[         R                  R                  SS/S/U[         R                  R                  S9n[
        R                  R                  SSU [
        R                  S	9n[
        R                  " U S   U S   U[
        R                  S
9n[        UU/U/SS9  g )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_diagonal)EyeLike.export_populate_off_main_diagonal(   s    {{$$5E!""(( % 
 IIa5AFF58U1X)<BJJO3C:		
r%    )returnN)	__name__
__module____qualname____firstlineno__staticmethodr#   r,   r5   __static_attributes__r7   r%   r"   r   r      s@    
Q 
Q N N 
 
r%   r   )numpyr   r   onnx.backend.test.case.baser   onnx.backend.test.case.noder   r   r7   r%   r"   <module>rB      s   
   , ./
d /
r%   