ColossalAI/colossalai/tensor/_ops/__init__.py

from .linear import colo_linear
from .element_wise import *
from .layernorm import colo_layernorm
from .loss import colo_cross_entropy
[tensor] reorganize files (#820) 2022-04-21 06:15:48 +00:00			`from .linear import colo_linear`
[tensor] customized op returns ColoTensor (#875) * [tensor] customized op returns ColoTensor * polish * polish code 2022-04-26 05:23:59 +00:00			`from .element_wise import *`
[tensor] add cross_entrophy_loss (#868) 2022-04-25 08:01:52 +00:00			`from .layernorm import colo_layernorm`
[Tensor] Add function to spec and update linear 1Drow and unit tests (#869) 2022-04-26 02:15:26 +00:00			`from .loss import colo_cross_entropy`