o
    
j8                     @   sJ   G d d dZ G dd dZG dd dZG dd dZG dd	 d	Zd
S )c                   @   ,   e Zd ZdZdZdZdZdZdZdZ	dZ
d	S )
	Operators	AttentionLayerNormalizationMultiHeadAttentionPackedAttentionPackedMultiHeadAttentionRemovePaddingRestorePaddingSkipLayerNormalizationN)__name__
__module____qualname__	ATTENTION	LAYERNORMMULTI_HEAD_ATTENTIONPACKEDATTENTIONPACKED_MULTI_HEAD_ATTENTIONREMOVEPADDINGRESTOREPADDINGSKIPLAYERNORM r   r   U/home/kuhnn/.local/lib/python3.10/site-packages/onnxruntime/transformers/constants.pyr          r   c                   @   s(   e Zd ZdZdZdZdZdZdZdZ	dS )	AttentionInputIDs                      N)
r   r   r   INPUTWEIGHTSBIAS
MASK_INDEXPASTATTENTION_BIASPAST_SEQUENCE_LENGTHr   r   r   r   r      s    r   c                   @   s   e Zd ZdZdZdS )AttentionOutputIDsr   r   N)r   r   r   OUTPUTPRESENTr   r   r   r   r(      s    r(   c                   @   r   )
MultiHeadAttentionInputIDsr   r   r   r   r   r   r       N)r   r   r   QUERYKEYVALUEr#   KEY_PADDING_MASKr&   PAST_KEY
PAST_VALUEr   r   r   r   r+   !   r   r+   c                   @   s   e Zd ZdZdZdZdS )MultiHeadAttentionOutputIDsr   r   r   N)r   r   r   r)   PRESENT_KEYPRESENT_VALUEr   r   r   r   r3   ,   s    r3   N)r   r   r(   r+   r3   r   r   r   r   <module>   s
   
