
    /j                        d dl mZ d dlZd dlZd dlZd dlZ G d dej                  Ze	dk    r ej
                     dS dS )    )annotationsNc                  X    e Zd Zej                            ddg          d             ZdS )NodeInferenceTest)GreaterOrEqual)LessOrEqualc                   t           j                            |ddgdg          }t           j                            |j        dd          }t           j                            t           j        j        ddg          }t           j                            t           j        j        ddg          }t           j	        
                    ||||d          }|                     t          |                                          dg           |                     |d         j        j        t           j        j                   |                     d	 |d         j        j        j        D             ddg           d S )
Nxyz       
   )r	   r
   c                    g | ]	}|j         
S  )	dim_value).0dims     h/home/longshao/multi-rider-rag/.venv/lib/python3.11/site-packages/onnx/test/node_shape_inference_test.py
<listcomp>z8NodeInferenceTest.test_comparison_op.<locals>.<listcomp>    s    HHHsS]HHH    )onnxhelper	make_nodedefs
get_schemaop_typemake_tensor_type_protoTensorProtoINT32shape_inferenceinfer_node_outputsassertEquallistkeystensor_type	elem_typeBOOLshaper   )selfr   nodeschemaxtypeytyperesults          r   test_comparison_opz$NodeInferenceTest.test_comparison_op   s?    {$$WsCj3%@@%%dlB;;2243C3IAr7SS2243C3IBPQ7SS%88DE22
 
 	fkkmm,,se4440:D<L<QRRRHHfSk&=&C&GHHHH	
 	
 	
 	
 	
r   N)__name__
__module____qualname__parameterizedexpandr0   r   r   r   r   r      sN         ''	
 
 
 
 
 
r   r   __main__)
__future__r   unittestr4   onnx.helperr   onnx.shape_inferenceTestCaser   r1   mainr   r   r   <module>r=      s    # " " " " "             
 
 
 
 
) 
 
 
. zHMOOOOO r   