ColossalAI/colossalai/legacy/nn/__init__.py

5 lines
83 B
Python

from ._ops import *
from .layer import *
from .loss import *
from .metric import *