U
    qhp                     @  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 )UpsampleNc                 C  st   |dkrZ| tj | krZ|}t|D ]&\}}|dkr>q,tj|t||d}q,|fS td|d|dd S )NZnearest   )axiszNot implemented for mode=z and scale=.)ZastypenpZint64tolist	enumeraterepeatintRuntimeError)selfxZscalemoderr   s r   B/tmp/pip-unpacked-wheel-xnis5xre/onnx/reference/ops/op_upsample.py_run   s     zUpsample._run)N)__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   