Ziyue Jiang f593a5637e [Tensor] add embedding tp1d row (#904)

2022-04-29 14:10:05 +08:00

..

_ops

[Tensor] add embedding tp1d row (#904)

2022-04-29 14:10:05 +08:00

optim

[Tensor] initialize the ColoOptimizer (#898)

2022-04-28 15:23:40 +08:00

__init__.py

[Tensor] initialize the ColoOptimizer (#898)

2022-04-28 15:23:40 +08:00

colo_tensor.py

[Tensor] add embedding tp1d row (#904)

2022-04-29 14:10:05 +08:00

op_wrapper.py

[tensor] reorganize files (#820)

2022-04-21 14:15:48 +08:00

spec.py

[Tensor] add ColoTensor TP1Dcol Embedding (#899)

2022-04-28 17:45:06 +08:00

utils.py

[Tensor] get named parameters for model using ColoTensors (#874)

2022-04-26 14:08:01 +08:00