U
    qh                     @  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 kr |d krt j}q:|j}n|tjkr2t j}nt|}|j}t|dkr^|d |d f}n"t|dkrp|}nt	d|dt j
|||dfS )N   r      z)EyeLike only accept 1D or 2D tensors not .)r   r   )npZfloat32r   r   STRINGZstr_r   shapelenRuntimeErrorZeye)selfdatar   r   argsZ_dtyper   sh r   A/tmp/pip-unpacked-wheel-xnis5xre/onnx/reference/ops/op_eyelike.py_run   s    
zEyeLike._run)__name__
__module____qualname__r   r   r   r   r   r      s   r   )
__future__r   Znumpyr   Zonnx.helperr   Zonnx.onnx_pbr   Zonnx.reference.op_runr   r   r   r   r   r   <module>   s
   