ColossalAI/docs/colossalai/colossalai.nn.parallel.layers.rst
ver217 d068af81a3
[doc] update rst and docstring (#1351)
* update rst

* add zero docstr

* fix docstr

* remove fx.tracer.meta_patch

* fix docstr

* fix docstr

* update fx rst

* fix fx docstr

* remove useless rst
2022-07-21 15:54:53 +08:00

15 lines
326 B
ReStructuredText

colossalai.nn.parallel.layers
=============================
.. automodule:: colossalai.nn.parallel.layers
:members:
.. toctree::
:maxdepth: 2
colossalai.nn.parallel.layers.colo_module
colossalai.nn.parallel.layers.embedding
colossalai.nn.parallel.layers.linear
colossalai.nn.parallel.layers.module_utils