U
    rh                     @  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                   @  s0   e Zd ZeddddZeddddZdS )MaxNone)returnc                  C  s   t dddgt j} t dddgt j}t dddgt j}t dddgt j}tjjdddd	g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   data_0data_1data_2resultinputsoutputsZtest_max_exampler   r   nameZtest_max_one_inputZtest_max_two_inputs)	nparrayastypeZfloat32onnxhelper	make_noder   maximum)r   r   r   r   node r   C/tmp/pip-unpacked-wheel-xnis5xre/onnx/backend/test/case/node/max.pyexport   sB       z
Max.exportc               	   C  s   t D ]} tdddg| }tdddg| }tdddg| }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   Z	test_max_r   )
r   r   r   r   r   r   r   r   Zdtyper   )Zop_dtyper   r   r   r   r   r   r    export_max_all_numeric_types2   s    z Max.export_max_all_numeric_typesN)__name__
__module____qualname__staticmethodr!   r"   r   r   r   r    r      s   "r   )
__future__r   Znumpyr   r   Zonnx.backend.test.case.baser   Zonnx.backend.test.case.noder   Zonnx.backend.test.case.utilsr   r   r   r   r   r    <module>   s   