github
/
ColossalAI
mirror of https://github.com/hpcaitech/ColossalAI
1
0
Fork 0
Code Issues Projects Releases Wiki Activity
ColossalAI/colossalai/tensor
History
Jiarui Fang 96211c2cc8
[tensor] customized op returns ColoTensor (#875) 
* [tensor] customized op returns ColoTensor

* polish

* polish code
 		2022-04-26 13:23:59 +08:00
..
_ops [tensor] customized op returns ColoTensor (#875) 2022-04-26 13:23:59 +08:00
__init__.py [Tensor] Add function to spec and update linear 1Drow and unit tests (#869) 2022-04-26 10:15:26 +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] reorganize files (#820) 2022-04-21 14:15:48 +08:00