U
    rh                     @  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                   @  s0   e Zd ZeddddZeddddZdS )DepthToSpaceNone)returnc                  C  s^  t jjddgdgddd} tddd	gd
ddggdddgdddggdddgdddggdddgdddggdd d!gd"d#d$ggd%d&d'gd(d)d*ggd+d,d-gd.d/d0ggd1d2d3gd4d5d6ggggtj}tddddd	dgdd+d d,d!d-gd
dddddgd"d.d#d/d$d0ggddddddgd%d1d&d2d'd3gddddddgd(d4d)d5d*d6ggggtj}t| |g|gd7d8 d S )9Nr   xy   Z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@Ztest_depthtospace_exampler   r   nameonnxhelperZ	make_nodenparrayZastypeZfloat32r   noder   r	    rI   L/tmp/pip-unpacked-wheel-xnis5xre/onnx/backend/test/case/node/depthtospace.pyexport_default_mode_example   sL        z(DepthToSpace.export_default_mode_examplec                  C  s^  t jjddgdgddd} tddd	gd
ddggdddgdddggdddgdddggdddgdddggdd d!gd"d#d$ggd%d&d'gd(d)d*ggd+d,d-gd.d/d0ggd1d2d3gd4d5d6ggggtj}tddddd	dgddddddgd
dddddgddddddggdd%d d&d!d'gd+d1d,d2d-d3gd"d(d#d)d$d*gd.d4d/d5d0d6ggggtj}t| |g|gd7d8 d S )9Nr   r   r	   r
   ZCRDr   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r    r!   r"   r#   r$   r%   r&   r'   r(   r)   r*   r+   r,   r-   r.   r/   r0   r1   r2   r3   r4   r5   r6   r7   r8   r9   r:   r;   r<   r=   r>   r?   Z"test_depthtospace_crd_mode_exampler@   rB   rG   rI   rI   rJ   export_crd_mode_example9   sL        z$DepthToSpace.export_crd_mode_exampleN)__name__
__module____qualname__staticmethodrK   rL   rI   rI   rI   rJ   r      s   *r   )

__future__r   ZnumpyrE   rC   Zonnx.backend.test.case.baser   Zonnx.backend.test.case.noder   r   rI   rI   rI   rJ   <module>   s
   