wehub-resource-sync
13 lines
519 B
Python
13 lines
519 B
Python
# Copyright (c) Microsoft Corporation.
|
|
# SPDX-License-Identifier: Apache-2.0
|
|
|
|
# DeepSpeed Team
|
|
|
|
from .attention_base import DSSelfAttentionRegistry, DSSelfAttentionBase
|
|
from .embedding_base import DSEmbeddingRegistry, DSEmbeddingBase
|
|
from .linear_base import DSLinearRegistry, DSLinearBase
|
|
from .moe_base import DSMoERegistry, DSMoEBase
|
|
from .post_norm_base import DSPostNormRegistry, DSPostNormBase
|
|
from .pre_norm_base import DSPreNormRegistry, DSPreNormBase
|
|
from .unembed_base import DSUnembedRegistry, DSUnembedBase
|