Move & update TBE docstring (pytorch#3169)

Summary: X-link: facebookresearch/FBGEMM#263 Move the TBE docstring from `fbgemm_gpu/fbgemm_gpu/docs/table_batched_embedding_ops.py` into `fbgemm_gpu/fbgemm_gpu/split_table_batched_embeddings_ops_training.py` Pull Request resolved: pytorch#3169 Test Plan: See example page https://deploy-preview-3169--pytorch-fbgemm-docs.netlify.app/fbgemm_gpu-python-api/table_batched_embedding_ops Reviewed By: q10, jianyuh Differential Revision: D63327715 Pulled By: sryap fbshipit-source-id: da372b41fdd651b8cc4ec3fd463b2dd35ae152a1
ROCm · Sep 25, 2024 · 73c4667 · 73c4667
1 parent 825e1a3
commit 73c4667
Show file tree

Hide file tree

Showing 3 changed files with 315 additions and 176 deletions.
diff --git a/fbgemm_gpu/docs/src/fbgemm_gpu-python-api/table_batched_embedding_ops.rst b/fbgemm_gpu/docs/src/fbgemm_gpu-python-api/table_batched_embedding_ops.rst
@@ -1,6 +1,5 @@
 Table Batched Embedding (TBE) Operators
 =======================================
 
-.. automodule:: fbgemm_gpu
-
-.. autofunction:: fbgemm_gpu.split_table_batched_embeddings_ops.SplitTableBatchedEmbeddingBagsCodegen
+.. autoclass:: fbgemm_gpu.split_table_batched_embeddings_ops_training.SplitTableBatchedEmbeddingBagsCodegen
+    :members:
diff --git a/fbgemm_gpu/fbgemm_gpu/docs/table_batched_embedding_ops.py b/fbgemm_gpu/fbgemm_gpu/docs/table_batched_embedding_ops.py