ColossalAI/colossalai/tensor/_ops/__init__.py

3 lines
98 B
Python
Raw Normal View History

2022-04-21 06:15:48 +00:00
from .init import colo_uniform
from .linear import colo_linear
from .element_wise import colo_mean