base_IIXIV / fla /ops /linear_attn /__init__.py
mainline777's picture
Duplicate from silx-ai/Quasar-Preview
41865df
Raw
History Blame Contribute Delete
250 Bytes
from .chunk import chunk_linear_attn
from .fused_chunk import fused_chunk_linear_attn
from .fused_recurrent import fused_recurrent_linear_attn
__all__ = [
'chunk_linear_attn',
'fused_chunk_linear_attn',
'fused_recurrent_linear_attn',
]