§
    1�Å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                  ón   — e Zd Zedd„¦   «         Zed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
             úYC:\Projects\Historical\venv\Lib\site-packages\onnx/backend/test/case/node/spacetodepth.pyÚexportzSpaceToDepth.export   sþ   € à)Ð)‰
ˆˆ1ˆa��UØˆ	ÝŒ{×$Ò$ØØ�5Ø�EØð	 %ñ 
ô 
ˆõ ŒI×#Ò# EÑ*Ô*×1Ò1µ"´*Ñ=Ô=ˆÝŒjØ��1�a˜9‘n i°°i±ÀÐKñ
ô 
ˆõ Œl˜3Ð 2Ð 2Ð 2Ñ3Ô3ˆÝŒJ�s˜Q  Y°¡\Ñ 2°A¸±NÀAÈÁNÐSÑTÔTˆÝˆt˜Q˜C¨!¨Ð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ð �Y�Y 	 	 	Ð*Ø�Y�Y   Ð,Ø!�\�\ < < <Ð0Ø!�\�\ < < <Ð0ð	ðñ	
ô 	
÷ Š&•”Ñ
Ô
ð 	
õ 	ˆt˜Q˜C¨!¨Ð3NÐOÑOÔOÐOÐOÐOr-   c                 óè  — t           j                             ddgdgdd¬¦  «        } t          j        g d¢g d¢g d	¢g d
¢g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 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   ÚDCR©r   r   r   Úmode)ç        ç      2@ç      ð?ç      3@ç       @ç      4@)ç      B@ç      K@ç     €B@ç     €K@ç      C@ç      L@)ç      @ç      5@ç      @ç      6@ç      @ç      7@)ç     €C@ç     €L@ç      D@ç      M@ç     €D@ç     €M@)ç      "@ç      ;@ç      $@ç      <@ç      &@ç      =@)ç     €F@ç     €O@ç      G@ç      P@ç     €G@ç     @P@)ç      (@ç      >@ç      *@ç      ?@ç      ,@ç      @@)ç      H@ç     €P@ç     €H@ç     ÀP@ç      I@ç      Q@©rI   rK   rM   ©rU   rW   rY   ©ra   rc   re   ©rm   ro   rq   ©rJ   rL   rN   ©rV   rX   rZ   ©rb   rd   rf   ©rn   rp   rr   ©rO   rQ   rS   ©r[   r]   r_   ©rg   ri   rk   ©rs   ru   rw   ©rP   rR   rT   ©r\   r^   r`   ©rh   rj   rl   ©rt   rv   rx   Ú"test_spacetodepth_dcr_mode_exampler   rA   rC   s      r+   Úexport_dcr_mode_examplez$SpaceToDepth.export_dcr_mode_exampleD   s²  € åŒ{×$Ò$ØØ�5Ø�EØØð %ñ 
ô 
ˆõ ŒHð :Ð9Ð9Ø<Ð<Ð<Ø9Ð9Ð9Ø<Ð<Ð<ð	ð <Ð;Ð;Ø<Ð<Ð<Ø<Ð<Ð<Ø<Ð<Ð<ð	ððñ
ô 
÷" Š&•”Ñ
Ô
ð# 	
õ( ŒHð %�_�_ o o oÐ6Ø&Ð&Ð&Ð(:Ð(:Ð(:Ð;Ø'Ð'Ð'Ð);Ð);Ð);Ð<Ø'Ð'Ð'Ð);Ð);Ð);Ð<Ø'Ð'Ð'Ð);Ð);Ð);Ð<Ø'Ð'Ð'Ð);Ð);Ð);Ð<Ø'Ð'Ð'Ð);Ð);Ð);Ð<Ø'Ð'Ð'Ð);Ð);Ð);Ð<ð	ðñ
ô 
÷ Š&•”Ñ
Ô
ð 	
õ 	ˆt˜Q˜C¨!¨Ð3WÐXÑXÔXÐXÐXÐXr-   c                 óè  — t           j                             ddgdgdd¬¦  «        } t          j        g d¢g d¢g d	¢g d
¢g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 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   ÚCRDrG   )rI   ra   rK   rc   rM   re   )rJ   rb   rL   rd   rN   rf   )rU   rm   rW   ro   rY   rq   )rV   rn   rX   rp   rZ   rr   )rO   rg   rQ   ri   rS   rk   )rP   rh   rR   rj   rT   rl   )r[   rs   r]   ru   r_   rw   )r\   rt   r^   rv   r`   rx   ry   rz   r{   r|   r}   r~   r   r€   r�   r‚   rƒ   r„   r…   r†   r‡   rˆ   Ú"test_spacetodepth_crd_mode_exampler   rA   rC   s      r+   Úexport_crd_mode_examplez$SpaceToDepth.export_crd_mode_examples   s²  € åŒ{×$Ò$ØØ�5Ø�EØØð %ñ 
ô 
ˆõ ŒHð 9Ð8Ð8Ø<Ð<Ð<Ø9Ð9Ð9Ø<Ð<Ð<ð	ð =Ð<Ð<Ø<Ð<Ð<Ø<Ð<Ð<Ø<Ð<Ð<ð	ððñ
ô 
÷" Š&•”Ñ
Ô
ð# 	
õ( ŒHð %�_�_ o o oÐ6Ø&Ð&Ð&Ð(:Ð(:Ð(:Ð;Ø'Ð'Ð'Ð);Ð);Ð);Ð<Ø'Ð'Ð'Ð);Ð);Ð);Ð<Ø'Ð'Ð'Ð);Ð);Ð);Ð<Ø'Ð'Ð'Ð);Ð);Ð);Ð<Ø'Ð'Ð'Ð);Ð);Ð);Ð<Ø'Ð'Ð'Ð);Ð);Ð);Ð<ð	ðñ
ô 
÷ Š&•”Ñ
Ô
ð 	
õ 	ˆt˜Q˜C¨!¨Ð3WÐXÑXÔXÐXÐXÐXr-   N)r   r	   )Ú__name__Ú
__module__Ú__qualname__Ústaticmethodr,   rD   rŠ   rŽ   © r-   r+   r   r      s¦   € € € € € ØðHð Hð Hñ „\ðHð" ð!Pð !Pð !Pñ „\ð!PðF ð,Yð ,Yð ,Yñ „\ð,Yð\ ð,Yð ,Yð ,Yñ „\ð,Yð ,Yð ,Yr-   r   )
Ú
__future__r   Únumpyr   r   Úonnx.backend.test.case.baser   Úonnx.backend.test.case.noder   r   r“   r-   r+   ú<module>r˜      sš   ðð #Ð "Ð "Ð "Ð "Ð "à Ð Ð Ð à €€€Ø ,Ð ,Ð ,Ð ,Ð ,Ð ,Ø .Ð .Ð .Ð .Ð .Ð .ðSYð SYð SYð SYð SY�4ñ SYô SYð SYð SYð SYr-   