ColossalAI/colossalai/nn/_ops
Jiarui Fang 177c374401
remove gather out in parallel action (#1163)
2022-06-23 16:35:05 +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 remove gather out in parallel action (#1163) 2022-06-23 16:35:05 +08:00
element_wise.py [tensor] add more element-wise ops (#1155) 2022-06-22 15:16:47 +08:00
embedding.py remove gather out in parallel action (#1163) 2022-06-23 16:35:05 +08:00
embedding_bag.py remove gather out in parallel action (#1163) 2022-06-23 16:35:05 +08:00
layernorm.py reorgnize colotensor directory (#1062) 2022-06-03 18:04:22 +08:00
linear.py remove gather out in parallel action (#1163) 2022-06-23 16:35:05 +08:00
loss.py reorgnize colotensor directory (#1062) 2022-06-03 18:04:22 +08:00