U
    q±ËhŸ  ã                   @  s4   d dl mZ d dlZd dlmZ G dd„ deƒZdS )é    )ÚannotationsN)ÚOpRunc                   @  s   e Zd Zddd„ZdS )ÚTriluNc                 C  s<   |d krdnt |ƒ}|r&t ||¡fS t ||¡ |j¡fS )Nr   )ÚintÚnpZtriuZtrilZastypeZdtype)ÚselfÚxÚkÚupper© r   ú?/tmp/pip-unpacked-wheel-xnis5xre/onnx/reference/ops/op_trilu.pyÚ_run   s    z
Trilu._run)NN)Ú__name__Ú
__module__Ú__qualname__r   r   r   r   r   r      s   r   )Ú
__future__r   Znumpyr   Zonnx.reference.op_runr   r   r   r   r   r   Ú<module>   s   