o
    wi                     @  sH   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 deZ	dS )    )annotationsN)Base)expectc                   @  s6   e Zd Zed
ddZed
ddZed
ddZd	S )EyeLikereturnNonec                  C  sb   d} t jjddgdgd}tjjdd| tjd}tj| d | d	 tjd
}t||g|gd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
    r!   `/home/ubuntu/sommelier/.venv/lib/python3.10/site-packages/onnx/backend/test/case/node/eyelike.pyexport_without_dtype   s   zEyeLike.export_without_dtypec                  C  sh   d} t jjddgdgt jjd}tjjdd| tjd}tj	| d | d	 tj
d
}t||g|gd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   r!   r!   r"   export_with_dtype   s   zEyeLike.export_with_dtypec                  C  sp   d} d}t jjddgdg|t jjd}tjjdd| tjd	}tj	| d | d |tj
d
}t||g|gd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
   r!   r!   r"   !export_populate_off_main_diagonal)   s"   
z)EyeLike.export_populate_off_main_diagonalN)r   r   )__name__
__module____qualname__staticmethodr#   r)   r0   r!   r!   r!   r"   r      s    r   )

__future__r   numpyr   r   onnx.backend.test.case.baser   onnx.backend.test.case.noder   r   r!   r!   r!   r"   <module>   s   