
    /j                    V    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	dS )    )annotationsN)Base)expectc                  >    e Zd Zedd            Zedd            ZdS )ReverseSequencereturnNonec                 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       p/home/longshao/multi-rider-rag/.venv/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,   r4   s       r6   export_reversesequence_batchz,ReverseSequence.export_reversesequence_batch3   s   {$$)E % 
 
 H$$$$$$&&&(((	 *
 
 
 RX>>>H$$$$$$&&&(((	 *
 
 
 	}%C-		
 	
 	
 	
 	
 	
r8   N)r   r	   )__name__
__module____qualname__staticmethodr7   r;    r8   r6   r   r      sS        "
 "
 "
 \"
H "
 "
 "
 \"
 "
 "
r8   r   )

__future__r   numpyr0   r-   onnx.backend.test.case.baser   onnx.backend.test.case.noder   r   r@   r8   r6   <module>rE      s    # " " " " "      , , , , , , . . . . . .I
 I
 I
 I
 I
d I
 I
 I
 I
 I
r8   