o
    toi                     @  s@   d dl mZ d dl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)OpRunc                   @  s$   e Zd Zedd Zedd ZdS )_CommonWindowc                 C  s4   t |}|dkr| }n| d }tj| |d}||fS )N   )dtype)r   nparange)sizeperiodicoutput_datatyper   N_1ni r   X/home/ubuntu/.local/lib/python3.10/site-packages/onnx/reference/ops/_op_common_window.py_begin   s   z_CommonWindow._beginc                 C  s   t |}||fS )N)r   astype)r
   resr   r   r   r   r   _end   s   z_CommonWindow._endN)__name__
__module____qualname__staticmethodr   r   r   r   r   r   r      s
    
	r   )	
__future__r   numpyr   onnx.helperr   onnx.reference.op_runr   r   r   r   r   r   <module>   s
   