Skip to content

Commit a5dc025

Browse files
Update docs tree.
1 parent b321bac commit a5dc025

File tree

1 file changed

+16
-0
lines changed

1 file changed

+16
-0
lines changed

docs/cugraph-docs/source/api_docs/cugraph-pyg/cugraph_pyg.rst

Lines changed: 16 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -20,6 +20,15 @@ Feature Storage
2020

2121
cugraph_pyg.data.feature_store.FeatureStore
2222

23+
Tensors and Embeddings
24+
----------------------
25+
.. autosummary::
26+
:toctree ../api/cugraph-pyg/
27+
28+
cugraph_pyg.tensor.dist_tensor.DistTensor
29+
cugraph_pyg.tensor.dist_tensor.DistEmbedding
30+
cugraph_pyg.tensor.dist_matrix.DistMatrix
31+
2332

2433
Data Loaders
2534
------------
@@ -29,6 +38,9 @@ Data Loaders
2938
cugraph_pyg.loader.node_loader.NodeLoader
3039
cugraph_pyg.loader.neighbor_loader.NeighborLoader
3140

41+
cugraph_pyg.loader.link_loader.LinkLoader
42+
cugraph_pyg.loader.link_neighbor_loader.LinkNeighborLoader
43+
3244
Samplers
3345
--------
3446
.. autosummary::
@@ -37,4 +49,8 @@ Samplers
3749
cugraph_pyg.sampler.sampler.BaseSampler
3850
cugraph_pyg.sampler.sampler.SampleReader
3951
cugraph_pyg.sampler.sampler.HomogeneousSampleReader
52+
cugraph_pyg.sampler.sampler.HeterogeneousSampleReader
4053
cugraph_pyg.sampler.sampler.SampleIterator
54+
55+
cugraph_pyg.sampler.distributed_sampler.BaseDistributedSampler
56+
cugraph_pyg.sampler.distributed_sampler.DistributedNeighborSampler

0 commit comments

Comments
 (0)