__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 |