
    /jN                    \    d dl mZ d dlZd dlZd dlmZ d dlmZ d Z	 G d de          Z
dS )    )annotationsN)Base)expectc                j    t           j                            d|                               |          S )N   )p)nprandombinomialastype)xdtypes     j/home/longshao/multi-rider-rag/.venv/lib/python3.11/site-packages/onnx/backend/test/case/node/bernoulli.py"bernoulli_reference_implementationr      s-    
 9a1%%,,U333    c                  V    e Zd Zedd            Zedd            Zedd            ZdS )	BernoullireturnNonec                    t           j                            ddgdg          } t          j                            ddd                              t                    }t          |t                    }t          | |g|gd	           d S )
Nr   r   y)inputsoutputs              ?
   test_bernoullir   r   name)
onnxhelper	make_noder	   r
   uniformr   floatr   r   noder   r   s      r   export_bernoulli_without_dtypez(Bernoulli.export_bernoulli_without_dtype   s    {$$5E % 
 
 Ic3++22599.q%88tQC!3CDDDDDDr   c                 >   t           j                            ddgdgt           j        j                  } t
          j                            ddd                              t
          j	                  }t          |t                    }t          | |g|gd	           d S )
Nr   r   r   )r   r   r   r   r   r   test_bernoulli_doubler   )r    r!   r"   TensorProtoDOUBLEr	   r
   r#   r   float32r   r$   r   r%   s      r   export_bernoulli_with_dtypez%Bernoulli.export_bernoulli_with_dtype"   s    {$$5E")	 % 
 
 Ic3++222:>>.q%88tQC!3JKKKKKKr   c                 H   t          d          } t          j                            ddgdg|           }t          j                            ddd                              t          j                  }t          |t          j                  }t          ||g|gd	
           d S )Nr   r   r   r   )r   r   seedr   r   r   test_bernoulli_seedr   )r$   r    r!   r"   r	   r
   r#   r   r,   r   r   )r/   r&   r   r   s       r   export_bernoulli_with_seedz$Bernoulli.export_bernoulli_with_seed/   s    Qxx{$$5E	 % 
 
 Ic3++222:>>.q"*==tQC!3HIIIIIIr   N)r   r   )__name__
__module____qualname__staticmethodr'   r-   r1    r   r   r   r      s        	E 	E 	E \	E 
L 
L 
L \
L J J J \J J Jr   r   )
__future__r   numpyr	   r    onnx.backend.test.case.baser   onnx.backend.test.case.noder   r   r   r6   r   r   <module>r;      s    # " " " " "      , , , , , , . . . . . .4 4 4&J &J &J &J &J &J &J &J &J &Jr   