
    SiN                    T    S SK Jr  S SKrS SKrS SKJr  S SKJr  S r	 " S S\5      r
g)    )annotationsN)Base)expectc                \    [         R                  R                  SU S9R                  U5      $ )N   )p)nprandombinomialastype)xdtypes     d/var/www/html/ai-image-ml/venv/lib/python3.13/site-packages/onnx/backend/test/case/node/bernoulli.py"bernoulli_reference_implementationr      s)    
 99a1%,,U33    c                  P    \ rS rSr\SS j5       r\SS j5       r\SS j5       rSrg)	Bernoulli   c                     [         R                  R                  SS/S/S9n [        R                  R                  SSS5      R                  [        5      n[        U[        5      n[        X/U/SS	9  g )
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_dtype(Bernoulli.export_bernoulli_without_dtype   sj    {{$$5E % 
 IIc3+2259.q%8tC!3CDr   c                 .   [         R                  R                  SS/S/[         R                  R                  S9n [
        R                  R                  SSS5      R                  [
        R                  5      n[        U[        5      n[        X/U/SS	9  g )
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_dtype%Bernoulli.export_bernoulli_with_dtype"   s}    {{$$5E""))	 % 
 IIc3+222::>.q%8tC!3JKr   c                 (   [        S5      n [        R                  R                  SS/S/U S9n[        R
                  R                  SSS5      R                  [        R                  5      n[        U[        R                  5      n[        X/U/S	S
9  g )Nr   r   r   r   )r   r   seedr   r   r   test_bernoulli_seedr   )r#   r   r    r!   r	   r
   r"   r   r,   r   r   )r0   r%   r   r   s       r   export_bernoulli_with_seed$Bernoulli.export_bernoulli_with_seed/   s~    Qx{{$$5E	 % 
 IIc3+222::>.q"**=tC!3HIr    N)returnNone)	__name__
__module____qualname____firstlineno__staticmethodr&   r-   r2   __static_attributes__r4   r   r   r   r      sB    	E 	E 
L 
L J Jr   r   )
__future__r   numpyr	   r   onnx.backend.test.case.baser   onnx.backend.test.case.noder   r   r   r4   r   r   <module>rA      s)    #   , .4&J &Jr   