U
    qhO                     @  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Zdtype)nparrayshapeint64)selfdata r   ?/tmp/pip-unpacked-wheel-xnis5xre/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ddddZd
dd	ZdS )Shape_15intz
int | Noneztuple[int, int] | None)nstartendreturnc                 C  sp   |dkr:|d kst |rd S |dk r2d| | fS d|fS |d ksLt |rT|| fS |dk rh|| | 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 kr2t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)NN)r   r   r   staticmethodr   r   r   r   r   r   r      s   r   )
__future__r   Znumpyr   Zonnx.reference.op_runr   r   r   r   r   r   r   <module>   s   