o
    soi                     @  sT   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 deZdS )    )annotationsN)Base)expect)all_numeric_dtypesc                   @  s(   e Zd ZedddZedddZdS )	MaxreturnNonec                  C  s   t g dt j} t g dt j}t g dt j}t g dt j}tjjdg ddgd}t|| ||g|gd	d
 tjjddgdgd}t|| g| gdd
 t | |}tjjdddgdgd}t|| |g|gd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   node r%   S/home/ubuntu/.local/lib/python3.10/site-packages/onnx/backend/test/case/node/max.pyexport   s<   
z
Max.exportc               	   C  s   t D ]>} tg d| }tg d| }tg d| }tjjdddgdgd}t|||g|gd	t| j	 d
 q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$   r%   r%   r&   export_max_all_numeric_types2   s    z Max.export_max_all_numeric_typesN)r   r   )__name__
__module____qualname__staticmethodr'   r+   r%   r%   r%   r&   r      s
    "r   )
__future__r   numpyr   r    onnx.backend.test.case.baser   onnx.backend.test.case.noder   onnx.backend.test.case.utilsr   r   r%   r%   r%   r&   <module>   s   