U
    qhy                     @  s4   d dl mZ d dlZd dlmZ G dd deZdS )    )annotationsN)OpRunBinaryNumpyc                   @  s   e Zd Zdd ZdS )BitShiftc                 C  sD   t | tj|| | jdkr.td| jd| jdkr@tj| _d S )N)LEFTRIGHTz Unexpected value for direction (z).r   )r   __init__npZright_shift	direction
ValueErrorZ
left_shiftZ	numpy_fct)selfZ	onnx_nodeZ
run_params r   B/tmp/pip-unpacked-wheel-xnis5xre/onnx/reference/ops/op_bitshift.pyr      s
    

zBitShift.__init__N)__name__
__module____qualname__r   r   r   r   r   r      s   r   )
__future__r   Znumpyr   Zonnx.reference.ops._opr   r   r   r   r   r   <module>   s   