ColossalAI/colossalai/tensor
Jiarui Fang 72cdc06875
[Tensor] make ColoTensor more robust for getattr (#886)
* [Tensor] make ColoTensor more robust for getattr

* polish

* polish
2022-04-27 10:57:49 +08:00
..
_ops [Tensor] make ColoTensor more robust for getattr (#886) 2022-04-27 10:57:49 +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 ColoTensor more robust for getattr (#886) 2022-04-27 10:57:49 +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