
    Ng                     J    d dl Zd dlZd dlmZ d dlmZ  G d de          ZdS )    N)Base)expectc                   >    e Zd Zedd            Zedd            ZdS )WherereturnNc                     t           j                            dg ddg          } t          j        ddgddggt
                    }t          j        ddgd	d
ggt          j                  }t          j        ddgddggt          j                  }t          j        |||          }t          | |||g|gd           d S )Nr   	conditionxyzinputsoutputs   r   dtype         	            test_where_exampler   r   name)	onnxhelper	make_nodenparrayboolfloat32wherer   noder
   r   r   r   s        ]/var/www/html/ai-engine/env/lib/python3.11/site-packages/onnx/backend/test/case/node/where.pyexportzWhere.export   s    {$$***E % 
 
 Hq!fq!f-T:::	Hq!fq!f%RZ888Hq!fq!f%RZ888HY1%%tY1-sAUVVVVVV    c                     t           j                            dg ddg          } t          j        ddgddggt
                    }t          j        ddgd	d
ggt          j                  }t          j        ddgddggt          j                  }t          j        |||          }t          | |||g|gd           d S )Nr   r	   r   r   r   r   r   r   r   r   r   r   r   r   test_where_long_exampler   )	r   r   r    r!   r"   r#   int64r%   r   r&   s        r(   export_longzWhere.export_long   s    {$$***E % 
 
 Hq!fq!f-T:::	Hq!fq!f%RX666Hq!fq!f%RX666HY1%%)Q*QC>W	
 	
 	
 	
 	
 	
r*   )r   N)__name__
__module____qualname__staticmethodr)   r.    r*   r(   r   r      sV        W W W \W 
 
 
 \
 
 
r*   r   )numpyr!   r   onnx.backend.test.case.baser   onnx.backend.test.case.noder   r   r3   r*   r(   <module>r7      st   
      , , , , , , . . . . . .
 
 
 
 
D 
 
 
 
 
r*   