
    Ngp                     J    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 )    N)Base)expectc                   >    e Zd Zedd            Zedd            ZdS )SpaceToDepthreturnNc            	         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
             d/var/www/html/ai-engine/env/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-   r.   )r5   r6   r7   )r/   r1   r3   )r8   r:   r<   )r0   r2   r4   )r9   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+   )r   N)__name__
__module____qualname__staticmethodr*   r@    r+   r)   r   r      s\        H H H \H" !P !P !P \!P !P !Pr+   r   )numpyr   r   onnx.backend.test.case.baser   onnx.backend.test.case.noder   r   rE   r+   r)   <module>rI      s~   
      , , , , , , . . . . . .5P 5P 5P 5P 5P4 5P 5P 5P 5P 5Pr+   