
    Ng;                     P    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 )    N)Base)expectc                 j    t           j                            d|                               |          S )N   )p)nprandombinomialastype)xdtypes     a/var/www/html/ai-engine/env/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returnNc                     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   )r   N)__name__
__module____qualname__staticmethodr%   r+   r/    r   r   r   r      s        	E 	E 	E \	E 
L 
L 
L \
L J J J \J J Jr   r   )	numpyr   r   onnx.backend.test.case.baser   onnx.backend.test.case.noder   r   r   r4   r   r   <module>r8      s   
      , , , , , , . . . . . .4 4 4&J &J &J &J &J &J &J &J &J &Jr   