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 )ReverseSequenceNone)returnc                  C  s   t jjdddgdgddd} tjdd	d
dgddddgddddgddddggtjd}tjddddgtjd}tjddddgddd
dgdd	ddgddddggtjd}t| ||g|gdd d S )Nr   xsequence_lensyr      inputsoutputsZ	time_axisZ
batch_axis              @       @      (@      ?      @      "@      *@       @      @      $@      ,@      @      @      &@      .@Zdtype         Ztest_reversesequence_timer   r   nameonnxhelperZ	make_nodenparrayZfloat32Zint64r   noder   r	   r
    r,   O/tmp/pip-unpacked-wheel-xnis5xre/onnx/backend/test/case/node/reversesequence.pyexport_reversesequence_time   s<    



	




z+ReverseSequence.export_reversesequence_timec                  C  s   t jjdddgdgddd} tjdd	d
dgddddgddddgddddggtjd}tjddddgtjd}tjdd	d
dgddddgddddgddddggtjd}t| ||g|gdd d S )Nr   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    Ztest_reversesequence_batchr#   r%   r*   r,   r,   r-   export_reversesequence_batch3   s<    



	




z,ReverseSequence.export_reversesequence_batchN)__name__
__module____qualname__staticmethodr.   r/   r,   r,   r,   r-   r      s   $r   )

__future__r   Znumpyr(   r&   Zonnx.backend.test.case.baser   Zonnx.backend.test.case.noder   r   r,   r,   r,   r-   <module>   s
   