U
    rh                     @  sH   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 deZ	dS )    )annotationsN)Base)expectc                   @  s0   e Zd ZeddddZeddddZdS )DetNone)returnc                  C  sT   t jjddgdgd} tdddtj}tj	|}t
| |g|gdd d S )	Nr   xyinputsoutputs      Ztest_det_2dr   r   name)onnxhelper	make_nodenpZarangeZreshapeastypefloat32linalgdetr   noder   r	    r   C/tmp/pip-unpacked-wheel-xnis5xre/onnx/backend/test/case/node/det.py	export_2d   s    zDet.export_2dc                  C  sv   t jjddgdgd} tddgddggddgddggddgddgggtj}tj|}t	| |g|gd	d
 d S )Nr   r   r	   r
      r      r   Ztest_det_ndr   )
r   r   r   r   arrayr   r   r   r   r   r   r   r   r   	export_nd   s    4zDet.export_ndN)__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   r   r   r   r   r   <module>   s
   