o
    wiy                     @  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  sH   t | tj|| | jdvrtd| jd| jdkr"tj| _d S d S )N)LEFTRIGHTz Unexpected value for direction (z).r   )r   __init__npright_shift	direction
ValueError
left_shift	numpy_fct)self	onnx_node
run_params r   [/home/ubuntu/sommelier/.venv/lib/python3.10/site-packages/onnx/reference/ops/op_bitshift.pyr      s   

zBitShift.__init__N)__name__
__module____qualname__r   r   r   r   r   r      s    r   )
__future__r   numpyr   onnx.reference.ops._opr   r   r   r   r   r   <module>   s   