o
    wi                     @  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rLt|d trLd }t||d D ]1\}}|jd ||jd |i}|d |}	|d u r8dd |	D }qt||	D ]	\}
}|
| q=qt|S tt|jdd  |}d }|D ],}|||jd < |jd ||d}	|d u rydd |	D }q\t||	D ]	\}
}|
| q~q\t|S )N   r   c                 S     g | ]}|gqS  r	   .0ir	   r	   _/home/ubuntu/sommelier/.venv/lib/python3.10/site-packages/onnx/reference/ops/op_sequence_map.py
<listcomp>       z$SequenceMap._run.<locals>.<listcomp>)r   c                 S  r   r	   r	   r
   r	   r	   r   r      r   )	len
isinstancelistzipinput_namesrunappendtupledict)selfinput_sequencer   r   additional_inputsresobj1obj2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   onnx.reference.op_runr   r   r	   r	   r	   r   <module>   s   