
    bIgA	                    N    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	y)    )annotationsN)Base)expectc                  B    e Zd Zedd       Zedd       Zedd       Zy)Identityc                     t         j                  j                  ddgdg      } t        j                  ddgddggggt        j
                  	      }t        | |g|gd
       y )Nr   xyinputsoutputs            dtypetest_identityr   r   nameonnxhelper	make_nodenparrayfloat32r   nodedatas     a/var/www/html/answerous/venv/lib/python3.12/site-packages/onnx/backend/test/case/node/identity.pyexportzIdentity.export   st    {{$$5E % 
 xx AA **

 	tTFTFI    c                 "   t         j                  j                  ddgdg      } t        j                  ddgddggggt        j
                  	      t        j                  ddgdd
ggggt        j
                  	      g}t        | |g|gd       y )Nr   r	   r
   r   r   r   r   r   r      test_identity_sequencer   r   r   s     r!   export_sequencezIdentity.export_sequence$   s    {{$$5E % 
 HH FF jj
 HH FF jj

2 	tTFTF9QRr#   c            
        t         j                  j                  t         j                  j                  dg      } t         j                  j                  |       }t         j                  j                  |      }t         j                  j                  ddgdg      }t        j                  g d      j                  t        j                        g}t        ||g|gdt         j                  j                  d	d
      g|g|g       y )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	   s        r!   export_identity_optzIdentity.export_identity_optG   s    KK66""1# 7 
	 KK88C	KK88C	--zI; . 
 XXo&--bjj9:3C$;;33B;<(k ){	
r#   N)returnNone)__name__
__module____qualname__staticmethodr"   r'   r=    r#   r!   r   r      sA    J J*  S  SD 
 
r#   r   )

__future__r   numpyr   r   onnx.backend.test.case.baser   onnx.backend.test.case.noder   r   rD   r#   r!   <module>rI      s$    #   , .P
t P
r#   