
    /j                    b    d dl mZ d dlZd dlZd dlmZ d dlmZ d dl	m
Z
  G d de          ZdS )    )annotationsN)Base)expect)all_numeric_dtypesc                  >    e Zd Zedd            Zedd            ZdS )MaxreturnNonec                 (   t          j        g d                              t           j                  } t          j        g d                              t           j                  }t          j        g d                              t           j                  }t          j        g d                              t           j                  }t          j                            dg ddg          }t          || ||g|gd	
           t          j                            ddgdg          }t          || g| gd
           t          j        | |          }t          j                            dddgdg          }t          || |g|gd
           d S )N         r      r   )r      r   )r   r   r   r   )data_0data_1data_2resultinputsoutputstest_max_exampler   r   namer   test_max_one_inputr   test_max_two_inputs)	nparrayastypefloat32onnxhelper	make_noder   maximum)r   r   r   r   nodes        d/home/longshao/multi-rider-rag/.venv/lib/python3.11/site-packages/onnx/backend/test/case/node/max.pyexportz
Max.export   s   )))$$++BJ77)))$$++BJ77)))$$++BJ77)))$$++BJ77{$$111J % 
 

 	FF+H#		
 	
 	
 	
 {$$:J % 
 

 	tVHvh=QRRRRFF++{$$h'J % 
 

 	&&)F8BW	
 	
 	
 	
 	
 	
    c            
        t           D ]} t          j        g d                              |           }t          j        g d                              |           }t          j        g d                              |           }t          j                            dddgdg          }t          |||g|gd	t          j        |           j	         
           d S )Nr   r   )r   r   r   r   r   r   r   r   	test_max_r   )
r   r   r    r!   r#   r$   r%   r   dtyper   )op_dtyper   r   r   r'   s        r(   export_max_all_numeric_typesz Max.export_max_all_numeric_types2   s    * 	 	HXiii((//99FXiii((//99FXiii((//99F;(( (+!
 )  D
 ':(!3!3!8::	    	 	r*   N)r	   r
   )__name__
__module____qualname__staticmethodr)   r/    r*   r(   r   r      sS         
  
  
 \ 
D    \  r*   r   )
__future__r   numpyr   r#   onnx.backend.test.case.baser   onnx.backend.test.case.noder   onnx.backend.test.case.utilsr   r   r4   r*   r(   <module>r:      s    # " " " " "      , , , , , , . . . . . . ; ; ; ; ; ;4 4 4 4 4$ 4 4 4 4 4r*   