o
    %Th8  ã                   @   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   úc/home/air/segue/gemini/back/venv/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
   
