U
    r±Ëh  ã                   @  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                   @  s0   e Zd Zeddœdd„ƒZeddœdd„ƒZdS )ÚScanÚNone)Úreturnc               	   C  sr  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 ddddddg¡ 	tj
¡ d¡}
t ddg¡ 	tj
¡ d¡}t ddddddg¡ 	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   Znum_scan_inputsÚbodyr   )r   r	   é   é   é   é   )r   r   r	   é	   é   Ztest_scan_sumé   ©r   r   ÚnameZopset_imports©ÚonnxÚhelperZmake_tensor_value_infoZTensorProtoÚFLOATZ	make_nodeZ
make_graphÚnpÚarrayZastypeZfloat32Zreshaper   Zmake_opsetid)r   r
   r   r   Úadd_nodeÚid_noder   Zno_sequence_lensÚnoder   r   r   r   © r-   úD/tmp/pip-unpacked-wheel-xnis5xre/onnx/backend/test/case/node/scan.pyÚexport_scan_8   sn      ÿ  ÿ  ÿ  ÿ  ÿ  ÿ   ÿû$$ûzScan.export_scan_8c               	   C  sl  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 ddddddg¡ 	tj
¡ d¡}	t ddg¡ 	tj
¡ d¡}
t ddddddg¡ 	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	   r   r    Ztest_scan9_sumr   r"   r$   )r   r
   r   r   r*   r+   r   r,   r   r   r   r   r-   r-   r.   Úexport_scan_9C   sl      ÿ  ÿ  ÿ  ÿ  ÿ  ÿ   ÿû$$ûzScan.export_scan_9N)Ú__name__Ú
__module__Ú__qualname__Ústaticmethodr/   r0   r-   r-   r-   r.   r      s   4r   )
Ú
__future__r   Znumpyr(   r%   Zonnx.backend.test.case.baser   Zonnx.backend.test.case.noder   r   r-   r-   r-   r.   Ú<module>   s
   