ColossalAI/tests/components_to_test
Ziyue Jiang f593a5637e
[Tensor] add embedding tp1d row (#904)
2022-04-29 14:10:05 +08:00
..
utils
__init__.py [Tensor] init a tp network training unittest (#849) 2022-04-24 16:43:44 +08:00
bert.py
nested_model.py
no_leaf_module.py
registry.py
repeated_computed_layer.py
resnet.py
simple_net.py [Tensor] add embedding tp1d row (#904) 2022-04-29 14:10:05 +08:00