File tree Expand file tree Collapse file tree 1 file changed +16
-0
lines changed
docs/cugraph-docs/source/api_docs/cugraph-pyg Expand file tree Collapse file tree 1 file changed +16
-0
lines changed Original file line number Diff line number Diff 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
2433Data 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+
3244Samplers
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
You can’t perform that action at this time.
0 commit comments