o
    toi@                     @  s8   d dl mZ d dlmZ d dlmZ G dd deZdS )    )annotations)np_dtype_to_tensor_dtype)_CommonRandomc                   @  s   e Zd ZdddZdS )	BernoulliNc                 C  sL   |d u r	t |j}| j||dd}| |}|jd|d|}||fS )NT)dtypedtype_first   )p)r   r   _dtype
_get_statebinomialastype)selfxr   seedstateres r   S/home/ubuntu/.local/lib/python3.10/site-packages/onnx/reference/ops/op_bernoulli.py_run   s   

zBernoulli._run)NN)__name__
__module____qualname__r   r   r   r   r   r   
   s    r   N)
__future__r   onnx.helperr   $onnx.reference.ops._op_common_randomr   r   r   r   r   r   <module>   s   