gla-16M-test / fla /ops /gla /__init__.py
zaydzuhri's picture
Training in progress, step 2048
2f9282b verified
raw
history blame contribute delete
225 Bytes
# -*- coding: utf-8 -*-
from .chunk import chunk_gla
from .fused_chunk import fused_chunk_gla
from .fused_recurrent import fused_recurrent_gla
__all__ = [
'chunk_gla',
'fused_chunk_gla',
'fused_recurrent_gla'
]