gsa-16M-test / fla /ops /scan /__init__.py
zaydzuhri's picture
Training in progress, step 2500
d152c99 verified
raw
history blame contribute delete
164 Bytes
# -*- coding: utf-8 -*-
from .parallel import parallel_scan
from .naive import naive_recurrent_scan
__all__ = [
'parallel_scan',
'naive_recurrent_scan'
]