ColossalAI/colossalai/tensor
Jiarui Fang e43f83aa5c
[Tensor] get named parameters for model using ColoTensors (#874)
2022-04-26 14:08:01 +08:00
..
_ops [tensor] customized op returns ColoTensor (#875) 2022-04-26 13:23:59 +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] Add function to spec and update linear 1Drow and unit tests (#869) 2022-04-26 10:15:26 +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