U
    qh                     @  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   npZarange)sizeZperiodicoutput_datatyper   ZN_1ni r   H/tmp/pip-unpacked-wheel-xnis5xre/onnx/reference/ops/_op_common_window.py_begin   s    z_CommonWindow._beginc                 C  s   t |}||fS )N)r   Z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   Znumpyr   Zonnx.helperr   Zonnx.reference.op_runr   r   r   r   r   r   <module>   s   