
    /jA	                    V    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	dS )    )annotationsN)Base)expectc                  V    e Zd Zedd            Zedd            Zedd            ZdS )IdentityreturnNonec                     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     i/home/longshao/multi-rider-rag/.venv/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_optG   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%   N)r   r	   )__name__
__module____qualname__staticmethodr$   r)   r?    r%   r#   r   r      s{        J J J \J*  S  S  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
 P
 P
 P
 P
t P
 P
 P
 P
 P
r%   