oneflow
oneflow copied to clipboard
Dev gcn spmm [don't merge]
This PR records the progress relating to GCN and SpMM. A prior PR is #9092.
Two components are added:
-
spmm
op & kernels - gcn model
The gcn model is implemented based on SpMM. Currently, it allows (1) mini-batch training based graph sampling and (2) full-batch training OneEmbedding.
The (unfinished) goal is to achieve mini-batch training with OneEmbedding.
The detailed explanantions can be found in /gcn
folder.
Thank you for your submission! We really appreciate it. Like many open source projects, we ask that you sign our Contributor License Agreement before we can accept your contribution.
You have signed the CLA already but the status is still pending? Let us recheck it.