ColossalAI/colossalai/nn/_ops
Jiarui Fang 1b657f9ce1
[tensor] revert local view back (#1178)
2022-06-27 18:38:34 +08:00
..
__init__.py [tensor] add embedding bag op (#1156) 2022-06-22 15:54:03 +08:00
_utils.py reorgnize colotensor directory (#1062) 2022-06-03 18:04:22 +08:00
addmm.py [Tensor] rename some APIs in TensorSpec and Polish view unittest (#1176) 2022-06-27 15:56:11 +08:00
element_wise.py [ColoTensor] rename APIs and add output_replicate to ComputeSpec (#1168) 2022-06-24 13:08:54 +08:00
embedding.py [tensor] revert local view back (#1178) 2022-06-27 18:38:34 +08:00
embedding_bag.py [Tensor] rename some APIs in TensorSpec and Polish view unittest (#1176) 2022-06-27 15:56:11 +08:00
layernorm.py [ColoTensor] rename APIs and add output_replicate to ComputeSpec (#1168) 2022-06-24 13:08:54 +08:00
linear.py [Tensor] rename some APIs in TensorSpec and Polish view unittest (#1176) 2022-06-27 15:56:11 +08:00
loss.py [Tensor] rename some APIs in TensorSpec and Polish view unittest (#1176) 2022-06-27 15:56:11 +08:00