
    bIg
                    t    d dl mZ d dlmZ d dlZd dlZd dlmZ d dl	m
Z
 	 d		 	 	 	 	 	 	 d
dZ G d de      Zy)    )annotations)AnyN)Base)expectc                t    t        |       }||d   }|j                  ||       |S |j                  |       |S )Nr   )listinsertappend)sequencetensorpositionseqinsert_positions        g/var/www/html/answerous/venv/lib/python3.12/site-packages/onnx/backend/test/case/node/sequenceinsert.py(sequence_insert_reference_implementationr      sF     x.C #1+

?F+ J 	

6J    c                      e Zd Zedd       Zy)SequenceInsertc                    t        j                  g d      j                  t         j                        gt        j                  g d      t        j                  dg      j                  t         j                        gd} t        j                  g d      j                  t         j                        t        j                  g d      j                  t         j                        t        j                  ddg      j                  t         j                        g}| j	                         D ]  \  }}|d   j                  t         j                        }t        |      d	kD  rNt        j                  j                  d
g ddg      }|d	   }t        |||      }t        ||||g|gd|z          t        j                  j                  d
ddgdg      }t        ||      }t        |||g|gd|z           y )N)
         )r   r   )at_backat_front)            )            	   r   r   )r   r   r   output_sequence)inputsoutputstest_sequence_insert_)r'   r(   namer   r   )nparrayastypeint64itemslenonnxhelper	make_noder   r   )
test_casesr   	test_nametest_inputsr   noder   inserteds           r   exportzSequenceInsert.export    s    .55bhh?@+.!0D0DRXX0NO


 HH\"))"((3HHY&&rxx0HHaV##BHH-
 '1&6&6&8 	"I{ ^**2884F;!#{{,,$=./ - 
 'q>Cfh $fh7%J09<	 {{,,$&1./ - 
 DHfU$f-%J09<	5	r   N)returnNone)__name__
__module____qualname__staticmethodr9    r   r   r   r      s    * *r   r   )N)r   	list[Any]r   
np.ndarrayr   rB   r:   rA   )
__future__r   typingr   numpyr+   r1   onnx.backend.test.case.baser   onnx.backend.test.case.noder   r   r   r@   r   r   <module>rH      sQ    #    , . EI!+7A ,T ,r   