mirrors / dmlc--dgl

文件历史

2,828 提交 34 分支 42 Git 标签
1.0.x
提交图

2 次代码提交

作者 SHA1 备注 提交日期
Hongzhi (Steve), Chen 89a4cc4db2 [Misc] Black auto fix. (#4694) 
Co-authored-by: Steve <ubuntu@ip-172-31-34-29.ap-northeast-1.compute.internal>
 2022-10-11 00:12:30 +08:00
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