Path	Lines of Code
mesh_tensorflow/auto_mtf/api_test.py	96
mesh_tensorflow/auto_mtf/graph_interface_test.py	247
mesh_tensorflow/auto_mtf/layout_optimizer_test.py	92
mesh_tensorflow/auto_mtf/memory_estimator_test.py	40
mesh_tensorflow/auto_mtf/scheduler_test.py	76
mesh_tensorflow/auto_mtf/valid_layouts_test.py	37
mesh_tensorflow/experimental/data_aug_lib_test.py	60
mesh_tensorflow/experimental/input_reader_test.py	80
mesh_tensorflow/import_test.py	11
mesh_tensorflow/layers_test.py	524
mesh_tensorflow/ops_test.py	550
mesh_tensorflow/simd_mesh_impl_test.py	42
mesh_tensorflow/test_utils.py	38
mesh_tensorflow/test_utils_test.py	36
mesh_tensorflow/transformer/adaptive_softmax_test.py	423
mesh_tensorflow/transformer/dataset_test.py	92
mesh_tensorflow/transformer/funnel_transformer_test.py	137
mesh_tensorflow/transformer/learning_rate_schedules_test.py	35
mesh_tensorflow/transformer/memory_layers_test.py	95
mesh_tensorflow/transformer/transformer_layers_test.py	263
mesh_tensorflow/transformer/utils_test.py	130
mesh_tensorflow/transformer/vocab_embeddings_test.py	400
mesh_tensorflow/utils_test.py	25
oss_scripts/oss_tests.sh	13