
    Ng	                     J    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 )    N)Base)expectc                   V    e Zd Zedd            Zedd            Zedd            ZdS )IdentityreturnNc                      t           j                            ddgdg          } t          j        ddgddggggt          j        	          }t          | |g|gd
           d S )Nr   xyinputsoutputs            dtypetest_identityr   r   nameonnxhelper	make_nodenparrayfloat32r   nodedatas     `/var/www/html/ai-engine/env/lib/python3.11/site-packages/onnx/backend/test/case/node/identity.pyexportzIdentity.export   s    {$$5E % 
 
 x AA *

 

 

 	tTFTFIIIIII    c                     t           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           d S )Nr   r	   r
   r   r   r   r   r   r      test_identity_sequencer   r   r   s     r!   export_sequencezIdentity.export_sequence#   s    {$$5E % 
 
 H FF j
 
 
 H FF j
 
 

2 	tTFTF9QRRRRRRr#   c            
         t           j                            t           j        j        dg          } t           j                            |           }t           j                            |          }t           j                            ddgdg          }t          j	        g d          
                    t          j                  g}t          ||g|gdt           j                            d	d
          g|g|g           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	   s        r!   export_identity_optzIdentity.export_identity_optF   s    K66"1# 7 
 
	 K88CC	K88CC	--zI; . 
 
 Xooo&&--bj99:3C$;33B;;<(k ){	
 	
 	
 	
 	
 	
r#   )r   N)__name__
__module____qualname__staticmethodr"   r'   r=    r#   r!   r   r      s{        J J J \J*  S  S  S \ SD 
 
 
 \
 
 
r#   r   )numpyr   r   onnx.backend.test.case.baser   onnx.backend.test.case.noder   r   rB   r#   r!   <module>rF      s~   
      , , , , , , . . . . . .P
 P
 P
 P
 P
t P
 P
 P
 P
 P
r#   