U
    qh"                     @  s4   d dl mZ d dlZd dlmZ G dd deZdS )    )annotationsN)OpRunc                   @  s   e Zd ZdddZdS )DepthToSpaceNc              	   C  s   t |jdkr td|jd|j\}}}}|dkrr||||||  ||f}||}	t|	dddddd	g}
n:||||  ||||f}||}	t|	dddd	ddg}
||||  || || f}t|
|}|fS )
N   zUnexpected shape .ZDCRr               )lenshapeRuntimeErrorZreshapenpZ	transpose)selfdata	blocksizemodebchwZtmpshapeZreshapedZ
transposedZ
finalshapey r   H/tmp/pip-unpacked-wheel-xnis5xre/onnx/reference/ops/op_depth_to_space.py_run   s:    




zDepthToSpace._run)NN)__name__
__module____qualname__r   r   r   r   r   r      s   r   )
__future__r   Znumpyr   Zonnx.reference.op_runr   r   r   r   r   r   <module>   s   