mirrors / dmlc--dgl

文件历史

2,272 提交 34 分支 42 Git 标签
0.8.x
提交图

1 次代码提交

作者 SHA1 备注 提交日期
nv-dlasalle 7359481497 [Feature][GPU] Add function for setting weights of a sparse embedding on multiple GPUs. (#3047) 
* add unit test

* Extend NDArrayPartition object

* Add method for setting embedding, and improve documentation

* Sync before returning

* Use name unique to sparse embedding class to avoid delete

Co-authored-by: xiang song(charlie.song) <classicxsong@gmail.com>
 2021-06-22 08:48:12 -07:00