
    /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 )ScanreturnNonec            
     d   t           j                            dt           j        j        dg          } t           j                            dt           j        j        dg          }t           j                            dt           j        j        dg          }t           j                            dt           j        j        dg          }t           j                            dddgdg          }t           j                            ddgdg          }t           j                            ||gd	| |g||g          }d
}t           j                            d|ddgddgd|          }t          j        ddg          	                    t          j
                                      d          }	t          j        g d          	                    t          j
                                      d          }
t          j        ddg          	                    t          j
                                      d          }t          j        g d          	                    t          j
                                      d          }t          ||	|
g||gdt           j                            d
d          g           d S )Nsum_in   nextsum_outscan_outAddinputsoutputsIdentity	scan_body r   initialxyz   r   r   num_scan_inputsbodyr   )r   r   r   r               )r   r    r   	      r   r   r!   r#   r$   r%   test_scan_sum   r   r   nameopset_importsonnxhelpermake_tensor_value_infoTensorProtoFLOAT	make_node
make_graphnparrayastypefloat32reshaper   make_opsetid)r   next_r   r   add_nodeid_noder   no_sequence_lensnoder   r   r   r   s                e/home/longshao/multi-rider-rag/.venv/lib/python3.11/site-packages/onnx/backend/test/case/node/scan.pyexport_scan_8zScan.export_scan_8   sy    33d&,qc
 
 2264;K;QTUSVWW+44t'-s
 
 ;55(.
 
 ;((8V,yk ) 
 
 +''	{ZL ( 
 
 K**wvuo?R
 
	 {$$$i5#J % 
 
 (Aq6""))"*55==fEEH'''((//
;;CCINNHaW$$RZ0088@@H((())00<<DDYOOQ<F ;33B::;	
 	
 	
 	
 	
 	
    c            
     ^   t           j                            dt           j        j        dg          } t           j                            dt           j        j        dg          }t           j                            dt           j        j        dg          }t           j                            dt           j        j        dg          }t           j                            dddgdg          }t           j                            ddgdg          }t           j                            ||gd	| |g||g          }t           j                            d
ddgddgd|          }t          j        ddg          	                    t          j
                                      d          }t          j        g d          	                    t          j
                                      d          }	t          j        ddg          	                    t          j
                                      d          }
t          j        g d          	                    t          j
                                      d          }t          |||	g|
|gdt           j                            dd          g           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&   test_scan9_sumr   r)   r,   )r   r:   r   r   r;   r<   r   r>   r   r   r   r   s               r?   export_scan_9zScan.export_scan_9A   sq    33d&,qc
 
 2264;K;QTUSVWW+44t'-s
 
 ;55(.
 
 ;((8V,yk ) 
 
 +''	{ZL ( 
 
 K**wvuo?R
 
	 {$$s##J % 
 
 (Aq6""))"*55==dCCH'''((//
;;CCFKKHaW$$RZ0088>>H((())00<<DDVLLQ<F!;33B::;	
 	
 	
 	
 	
 	
rA   N)r   r	   )__name__
__module____qualname__staticmethodr@   rD    rA   r?   r   r      sS        0
 0
 0
 \0
d /
 /
 /
 \/
 /
 /
rA   r   )

__future__r   numpyr4   r-   onnx.backend.test.case.baser   onnx.backend.test.case.noder   r   rI   rA   r?   <module>rN      s    # " " " " "      , , , , , , . . . . . .d
 d
 d
 d
 d
4 d
 d
 d
 d
 d
rA   