From a5dc025c70934ba712b77fac5dfd7706432010d2 Mon Sep 17 00:00:00 2001
From: Alex Barghi <105237337+alexbarghi-nv@users.noreply.github.com>
Date: Tue, 7 Oct 2025 10:17:47 -0700
Subject: [PATCH] Update docs tree.

---
 .../source/api_docs/cugraph-pyg/cugraph_pyg.rst  | 16 ++++++++++++++++
 1 file changed, 16 insertions(+)

diff --git a/docs/cugraph-docs/source/api_docs/cugraph-pyg/cugraph_pyg.rst b/docs/cugraph-docs/source/api_docs/cugraph-pyg/cugraph_pyg.rst
index 3b1cfcd..2552206 100644
--- a/docs/cugraph-docs/source/api_docs/cugraph-pyg/cugraph_pyg.rst
+++ b/docs/cugraph-docs/source/api_docs/cugraph-pyg/cugraph_pyg.rst
@@ -20,6 +20,15 @@ Feature Storage
    
    cugraph_pyg.data.feature_store.FeatureStore
 
+Tensors and Embeddings
+----------------------
+.. autosummary::
+   :toctree ../api/cugraph-pyg/
+
+   cugraph_pyg.tensor.dist_tensor.DistTensor
+   cugraph_pyg.tensor.dist_tensor.DistEmbedding
+   cugraph_pyg.tensor.dist_matrix.DistMatrix
+
 
 Data Loaders
 ------------
@@ -29,6 +38,9 @@ Data Loaders
    cugraph_pyg.loader.node_loader.NodeLoader
    cugraph_pyg.loader.neighbor_loader.NeighborLoader
 
+   cugraph_pyg.loader.link_loader.LinkLoader
+   cugraph_pyg.loader.link_neighbor_loader.LinkNeighborLoader
+
 Samplers
 --------
 .. autosummary::
@@ -37,4 +49,8 @@ Samplers
    cugraph_pyg.sampler.sampler.BaseSampler
    cugraph_pyg.sampler.sampler.SampleReader
    cugraph_pyg.sampler.sampler.HomogeneousSampleReader
+   cugraph_pyg.sampler.sampler.HeterogeneousSampleReader
    cugraph_pyg.sampler.sampler.SampleIterator
+
+   cugraph_pyg.sampler.distributed_sampler.BaseDistributedSampler
+   cugraph_pyg.sampler.distributed_sampler.DistributedNeighborSampler