Jiarui Fang 3d907faede [Gemini] add an inline_op_module to common test models and polish unitests. (#2004)

2022-11-23 16:55:54 +08:00

..

utils

[unit test] Refactored test cases with component func (#339)

2022-03-11 15:50:28 +08:00

__init__.py

[Gemini] add an inline_op_module to common test models and polish unitests. (#2004)

2022-11-23 16:55:54 +08:00

bert.py

[Tensor] fix test_model (#916)

2022-05-06 18:06:22 +08:00

gpt.py

[NFC] polish test component gpt code style (#1567)

2022-09-08 16:34:09 +08:00

inline_op_model.py

[Gemini] add an inline_op_module to common test models and polish unitests. (#2004)

2022-11-23 16:55:54 +08:00

nested_model.py

[zero] new interface for ShardedOptimv2 (#406)

2022-03-14 20:48:41 +08:00

no_leaf_module.py

[zero] adapt zero for unsharded paramters (Optimizer part) (#601)

2022-04-01 20:10:47 +08:00

registry.py

[test] refactored testing components (#324)

2022-03-11 15:50:28 +08:00

repeated_computed_layer.py

[zero] new interface for ShardedOptimv2 (#406)

2022-03-14 20:48:41 +08:00

resnet.py

[zero] new interface for ShardedOptimv2 (#406)

2022-03-14 20:48:41 +08:00

simple_net.py

[Tensor] add 1d vocab loss (#918)

2022-05-07 15:49:14 +08:00