o
    wiO                     @  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                   @  s(   e Zd ZedddZedddZdS )	TilereturnNonec                  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size	test_tiler   r   name)onnxhelper	make_nodenprandomrandastypefloat32randintndimint64tiler   noder   repeatsr
    r)   ]/home/ubuntu/sommelier/.venv/lib/python3.10/site-packages/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g dg dg dg dgtjd
}t| ||g|gdd d S )Nr   r   r	   r
   r   r   r   r   r   )dtype)r   r   r   r   )r   r   r   r   test_tile_precomputedr   )r   r   r   r   arrayr!   r$   r   r&   r)   r)   r*   export_tile_precomputed   s   zTile.export_tile_precomputedN)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   r   r)   r)   r)   r*   <module>   s   