o
    soiA	                     @  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                   @  s6   e Zd Zed
ddZed
ddZed
ddZd	S )IdentityreturnNonec                  C  sN   t jjddgdgd} tjddgddggggtjd	}t| |g|gd
d d S )Nr   xyinputsoutputs            dtypetest_identityr   r   nameonnxhelper	make_nodenparrayfloat32r   nodedata r    X/home/ubuntu/.local/lib/python3.10/site-packages/onnx/backend/test/case/node/identity.pyexport   s   zIdentity.exportc                  C  sn   t jjddgdgd} tjddgddggggtjd	tjddgdd
ggggtjd	g}t| |g|gdd d S )Nr   r   r	   r
   r   r   r   r   r      test_identity_sequencer   r   r   r    r    r!   export_sequence$   s.   zIdentity.export_sequencec               	   C  s   t jjt jjdgd} t j| }t j|}t jjddgdgd}t	g d
tjg}t||g|gdt jd	d
g|g|gd d S )Nr#   )shaper   opt_inopt_outr
   )r   r   r   r   r#   test_identity_opt    )r   r   r   opset_importsinput_type_protosoutput_type_protos)r   r   make_tensor_type_protoTensorProtoFLOATmake_sequence_type_protomake_optional_type_protor   r   r   astyper   r   make_opsetid)	ten_in_tp	seq_in_tp	opt_in_tpidentity_noder   r    r    r!   export_identity_optG   s$   


zIdentity.export_identity_optN)r   r   )__name__
__module____qualname__staticmethodr"   r%   r:   r    r    r    r!   r      s    "r   )

__future__r   numpyr   r   onnx.backend.test.case.baser   onnx.backend.test.case.noder   r   r    r    r    r!   <module>   s   