o
    wi                     @  sH   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 deZ	dS )    )annotationsN)Base)expectc                   @  s(   e Zd ZedddZedddZdS )	ScanreturnNonec               	   C  sb  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jdddgdgd}t jjddgdgd}t j||gd	| |g||g}d
}t jjd|ddgddgd|d}tddg	tj
d}	tg d	tj
d}
tddg	tj
d}tg d	tj
d}t||	|
g||gdt jd
dgd 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   r
   r   r   add_nodeid_noder   no_sequence_lensnoder   r   r   r    r;   ]/home/ubuntu/sommelier/.venv/lib/python3.10/site-packages/onnx/backend/test/case/node/scan.pyexport_scan_8   sP   

zScan.export_scan_8c               	   C  s\  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jdddgdgd}t jjddgdgd}t j||gd	| |g||g}t jjd
ddgddgd|d}tddg	tj
d}tg d	tj
d}	tddg	tj
d}
tg d	tj
d}t|||	g|
|gdt jdd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#   test_scan9_sumr   r&   r)   )r   r
   r   r   r7   r8   r   r:   r   r   r   r   r;   r;   r<   export_scan_9C   sN   

zScan.export_scan_9N)r   r   )__name__
__module____qualname__staticmethodr=   r?   r;   r;   r;   r<   r      s
    4r   )

__future__r   numpyr1   r*   onnx.backend.test.case.baser   onnx.backend.test.case.noder   r   r;   r;   r;   r<   <module>   s   