ColossalAI/colossalai/tensor/_ops
Jiarui Fang 802ac297cc
[Tensor] remove useless import in tensor dir (#997)
2022-05-18 14:54:51 +08:00
..
__init__.py add DistSpec for loss and test_model (#947) 2022-05-13 20:29:50 +08:00
addmm.py [Tensor] remove useless import in tensor dir (#997) 2022-05-18 14:54:51 +08:00
element_wise.py [Tensor] make ColoTensor more robust for getattr (#886) 2022-04-27 10:57:49 +08:00
embedding.py [Tensor] remove useless import in tensor dir (#997) 2022-05-18 14:54:51 +08:00
layernorm.py add DistSpec for loss and test_model (#947) 2022-05-13 20:29:50 +08:00
linear.py [tensor] derive compute pattern from dist spec (#971) 2022-05-16 14:58:08 +08:00
loss.py [tensor] derive compute pattern from dist spec (#971) 2022-05-16 14:58:08 +08:00