
    /j                    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 )SpaceToDepthreturnNonec            	        dx\  } }}}}d}t           j                            ddgdg|          }t          j                            |                              t          j                  }t          j        || |||z  |||z  |g          }t          j	        |g d          }t          j        || ||dz  z  ||z  ||z  g          }	t          ||g|	gd	           d S )
N)   r      r   r   r   xyinputsoutputs	blocksize)r            r      test_spacetodepthr   r   name)onnxhelper	make_nodenprandomrandom_sampleastypefloat32reshape	transposer   )
bchwshaper   noder   tmpr   s
             m/home/longshao/multi-rider-rag/.venv/lib/python3.11/site-packages/onnx/backend/test/case/node/spacetodepth.pyexportzSpaceToDepth.export   s    ))
1aU	{$$5E	 % 
 
 I##E**11"*==j1a9niiK
 
 l3 2 2 233JsQY\ 2ANANSTTtQC!3FGGGGGG    c                    t           j                            ddgdgd          } t          j        g dg dg dg d	ggg                              t          j                  }t          j        g d
g dgg dg dgg dg dgg dg dggg                              t          j                  }t          | |g|gd           d S )Nr   r   r   r   r   )r   r   r      r      )                  )r   	   r   
   r      )                  )r   r   r   )r   r   r   )r   r/   r0   )r7   r8   r9   )r1   r3   r5   )r:   r<   r>   )r2   r4   r6   )r;   r=   r?   test_spacetodepth_exampler   )r   r   r   r   arrayr    r!   r   )r)   r   r   s      r+   export_examplezSpaceToDepth.export_example    s!   {$$5E	 % 
 
 H +**000,,,000		
 
 &

 	
 H YY			*YY,!\\<<<0!\\<<<0		
 	
 &

 	
 	tQC!3NOOOOOOr-   N)r   r	   )__name__
__module____qualname__staticmethodr,   rB    r-   r+   r   r      s\        H H H \H" !P !P !P \!P !P !Pr-   r   )

__future__r   numpyr   r   onnx.backend.test.case.baser   onnx.backend.test.case.noder   r   rG   r-   r+   <module>rL      s    # " " " " "      , , , , , , . . . . . .5P 5P 5P 5P 5P4 5P 5P 5P 5P 5Pr-   