ColossalAI/colossalai/nn/layer/parallel_2p5d
Zirui Zhu 598cde4a0f [NFC] polish colossalai/nn/layer/parallel_2p5d/layers.py code style (#972) 2022-05-17 10:25:06 +08:00
..
__init__.py
_operation.py [TP] change the check assert in split batch 2d (#772) 2022-04-16 21:29:57 +08:00
_utils.py
layers.py [NFC] polish colossalai/nn/layer/parallel_2p5d/layers.py code style (#972) 2022-05-17 10:25:06 +08:00