From e4df36cb09cdb4b23e98a9c8d347934197860821 Mon Sep 17 00:00:00 2001 From: Paul Zhang Date: Tue, 9 Jan 2024 12:55:40 -0800 Subject: [PATCH] Expose ZCH modules in Documentation (#1613) Summary: Expose the ZCH modules in the documentation for the TorchRec page Differential Revision: D52603324 --- docs/source/torchrec.distributed.rst | 16 ++++++++++++++++ docs/source/torchrec.modules.rst | 12 ++++++++++-- 2 files changed, 26 insertions(+), 2 deletions(-) diff --git a/docs/source/torchrec.distributed.rst b/docs/source/torchrec.distributed.rst index 9fdcb4282..1329221ab 100644 --- a/docs/source/torchrec.distributed.rst +++ b/docs/source/torchrec.distributed.rst @@ -124,3 +124,19 @@ torchrec.distributed.utils :members: :undoc-members: :show-inheritance: + +torchrec.distributed.mc_modules +-------------------------- + +.. automodule:: torchrec.distributed.mc_modules + :members: + :undoc-members: + :show-inheritance: + +torchrec.distributed.mc_embeddingbag +-------------------------- + +.. automodule:: torchrec.distributed.mc_embeddingbag + :members: + :undoc-members: + :show-inheritance: diff --git a/docs/source/torchrec.modules.rst b/docs/source/torchrec.modules.rst index 95dcf35db..e2c9ee771 100644 --- a/docs/source/torchrec.modules.rst +++ b/docs/source/torchrec.modules.rst @@ -76,10 +76,18 @@ torchrec.modules.utils :undoc-members: :show-inheritance: -Module contents +torchrec.modules.mc_modules --------------- -.. automodule:: torchrec.modules +.. automodule:: torchrec.modules.mc_modules + :members: + :undoc-members: + :show-inheritance: + +torchrec.modules.mc_embedding_modules +--------------- + +.. automodule:: torchrec.modules.mc_embedding_modules :members: :undoc-members: :show-inheritance: