Skip to content
This repository has been archived by the owner on Nov 17, 2023. It is now read-only.

[DOC] Update dgl_graph.cc #19827

Merged
merged 1 commit into from
Feb 3, 2021
Merged

[DOC] Update dgl_graph.cc #19827

Changes from all commits
Commits
Show all changes
1 commit
Select commit
File filter

Filter by extension

Filter by extension
Conversations
Failed to load comments.
Jump to
Jump to file
Failed to load files.
Diff view
Diff view
4 changes: 2 additions & 2 deletions src/operator/contrib/dgl_graph.cc
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -763,7 +763,7 @@ NNVM_REGISTER_OP(_contrib_dgl_csr_neighbor_uniform_sample)
uniform probability. The operator is designed for DGL.

The operator outputs three sets of NDArrays to represent the sampled results
(the number of NDArrays in each set is the same as the number of seed NDArrays):
(the number of NDArrays in each set is the same as the number of seed NDArrays minus two (csr matrix and probability)):
1) a set of 1D NDArrays containing the sampled vertices, 2) a set of CSRNDArrays representing
the sampled edges, 3) a set of 1D NDArrays indicating the layer where a vertex is sampled.
The first set of 1D NDArrays have a length of max_num_vertices+1. The last element in an NDArray
Expand Down Expand Up @@ -868,7 +868,7 @@ NNVM_REGISTER_OP(_contrib_dgl_csr_neighbor_non_uniform_sample)
non-uniform probability. The operator is designed for DGL.

The operator outputs four sets of NDArrays to represent the sampled results
(the number of NDArrays in each set is the same as the number of seed NDArrays):
(the number of NDArrays in each set is the same as the number of seed NDArrays minus two (csr matrix and probability)):
1) a set of 1D NDArrays containing the sampled vertices, 2) a set of CSRNDArrays representing
the sampled edges, 3) a set of 1D NDArrays with the probability that vertices are sampled,
4) a set of 1D NDArrays indicating the layer where a vertex is sampled.
Expand Down