o
    soi                     @  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 )	ReverseSequencereturnNonec                  C  s   t jjdddgdgddd} tjg dg d	g d
g dgtjd}tjg dtjd}tjg dg dg dg dgtjd}t| ||g|gdd d S )Nr   xsequence_lensyr      inputsoutputs	time_axis
batch_axis)              @       @      (@)      ?      @      "@      *@)       @      @      $@      ,@)      @      @      &@      .@dtype)         r   )r   r   r   r   )r   r   r   r   )r   r   r   r   )r   r   r   r    test_reversesequence_timer   r   nameonnxhelper	make_nodenparrayfloat32int64r   noder   r	   r
    r3   _/home/ubuntu/.local/lib/python3.10/site-packages/onnx/backend/test/case/node/reversesequence.pyexport_reversesequence_time   <   	

z+ReverseSequence.export_reversesequence_timec                  C  s   t jjdddgdgddd} tjg dg d	g d
g dgtjd}tjg dtjd}tjg dg dg dg d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$   r#   )r   r   r   r   )r   r   r   r   )r    r   r   r   test_reversesequence_batchr'   r)   r1   r3   r3   r4   export_reversesequence_batch3   r6   z,ReverseSequence.export_reversesequence_batchN)r   r   )__name__
__module____qualname__staticmethodr5   r8   r3   r3   r3   r4   r      s
    $r   )

__future__r   numpyr-   r*   onnx.backend.test.case.baser   onnx.backend.test.case.noder   r   r3   r3   r3   r4   <module>   s   