
    bIg                    N    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e      Z	y)    )annotationsN)Base)expectc                  0    e Zd Zedd       Zedd       Zy)ReverseSequencec                    t         j                  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       y )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       h/var/www/html/answerous/venv/lib/python3.12/site-packages/onnx/backend/test/case/node/reversesequence.pyexport_reversesequence_timez+ReverseSequence.export_reversesequence_time   s    {{$$)E % 
 HH%%&&	 **
 RXX>HH%%&&	 **
 	}%C,		
    c                    t         j                  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       y )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_batch3   s    {{$$)E % 
 HH$$&(	 **
 RXX>HH$$&(	 **
 	}%C-		
r6   N)returnNone)__name__
__module____qualname__staticmethodr5   r9    r6   r4   r   r      s)    "
 "
H "
 "
r6   r   )

__future__r   numpyr.   r+   onnx.backend.test.case.baser   onnx.backend.test.case.noder   r   r@   r6   r4   <module>rE      s$    #   , .I
d I
r6   