U
    qhl                     @  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 )Wherec                 C  sd   |j |j krL|j tfkrL|j jtjk	rL|j jtjk	rLtd|j  d|j  t||||j fS )Nz$x and y should share the same dtype z != )ZdtypeobjecttypenpZstr_RuntimeErrorwhereZastype)self	conditionxy r   ?/tmp/pip-unpacked-wheel-xnis5xre/onnx/reference/ops/op_where.py_run   s    

z
Where._runN)__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   