Commit be85a0f3 authored by Frank Lee's avatar Frank Lee Committed by ver217
Browse files

removed tutorial markdown and refreshed rst files for consistency

parent ca4ae52d
colossalai.nn.layer.moe
=======================
.. automodule:: colossalai.nn.layer.moe
:members:
.. toctree::
:maxdepth: 2
colossalai.nn.layer.moe.layers
colossalai.nn.layer.non\_parallel\_layers
======================================
.. automodule:: colossalai.nn.layer.non_parallel_layers
:members:
colossalai.nn.layer.parallel\_1d
================================
.. automodule:: colossalai.nn.layer.parallel_1d
:members:
.. toctree::
:maxdepth: 2
colossalai.nn.layer.parallel_1d.layers
.. automodule:: colossalai.nn.layer.parallel_1d
:members:
colossalai.nn.layer.parallel\_2d
================================
.. automodule:: colossalai.nn.layer.parallel_2d
:members:
.. toctree::
:maxdepth: 2
colossalai.nn.layer.parallel_2d.layers
.. automodule:: colossalai.nn.layer.parallel_2d
:members:
colossalai.nn.layer.parallel\_2p5d
==================================
.. automodule:: colossalai.nn.layer.parallel_2p5d
:members:
.. toctree::
:maxdepth: 2
colossalai.nn.layer.parallel_2p5d.layers
.. automodule:: colossalai.nn.layer.parallel_2p5d
:members:
colossalai.nn.layer.parallel\_3d
================================
.. automodule:: colossalai.nn.layer.parallel_3d
:members:
.. toctree::
:maxdepth: 2
colossalai.nn.layer.parallel_3d.layers
.. automodule:: colossalai.nn.layer.parallel_3d
:members:
colossalai.nn.layer.parallel\_sequence
======================================
.. automodule:: colossalai.nn.layer.parallel_sequence
:members:
.. toctree::
:maxdepth: 2
colossalai.nn.layer.parallel_sequence.layers
.. automodule:: colossalai.nn.layer.parallel_sequence
:members:
colossalai.nn.layer
===================
.. automodule:: colossalai.nn.layer
:members:
.. toctree::
:maxdepth: 2
colossalai.nn.layer.colossalai_layer
colossalai.nn.layer.moe
colossalai.nn.layer.parallel_1d
colossalai.nn.layer.parallel_2d
colossalai.nn.layer.parallel_2p5d
colossalai.nn.layer.parallel_3d
colossalai.nn.layer.parallel_sequence
colossalai.nn.layer.non_parallel_layers
colossalai.nn.layer.utils
colossalai.nn.layer.vanilla
colossalai.nn.layer.wrapper
colossalai.nn.layer.base_layer
.. automodule:: colossalai.nn.layer
:members:
.. toctree::
:maxdepth: 2
colossalai.nn.layer.base_layer
colossalai.nn.layer.utils.common
================================
.. automodule:: colossalai.nn.layer.utils.common
:members:
colossalai.nn.layer.utils
=========================
.. automodule:: colossalai.nn.layer.utils
:members:
.. toctree::
:maxdepth: 2
colossalai.nn.layer.utils.common
colossalai.nn.layer.vanilla.layers
==================================
.. automodule:: colossalai.nn.layer.vanilla.layers
:members:
colossalai.nn.layer.vanilla
===========================
.. automodule:: colossalai.nn.layer.vanilla
:members:
.. toctree::
:maxdepth: 2
colossalai.nn.layer.vanilla.layers
colossalai.nn.layer.wrapper.pipeline\_wrapper
=============================================
.. automodule:: colossalai.nn.layer.wrapper.pipeline_wrapper
:members:
......@@ -9,3 +9,4 @@ colossalai.nn.layer.wrapper
:maxdepth: 2
colossalai.nn.layer.wrapper.lambda_wrapper
colossalai.nn.layer.wrapper.pipeline_wrapper
colossalai.nn.loss.cross\_entropy\_2d
=====================================
.. automodule:: colossalai.nn.loss.cross_entropy_2d
:members:
colossalai.nn.loss.cross\_entropy\_2p5d
=======================================
.. automodule:: colossalai.nn.loss.cross_entropy_2p5d
:members:
colossalai.nn.loss.cross\_entropy\_3d
=====================================
.. automodule:: colossalai.nn.loss.cross_entropy_3d
:members:
colossalai.nn.loss.loss\_2d
===========================
.. automodule:: colossalai.nn.loss.loss_2d
:members:
colossalai.nn.loss.loss\_2p5d
=============================
.. automodule:: colossalai.nn.loss.loss_2p5d
:members:
colossalai.nn.loss.loss\_3d
===========================
.. automodule:: colossalai.nn.loss.loss_3d
:members:
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment