
    Ng*                     ^    d dl mZmZ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 )    )AnyDictSequenceN)Base)expectc                   &    e Zd Zedd            ZdS )ConcatreturnNc                     ddgddgfddgddggddgddggfddgddggddgddgggd	d
gddggddgddgggfd} |                                  D ]\  }}d |D             }t          t          |d         j                            D ]}d t          t          |                    D             }t          j                            dt          |          dg|          }t          j	        ||          }t          |t          |          |gd|z   dz   t          |          z              t          t          |d         j                   d          D ]}d t          t          |                    D             }t          j                            dt          |          dg|          }t          j	        ||          }t          |t          |          |gd|z   dz   t          t          |                    z              d S )N                        	   
                     )1d2d3dc                 N    g | ]"}t          j        |t           j                   #S ))dtype)npasarrayfloat32).0vs     ^/var/www/html/ai-engine/env/lib/python3.11/site-packages/onnx/backend/test/case/node/concat.py
<listcomp>z!Concat.export.<locals>.<listcomp>   s)    GGG!bj"*555GGG    r   c                 2    g | ]}d t          |          z   S valuestrr$   ks     r&   r'   z!Concat.export.<locals>.<listcomp>   #    HHH7SVV+HHHr(   r	   output)inputsoutputsaxistest_concat__axis_)r2   r3   namec                 2    g | ]}d t          |          z   S r*   r,   r.   s     r&   r'   z!Concat.export.<locals>.<listcomp>*   r0   r(   _axis_negative_)itemsrangelenshapeonnxhelper	make_nodelistr!   concatenater   r-   abs)
test_cases	test_casevalues_valuesiin_argsnoder1   s           r&   exportzConcat.export   st    q6Aq6"FQF#q!fq!f%56a&1a&!QFQF#34b'B8$Bx"b&:;0
 0

 #-"2"2"4"4 	 	IwGGwGGGF3vay//00  HHU3v;;5G5GHHH{,,T']]XJQ -   22<<#H')3h>QG	     Cq	000!44  HHU3v;;5G5GHHH{,,T']]XJQ -   22<<#H')36GG#cRSff++U	    	 	r(   )r
   N)__name__
__module____qualname__staticmethodrK    r(   r&   r	   r	      s2        $ $ $ \$ $ $r(   r	   )typingr   r   r   numpyr!   r>   onnx.backend.test.case.baser   onnx.backend.test.case.noder   r	   rP   r(   r&   <module>rU      s   
 ' & & & & & & & & &      , , , , , , . . . . . .& & & & &T & & & & &r(   