Kernels:
Trusted publisher
File size: 347 Bytes
e19323e 888b93f e19323e 888b93f e19323e | 1 2 3 4 5 6 7 8 9 10 11 12 13 | from . import layers
from .ops.gated_delta_rule import chunk_gated_delta_rule, fused_recurrent_gated_delta_rule
from .ops.kda import chunk_kda, fused_recurrent_kda
__all__ = [
# original functions directly
"chunk_kda", "fused_recurrent_kda",
"chunk_gated_delta_rule", "fused_recurrent_gated_delta_rule",
# layers
"layers",
]
|