o
    soiò  ã                   @  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 )	ÚDepthToSpaceÚreturnÚNonec                  C  óþ   t jjddgdgddd} t 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 d¢g d¢g d¢g d¢gg d¢g d¢g d¢g d¢ggg¡ tj¡}t| |g|gdd  d S )!Nr   ÚxÚyé   ÚDCR©ÚinputsÚoutputsÚ	blocksizeÚmode©ç        ç      ð?ç       @©ç      @ç      @ç      @©ç      "@ç      $@ç      &@©ç      (@ç      *@ç      ,@©ç      2@ç      3@ç      4@©ç      5@ç      6@ç      7@©ç      ;@ç      <@ç      =@©ç      >@ç      ?@ç      @@©ç      B@ç     B@ç      C@©ç     C@ç      D@ç     D@©ç     F@ç      G@ç     G@©ç      H@ç     H@ç      I@©ç      K@ç     K@ç      L@©ç     L@ç      M@ç     M@©ç     O@ç      P@ç     @P@©ç     P@ç     ÀP@ç      Q@)r   r#   r   r$   r   r%   )r3   rC   r4   rD   r5   rE   )r   r'   r   r(   r   r)   )r7   rG   r8   rH   r9   rI   )r   r+   r   r,   r   r-   )r;   rK   r<   rL   r=   rM   )r   r/   r    r0   r!   r1   )r?   rO   r@   rP   rA   rQ   Útest_depthtospace_example©r   r   Úname©ÚonnxÚhelperÚ	make_nodeÚnpÚarrayÚastypeÚfloat32r   ©Únoder	   r
   © r_   ú\/home/ubuntu/.local/lib/python3.10/site-packages/onnx/backend/test/case/node/depthtospace.pyÚexport_default_mode_example   óD   ÿøÿÿóüüùÿÿïz(DepthToSpace.export_default_mode_examplec                  C  r   )!Nr   r	   r
   r   ÚCRDr   r   r   r   r   r"   r&   r*   r.   r2   r6   r:   r>   rB   rF   rJ   rN   )r   r   r   r   r   r   )r#   r+   r$   r,   r%   r-   )r   r   r   r    r   r!   )r'   r/   r(   r0   r)   r1   )r3   r;   r4   r<   r5   r=   )rC   rK   rD   rL   rE   rM   )r7   r?   r8   r@   r9   rA   )rG   rO   rH   rP   rI   rQ   Ú"test_depthtospace_crd_mode_examplerS   rU   r]   r_   r_   r`   Úexport_crd_mode_example9   rb   z$DepthToSpace.export_crd_mode_exampleN)r   r   )Ú__name__Ú
__module__Ú__qualname__Ústaticmethodra   re   r_   r_   r_   r`   r      s
    *r   )
Ú
__future__r   ÚnumpyrY   rV   Úonnx.backend.test.case.baser   Úonnx.backend.test.case.noder   r   r_   r_   r_   r`   Ú<module>   s   