ColossalAI/colossalai/tensor/_ops
ver217 a3b66f6def
[tensor] refactor parallel action (#1007)
* refactor parallel action

* polish unit tests
2022-05-20 20:19:58 +08:00
..
__init__.py add DistSpec for loss and test_model (#947) 2022-05-13 20:29:50 +08:00
_utils.py [tensor] refactor colo-tensor (#992) 2022-05-19 12:44:59 +08:00
addmm.py [tensor] refactor parallel action (#1007) 2022-05-20 20:19:58 +08:00
element_wise.py [tensor] refactor colo-tensor (#992) 2022-05-19 12:44:59 +08:00
embedding.py [tensor] refactor parallel action (#1007) 2022-05-20 20:19:58 +08:00
layernorm.py [tensor] refactor colo-tensor (#992) 2022-05-19 12:44:59 +08:00
linear.py [tensor] refactor parallel action (#1007) 2022-05-20 20:19:58 +08:00
loss.py [tensor] refactor parallel action (#1007) 2022-05-20 20:19:58 +08:00