
    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                   t    \ rS rSr\S 5       r\S 5       r\S 5       r\S 5       r\S 5       r	\S 5       r
Srg	)
SequenceMap   c            
         [         R                  R                  [         R                  R                  SS/S/5      /S[         R                  R	                  S[         R
                  R                  S/5      /[         R                  R	                  S[         R
                  R                  S/5      /5      n [         R                  R                  SS/S	/U S
9n[        S5       Vs/ s HA  n[        R                  R                  SSS5      R                  [        R                  5      PMC     nnUn[         R                  R                  [         R                  R                  [         R
                  R                  S/5      5      /n[         R                  R                  [         R                  R                  [         R
                  R                  S/5      5      /n[        UU/U/UUSS9  g s  snf )NIdentityin0out0seq_map_bodyNMr   xyinputsoutputsbody                 ?
   %test_sequence_map_identity_1_sequencer   r   input_type_protosoutput_type_protosnameonnxhelper
make_graph	make_nodemake_tensor_value_infoTensorProtoFLOATrangenprandomuniformastypefloat32make_sequence_type_protomake_tensor_type_protor   )r   node_r   r   r   r   s          b/var/www/fran/franai/venv/lib/python3.13/site-packages/onnx/backend/test/case/node/sequence_map.py'export_sequence_map_identity_1_sequence3SequenceMap.export_sequence_map_identity_1_sequence   s   {{%%[["":wAB[[//t7G7G7M7MPSuUV[[//8H8H8N8NQTPUVW	
 {{$$3%#T % 
 JOqRARYYsC,33BJJ?RKK002243C3C3I3IC5Q
 KK002243C3C3I3IC5Q

 	3C/18	
 Ss   &AG1c                     [         R                  R                  [         R                  R                  SS/S/5      [         R                  R                  SS/S/5      /S[         R                  R	                  S[         R
                  R                  S/5      [         R                  R	                  S[         R
                  R                  S/5      /[         R                  R	                  S[         R
                  R                  S/5      [         R                  R	                  S[         R
                  R                  S/5      /5      n [         R                  R                  S	S
S/SS/U S9n[        S5       Vs/ s H_  n[        R                  R                  SS[        R                  R                  SS5      5      R                  [        R                  5      PMa     nn[        S5       Vs/ s H_  n[        R                  R                  SS[        R                  R                  SS5      5      R                  [        R                  5      PMa     nnUnUn[         R                  R                  [         R                  R                  [         R
                  R                  S/5      5      [         R                  R                  [         R                  R                  [         R
                  R                  S/5      5      /n[         R                  R                  [         R                  R                  [         R
                  R                  S/5      5      [         R                  R                  [         R                  R                  [         R
                  R                  S/5      5      /n[!        UX4/XV/UUSS9  g s  snf s  snf )Nr	   r
   r   in1out1r   r   r   r   x0x1y0y1r   r   r   r      r   &test_sequence_map_identity_2_sequencesr   r   r    r!   r"   r#   r$   r%   r&   r'   r(   r)   randintr*   r+   r,   r-   r   	r   r.   r/   r6   r7   r8   r9   r   r   s	            r0   (export_sequence_map_identity_2_sequences4SequenceMap.export_sequence_map_identity_2_sequences/   s   {{%%%%j5'F8D%%j5'F8D 224++11C5 224++11C5	 22D,,22SE 22D,,22SE	
0 {{$$4,t4 % 
 1X
 IIc3		(9(9!R(@AHHT 	 
 1X
 IIc3		(9(9!R(@AHHT 	 
 KK002243C3C3I3IC5Q KK002243C3C3I3IC5Q	
 KK002243C3C3I3IC5Q KK002243C3C3I3IC5Q	
 	8H/19	
5

s   <A&O1A&Oc                  d   [         R                  R                  [         R                  R                  SS/S/5      [         R                  R                  SS/S/5      /S[         R                  R	                  S[         R
                  R                  S/5      [         R                  R	                  S[         R
                  R                  S/5      /[         R                  R	                  S[         R
                  R                  S/5      [         R                  R	                  S[         R
                  R                  S/5      /5      n [         R                  R                  S	S
S/SS/U S9n[        S5       Vs/ s H_  n[        R                  R                  SS[        R                  R                  SS5      5      R                  [        R                  5      PMa     nn[        R                  R                  SS[        R                  R                  SS5      5      R                  [        R                  5      nUn[        S5       Vs/ s H  o$PM     nn[         R                  R                  [         R                  R                  [         R
                  R                  S/5      5      [         R                  R                  [         R
                  R                  S/5      /n[         R                  R                  [         R                  R                  [         R
                  R                  S/5      5      [         R                  R                  [         R                  R                  [         R
                  R                  S/5      5      /n[!        UX4/XV/UUSS9  g s  snf s  snf )Nr	   r
   r   r4   r5   r   r   r   r   r6   r7   r8   r9   r   r   r   r   r:   r   .test_sequence_map_identity_1_sequence_1_tensorr   r<   r>   s	            r0   0export_sequence_map_identity_1_sequence_1_tensor<SequenceMap.export_sequence_map_identity_1_sequence_1_tensorp   s   {{%%%%j5'F8D%%j5'F8D 224++11C5 224++11C5	 22D,,22SE 22D,,22SE	
0 {{$$4,t4 % 
 1X
 IIc3		(9(9!R(@AHHT 	 
 YYsC):):1b)ABII"**U(#(Qb(#KK002243C3C3I3IC5Q KK..t/?/?/E/EuM	
 KK002243C3C3I3IC5Q KK002243C3C3I3IC5Q	
 	8H/1A	
+
 $s   <A&N(N-c            
         [         R                  R                  [         R                  R                  SSS/S/5      /S[         R                  R	                  S[         R
                  R                  S/5      [         R                  R	                  S[         R
                  R                  S/5      /[         R                  R	                  S[         R
                  R                  S/5      /5      n [         R                  R                  SSS	/S
/U S9n[        S5       Vs/ s H#  n[        R                  R                  SS5      PM%     nn[        S5       Vs/ s HC  n[        R                  R                  SSX4   5      R                  [        R                  5      PME     nn[        S5       Vs/ s HC  n[        R                  R                  SSX4   5      R                  [        R                  5      PME     nn[        S5       Vs/ s H  oEU   Xd   -   PM     nn[         R                  R                  [         R                  R                  [         R
                  R                  S/5      5      [         R                  R                  [         R                  R                  [         R
                  R                  S/5      5      /n[         R                  R                  [         R                  R                  [         R
                  R                  S/5      5      /n	[!        UXV/U/UU	SS9  g s  snf s  snf s  snf s  snf )NAddr
   r4   r   r   r   r   r6   r7   r8   r   r   r:   r   r   r   !test_sequence_map_add_2_sequencesr   )r   r    r!   r"   r#   r$   r%   r&   r'   r(   r=   r)   r*   r+   r,   r-   r   )
r   r.   r/   r   kr6   r7   r8   r   r   s
             r0   #export_sequence_map_add_2_sequences/SequenceMap.export_sequence_map_add_2_sequences   s   {{%%[[""55%.6(CD224++11C5 224++11C5	 [[//8H8H8N8NQTPUVW
 {{$$4,T % 
 05Qx8x!RYYq"%x8LQRSHUHqbiiS!$/66rzzBHULQRSHUHqbiiS!$/66rzzBHU%*1X.XebemX.KK002243C3C3I3IC5Q KK002243C3C3I3IC5Q	
 KK002243C3C3I3IC5Q

 	8D/14	
# 9UU.s   !*L5A
L:3A
L?Mc            
         [         R                  R                  [         R                  R                  SSS/S/5      /S[         R                  R	                  S[         R
                  R                  S/5      [         R                  R	                  S[         R
                  R                  S/5      /[         R                  R	                  S[         R
                  R                  S/5      /5      n [         R                  R                  SSS	/S
/U S9n[        S5       Vs/ s HA  n[        R                  R                  SSS5      R                  [        R                  5      PMC     nn[        R                  R                  SSS5      R                  [        R                  5      n[        S5       Vs/ s H
  oSU   U-   PM     nn[         R                  R                  [         R                  R                  [         R
                  R                  S/5      5      [         R                  R                  [         R
                  R                  S/5      /n[         R                  R                  [         R                  R                  [         R
                  R                  S/5      5      /n[        UX4/U/UUSS9  g s  snf s  snf )NrF   r
   r4   r   r   r   r   r6   r7   r8   r   r   r   r   r   )test_sequence_map_add_1_sequence_1_tensorr   r   )	r   r.   rH   r6   r7   ir8   r   r   s	            r0   +export_sequence_map_add_1_sequence_1_tensor7SequenceMap.export_sequence_map_add_1_sequence_1_tensor   s-   {{%%[[""55%.6(CD224++11C5 224++11C5	 [[//8H8H8N8NQTPUVW
 {{$$4,T % 
 KPPQ(S(QbiiS"-44RZZ@(SYYsC,33BJJ?"'(+(Qebj(+KK002243C3C3I3IC5Q KK..t/?/?/E/EuM	
 KK002243C3C3I3IC5Q

 	8D/1<	
 T+s   !AK 6Kc            
         [         R                  R                  [         R                  R                  SS/S/5      /S[         R                  R	                  S[         R
                  R                  / SQ5      /[         R                  R	                  S[         R
                  R                  S/5      /5      n [         R                  R                  SS/S	/U S
9n[        R                  " / SQ[        R                  S9[        R                  " / SQ[        R                  S9[        R                  " / SQ[        R                  S9/nU Vs/ s H&  n[        R                  " U[        R                  S9PM(     nn[         R                  R                  [         R                  R                  [         R
                  R                  / SQ5      5      /n[         R                  R                  [         R                  R                  [         R
                  R                  S/5      5      /n[        UU/U/UUSS9  g s  snf )NShaper   shaper   )HWCr   r   in_seqshapesr   )(      r   )dtype)   r   r   )r      r    test_sequence_map_extract_shapesr   )r   r    r!   r"   r#   r$   r%   INT64r'   arrayint64zerosr+   r,   r-   r   )r   r.   rW   rR   r6   r   r   s          r0   "export_sequence_map_extract_shapes.SequenceMap.export_sequence_map_extract_shapes  s   {{%%[[""7SEG9=>22))//
 [[//9I9I9O9ORSQTUV	
 {{$$8*xjt % 

 HH[1HH[1HHZrxx0

 >DDVEbhhuBJJ/VDKK0022$$**O
 KK002243C3C3I3IA3O

 	4H/13	
 Es   -H; N)__name__
__module____qualname____firstlineno__staticmethodr1   r?   rC   rI   rN   rb   __static_attributes__rd       r0   r   r      s}    
 
B >
 >
@ 9
 9
v +
 +
Z (
 (
T )
 )
rk   r   )numpyr'   r   onnx.backend.test.case.baser   onnx.backend.test.case.noder   r   rd   rk   r0   <module>ro      s!   
   , .c
$ c
rk   