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: