o
    àÙ·i•  ã                   @   s    d dl mZ G dd„ deƒZdS )é    )ÚEnumc                   @   s8   e Zd Zdd„ ZdZdZdZdZdZdZ	e
d	d
„ ƒZdS )ÚForwardPatternc                 C   s(   || _ || _|| _|| _|| _|| _d S ©N)ÚReturn_H_FirstÚReturn_H_OnlyÚForward_H_onlyÚInÚOutÚ	Supported)Úselfr   r   r   r   r	   r
   © r   úU/home/ubuntu/.local/lib/python3.10/site-packages/cache_dit/caching/forward_pattern.pyÚ__init__   s   	
zForwardPattern.__init__)TFF©Úhidden_statesÚencoder_hidden_statesr   T)FFFr   ©r   r   T)FTFr   ©r   T)FTTr   r   T)TFTr   r   T)FFTr   r   Tc                   C   s   t jt jt jt jt jt jgS r   )r   Ú	Pattern_0Ú	Pattern_1Ú	Pattern_2Ú	Pattern_3Ú	Pattern_4Ú	Pattern_5r   r   r   r   Úsupported_patternsK   s   úz!ForwardPattern.supported_patternsN)Ú__name__Ú
__module__Ú__qualname__r   r   r   r   r   r   r   Ústaticmethodr   r   r   r   r   r      s    						r   N)Úenumr   r   r   r   r   r   Ú<module>   s    