ColossalAI/docs/colossalai/colossalai.context.process_...

6 lines
255 B
ReStructuredText

colossalai.context.process\_group\_initializer.initializer\_2d
==============================================================
.. automodule:: colossalai.context.process_group_initializer.initializer_2d
:members: Initializer_2D_Row, Initializer_2D_Col