o
    wÖir  ã                   @  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„ ZdS )ÚNonZeroc                 C  s   t  t  |¡¡ t j¡}|fS )N)ÚnpÚvstackÚnonzeroÚastypeÚint64)ÚselfÚxÚres© r   ú[/home/ubuntu/sommelier/.venv/lib/python3.10/site-packages/onnx/reference/ops/op_non_zero.pyÚ_run   s   zNonZero._runN)Ú__name__Ú
__module__Ú__qualname__r   r   r   r   r   r      s    r   )Ú
__future__r   Únumpyr   Úonnx.reference.op_runr   r   r   r   r   r   Ú<module>   s   