
    h                     B    S SK rS SKrS SKJr  S SKJr   " S S\5      rg)    N)Base)expectc                   <    \ rS rSr\SS j5       r\SS j5       rSrg)ReverseSequence   Nc                  n   [         R                  R                  SSS/S/SSS9n [        R                  " / SQ/ S	Q/ S
Q/ SQ/[        R
                  S9n[        R                  " / SQ[        R                  S9n[        R                  " / SQ/ SQ/ SQ/ SQ/[        R
                  S9n[        U X/U/SS9  g )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       e/var/www/fran/franai/venv/lib/python3.13/site-packages/onnx/backend/test/case/node/reversesequence.pyexport_reversesequence_time+ReverseSequence.export_reversesequence_time   s    {{$$)E % 
 HH%%&&	 **
 RXX>HH%%&&	 **
 	%C,		
    c                  n   [         R                  R                  SSS/S/SSS9n [        R                  " / SQ/ S	Q/ S
Q/ SQ/[        R
                  S9n[        R                  " / SQ[        R                  S9n[        R                  " / SQ/ SQ/ SQ/ SQ/[        R
                  S9n[        U X/U/SS9  g )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_batch,ReverseSequence.export_reversesequence_batch2   s    {{$$)E % 
 HH$$&(	 **
 RXX>HH$$&(	 **
 	%C-		
r7    )returnN)__name__
__module____qualname____firstlineno__staticmethodr5   r:   __static_attributes__r<   r7   r4   r   r      s)    "
 "
H "
 "
r7   r   )numpyr.   r+   onnx.backend.test.case.baser   onnx.backend.test.case.noder   r   r<   r7   r4   <module>rG      s!   
   , .I
d I
r7   