
    Ng                     J    d dl Zd dlZd dlmZ d dlmZ  G d de          ZdS )    N)Base)expectc                   >    e Zd Zedd            Zedd            ZdS )ReverseSequencereturnNc                  t   t           j                            dddgdgdd          } t          j        g dg d	g d
g dgt          j                  }t          j        g dt          j                  }t          j        g dg dg dg dgt          j                  }t          | ||g|g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   s       g/var/www/html/ai-engine/env/lib/python3.11/site-packages/onnx/backend/test/case/node/reversesequence.pyexport_reversesequence_timez+ReverseSequence.export_reversesequence_time   s   {$$)E % 
 
 H%%%%%%&&&&&&	 *
 
 
 RX>>>H%%%%%%&&&&&&	 *
 
 
 	}%C,		
 	
 	
 	
 	
 	
    c                  t   t           j                            dddgdgdd          } t          j        g dg d	g d
g dgt          j                  }t          j        g dt          j                  }t          j        g dg dg dg dgt          j                  }t          | ||g|g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*   r2   s       r4   export_reversesequence_batchz,ReverseSequence.export_reversesequence_batch2   s   {$$)E % 
 
 H$$$$$$&&&(((	 *
 
 
 RX>>>H$$$$$$&&&(((	 *
 
 
 	}%C-		
 	
 	
 	
 	
 	
r6   )r   N)__name__
__module____qualname__staticmethodr5   r9    r6   r4   r   r      sS        "
 "
 "
 \"
H "
 "
 "
 \"
 "
 "
r6   r   )numpyr.   r+   onnx.backend.test.case.baser   onnx.backend.test.case.noder   r   r>   r6   r4   <module>rB      s~   
      , , , , , , . . . . . .I
 I
 I
 I
 I
d I
 I
 I
 I
 I
r6   