Path	Lines of Code
contrib/k8s/test_train_mp_mnist.yaml	35
test/args_parse.py	40
test/bench.py	106
test/cpp/cpp_test_util.cpp	333
test/cpp/cpp_test_util.h	73
test/cpp/main.cpp	7
test/cpp/metrics_snapshot.cpp	76
test/cpp/metrics_snapshot.h	39
test/cpp/run_tests.sh	62
test/cpp/test_async_task.cpp	51
test/cpp/test_aten_xla_tensor.cpp	9990
test/cpp/test_ir.cpp	67
test/cpp/test_mayberef.cpp	18
test/cpp/test_op_by_op_executor.cpp	57
test/cpp/test_replication.cpp	80
test/cpp/test_tensor.cpp	663
test/cpp/test_xla_util_cache.cpp	32
test/cpp/torch_xla_test.cpp	68
test/cpp/torch_xla_test.h	32
test/gcsfs_test.py	96
test/metrics_compare_utils_test.py	160
test/pytorch_test_base.py	548
test/run_tests.sh	86
test/schedulers.py	79
test/test_async_closures.py	58
test/test_autocast.py	178
test/test_data_type.py	53
test/test_mp_all_gather.py	24
test/test_mp_all_to_all.py	29
test/test_mp_collective_permute.py	28
test/test_mp_distributed_mm.py	35
test/test_mp_mesh_reduce.py	23
test/test_mp_reduce_scatter.py	27
test/test_mp_rendezvous.py	29
test/test_mp_replication.py	25
test/test_mp_save.py	53
test/test_mp_sync_batch_norm.py	103
test/test_operations.py	1687
test/test_ops.py	278
test/test_profile_mp_mnist.py	182
test/test_profiler.py	76
test/test_syncfree_optimizers.py	101
test/test_train_mp_imagenet.py	243
test/test_train_mp_imagenet_amp.py	270
test/test_train_mp_mnist.py	165
test/test_train_mp_mnist_amp.py	184
test/test_xla_dist.py	772
torch_xla/test/__init__.py	1
torch_xla/test/test_utils.py	82