
    /j2                    V    d dl mZ 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 )    )annotationsN)Base)expectc                  >    e Zd Zedd            Zedd            ZdS )WherereturnNonec                    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        f/home/longshao/multi-rider-rag/.venv/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,   N)r   r	   )__name__
__module____qualname__staticmethodr+   r0    r,   r*   r   r      sV        W W W \W 
 
 
 \
 
 
r,   r   )

__future__r   numpyr#   r    onnx.backend.test.case.baser   onnx.backend.test.case.noder   r   r5   r,   r*   <module>r:      s    # " " " " "      , , , , , , . . . . . .
 
 
 
 
D 
 
 
 
 
r,   