github/ColossalAI
1
0
Fork 0
mirror of https://github.com/hpcaitech/ColossalAI.git synced 2026-05-02 08:37:36 +00:00
Code Issues Packages Projects Releases Wiki Activity
Files
a9d64377bb237f34fdafaeec2abcfdfb6e080091
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