You can not select more than 25 topics Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
ColossalAI/colossalai/nn/parallel/layers
Ziyue Jiang dd0420909f
[Tensor] rename parallel_action (#1174)
2 years ago
..
__init__.py [refactory] add nn.parallel module (#1068) 3 years ago
colo_module.py [refactory] add nn.parallel module (#1068) 3 years ago
embedding.py [refactory] add nn.parallel module (#1068) 3 years ago
linear.py [refactory] add nn.parallel module (#1068) 3 years ago
module_utils.py [Tensor] rename parallel_action (#1174) 2 years ago