U
    qh                     @  s,   d dl mZ d dlmZ G dd deZdS )    )annotations)OpRunc                   @  s   e Zd ZdddddZdS )SequenceMapN)body
attributesc                G  s  t |dkrt|d trd }t||d D ]b\}}|jd ||jd |i}|d |}	|d krpdd |	D }q,t||	D ]\}
}|
| qzq,t|S tt|jdd  |}d }|D ]X}|||jd < |jd ||d}	|d krdd |	D }qt||	D ]\}
}|
| qqt|S )N   r   c                 S  s   g | ]
}|gqS  r   .0ir   r   F/tmp/pip-unpacked-wheel-xnis5xre/onnx/reference/ops/op_sequence_map.py
<listcomp>   s     z$SequenceMap._run.<locals>.<listcomp>)r   c                 S  s   g | ]
}|gqS r   r   r	   r   r   r   r      s     )	len
isinstancelistzipZinput_namesrunappendtupledict)selfZinput_sequencer   r   Zadditional_inputsresZobj1obj2Zfeedsrsr   objr   r   r   _run
   s(    zSequenceMap._run)__name__
__module____qualname__r   r   r   r   r   r   	   s   r   N)
__future__r   Zonnx.reference.op_runr   r   r   r   r   r   <module>   s   