2022-04-21 06:15:48 +00:00
|
|
|
from .init import colo_uniform
|
|
|
|
|
from .linear import colo_linear
|
2022-04-25 03:49:20 +00:00
|
|
|
from .element_wise import colo_mean
|
2022-04-25 08:01:52 +00:00
|
|
|
from .layernorm import colo_layernorm
|
|
|
|
|
from .loss import colo_cross_entropy
|
