o
    wiO                     @  sD   d dl mZ d dlZd dlmZ G dd deZG dd deZdS )    )annotationsN)OpRunc                   @  s   e Zd Zdd ZdS )Shape_1c                 C  s   t j|jt jdfS )Ndtype)nparrayshapeint64)selfdata r   X/home/ubuntu/sommelier/.venv/lib/python3.10/site-packages/onnx/reference/ops/op_shape.py_run   s   zShape_1._runN)__name__
__module____qualname__r   r   r   r   r   r      s    r   c                   @  s$   e Zd Zeddd	ZdddZd
S )Shape_15nintstart
int | Noneendreturntuple[int, int] | Nonec                 C  sp   |dkr|d u st |rd S |dk rd| | fS d|fS |d u s&t |r*|| fS |dk r4|| | fS ||fS )Nr   )r   isnan)r   r   r   r   r   r   	_interval   s   zShape_15._intervalNc                 C  sV   | j t|j||d}|d u rtj|jtjdfS tj|j|d |d  tjdfS )N)r   r   r   r      )r   lenr	   r   r   r
   )r   r   r   r   abr   r   r   r      s   $zShape_15._run)r   r   r   r   r   r   r   r   )NN)r   r   r   staticmethodr   r   r   r   r   r   r      s    r   )
__future__r   numpyr   onnx.reference.op_runr   r   r   r   r   r   r   <module>   s
   