U
    rhO                     @  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 )TileNone)returnc                  C  sz   t jjdddgdgd} tjdddd	tj}tjjd
dt	|fdtj
}t||}t| ||g|gdd d S )Nr   xyzinputsoutputs               
   )lowhighsizeZ	test_tiler   r   name)onnxhelper	make_nodenprandomZrandZastypefloat32randintndimint64Ztiler   noder   Zrepeatsr
    r$   D/tmp/pip-unpacked-wheel-xnis5xre/onnx/backend/test/case/node/tile.pyexport_tile   s
    "zTile.export_tilec                  C  s   t jjdddgdgd} tjddgdd	ggtjd
}tjddgtjd
}tjddddgdd	dd	gddddgdd	dd	ggtjd
}t| ||g|gdd d S )Nr   r   r	   r
   r   r   r   r   r   )ZdtypeZtest_tile_precomputedr   )r   r   r   r   arrayr   r!   r   r"   r$   r$   r%   export_tile_precomputed   s    * zTile.export_tile_precomputedN)__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
   