o
    toi                     @  sL   d dl mZ d dlZd dlmZ d dlmZ d dlm	Z	 G dd de	Z
dS )    )annotationsN)tensor_dtype_to_np_dtype)TensorProto)OpRunc                   @  s   e Zd ZdddddZdS )EyeLikeN)dtypekc                G  s   |d u r|d u rt j}n|j}n|tjkrt j}nt|}|j}t|dkr/|d |d f}nt|dkr8|}nt	d|dt j
|||dfS )N   r      z)EyeLike only accept 1D or 2D tensors not .)r   r   )npfloat32r   r   STRINGstr_r   shapelenRuntimeErroreye)selfdatar   r   args_dtyper   sh r   Q/home/ubuntu/.local/lib/python3.10/site-packages/onnx/reference/ops/op_eyelike.py_run   s   
zEyeLike._run)__name__
__module____qualname__r   r   r   r   r   r      s    r   )
__future__r   numpyr   onnx.helperr   onnx.onnx_pbr   onnx.reference.op_runr   r   r   r   r   r   <module>   s   