github/ColossalAI
1
0
Fork 0
mirror of https://github.com/hpcaitech/ColossalAI.git synced 2026-07-13 12:05:48 +00:00
Code Issues Packages Projects Releases Wiki Activity
Files
cb413ccf28a82602eedfeebb3be5224aa02486dc
ColossalAI/colossalai/nn/layer/parallel_1d/_utils.py
Ziyue Jiang 0653c63eaa [Tensor] 1d row embedding (#1075) 
* Add CPU 1d row embedding

* polish
2022-06-08 12:04:59 +08:00

4.9 KiB
Raw Blame History

View Raw
Reference in New Issue View Git Blame Copy Permalink