Jiarui Fang 7f76517a85 [Tensor] make a simple net works with 1D row TP (#879)

2022-04-26 18:11:47 +08:00

..

_ops

[Tensor] Add some attributes to ColoTensor (#877)

2022-04-26 15:10:47 +08:00

__init__.py

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

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

colo_tensor.py

[Tensor] make a simple net works with 1D row TP (#879)

2022-04-26 18:11:47 +08:00

op_wrapper.py

[tensor] reorganize files (#820)

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

spec.py

[tensor] customized op returns ColoTensor (#875)

2022-04-26 13:23:59 +08:00

utils.py

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

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