
    /jO                    V    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e          Z	dS )    )annotationsN)Base)expectc                  >    e Zd Zedd            Zedd            ZdS )TilereturnNonec                    t           j                            dddgdg          } t          j                            dddd	                              t          j                  }t          j                            d
dt          j	        |          f                              t          j
                  }t          j        ||          }t          | ||g|g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   s       e/home/longshao/multi-rider-rag/.venv/lib/python3.11/site-packages/onnx/backend/test/case/node/tile.pyexport_tilezTile.export_tile   s    {$$VS#J$NNINN1aA&&--bj99)##"'!**#GGNNrxXXGAwtQL1#KHHHHHH    c                 d   t           j                            dddgdg          } t          j        ddgdd	ggt          j        
          }t          j        ddgt          j        
          }t          j        g dg dg dg dgt          j        
          }t          | ||g|g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)   s       r,   export_tile_precomputedzTile.export_tile_precomputed   s    {$$VS#J$NNHq!fq!f%RZ888(Aq6222H\\<<<|||DBJ
 
 
 	tQL1#<STTTTTTr.   N)r   r	   )__name__
__module____qualname__staticmethodr-   r3    r.   r,   r   r      s\        	I 	I 	I \	I U U U \U U Ur.   r   )

__future__r   numpyr    r   onnx.backend.test.case.baser   onnx.backend.test.case.noder   r   r8   r.   r,   <module>r=      s    # " " " " "      , , , , , , . . . . . .U U U U U4 U U U U Ur.   