ColossalAI/docs/colossalai/colossalai.nn.layer.paralle...

5 lines
127 B
ReStructuredText

colossalai.nn.layer.parallel\_2d
================================
.. automodule:: colossalai.nn.layer.parallel_2d
:members: